[프로그래머스] level.3 - 선입 선출 스케줄링 (C++)

https://school.programmers.co.kr/learn/courses/30/lessons/12920

문제 설명

처리해야 할 동일한 작업이 n 개가 있고, 이를 처리하기 위한 CPU가 있습니다.

이 CPU는 다음과 같은 특징이 있습니다.

• CPU에는 여러 개의 코어가 있고, 코어별로 한 작업을 처리하는 시간이 다릅니다.
• 한 코어에서 작업이 끝나면 작업이 없는 코어가 바로 다음 작업을 수행합니다.
• 2개 이상의 코어가 남을 경우 앞의 코어부터 작업을 처리 합니다.

처리해야 될 작업의 개수 n과, 각 코어의 처리시간이 담긴 배열 cores 가 매개변수로 주어질 때, 마지막 작업을 처리하는 코어의 번호를 return 하는 solution 함수를 완성해주세요.

제한 사항

• 코어의 수는 10,000 이하 2이상 입니다.
• 코어당 작업을 처리하는 시간은 10,000이하 입니다.
• 처리해야 하는 일의 개수는 50,000개를 넘기지 않습니다.

---

입출력 예

n	cores	result
6	[1,2,3]	2

입출력 예 설명

입출력 예 #1

처음 3개의 작업은 각각 1,2,3번에 들어가고, 1시간 뒤 1번 코어에 4번째 작업,다시 1시간 뒤 1,2번 코어에 5,6번째 작업이 들어가므로 2를 반환해주면 됩니다.

내 해답

#include <string>
#include <vector>

using namespace std;

int solution(int n, vector<int> cores) 
{
    if(n <= cores.size())
        return n;
    
    int min_time = -1;
    int max_time = 2e5;
    while(min_time+1 < max_time)
    {
        int avg_time = (min_time+max_time)/2;
        int count = cores.size();            
        
        for(const auto& i : cores)
            count += (avg_time / i);
            
        if(count >= n)
            max_time = avg_time;
        else
            min_time = avg_time;
    }
    
    int count = cores.size();
    
    for(int i = 0; i < cores.size(); ++i)
        count += min_time/cores[i];
    for(int i = 0; i < cores.size(); ++i)
    {
        if((min_time+1) % cores[i] == 0) 
            ++count;
        if(count == n)
            return i+1;
    }
    
    return 0;
}