[프로그래머스] 두 큐 합 같게 만들기(C++)

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

프로그래머스

큐1와 큐2를 하나의 큐로 합쳐서 사용한다.

테케 1을 예시로 들자면,

 

원래 각각의 시작점을 a,b로 두어 접근할 수 있도록 하는 것이다.

sum 비교에 따라 빠져나오거나, sum과 a,b를 조정해주면 된다.

b가 큐 크기를 벗어난다면, 더 이상 접근 불가능한 것이므로 -1으로 지정하고 빠져나온다.

 

#include <string>
#include <vector>
#include <numeric>

using namespace std;

int solution(vector<int> queue1, vector<int> queue2) {
    int answer = 0;

    long long int sum1 = accumulate(queue1.begin(), queue1.end(), 0);
    long long int sum2 = accumulate(queue2.begin(), queue2.end(), 0);
    vector<int>queue;
    for(int i=0;i<queue1.size();i++)
        queue.push_back(queue1[i]);
    for(int i=0;i<queue2.size();i++)
        queue.push_back(queue2[i]);
    
    //접근할 인덱스
    int a=0;
    int b=queue1.size();
    
    while(1)
    {
        //더이상 접근 불가
        if(b>=queue.size())
        {
            answer=-1;
            break;
        }
        
        if(sum1==sum2)
            break;
        else if(sum1>sum2)
        {
            sum1-=queue[a];
            sum2+=queue[a];
            a++;
        }
        else
        {
            sum2-=queue[b];
            sum1+=queue[b];
            b++;
        }
        answer++;
    }
    
    return answer;
}