[Programmers] 두 큐 합 같게 만들기

문제 링크입니다: https://school.programmers.co.kr/learn/courses/30/lessons/118667

프로그래머스

문제에서 주어진대로 풀면 되는 문제였습니다.

어떤 방법으로도 두 큐의 합을 같게 만들 수 없는 경우 하나의 큐를 두 벡터의 크기 합만큼 pop 했는데도 불구하고 조건을 만족하지 못한 경우입니다.

 

	#include <string>
	#include <vector>
	#include <queue>
	using namespace std;
	int solution(vector<int> queue1, vector<int> queue2) {
	long long sum = 0, sum2 = 0;
	queue<int> q, q2;
	for (int num : queue1) {
	sum += num;
	q.push(num);
	}
	for (int num : queue2) {
	sum2 += num;
	q2.push(num);
	}
	int idx = 0, idx2 = 0;
	int maxIdx = queue1.size() + queue2.size();
	int answer = 0;
	while (idx < maxIdx && idx2 < maxIdx)
	{
	if (sum == sum2)
	{
	return answer;
	}
	answer++;
	if (sum < sum2)
	{
	int cur = q2.front();
	q2.pop();
	q.push(cur);
	sum += cur;
	sum2 -= cur;
	idx2++;
	}
	else
	{
	int cur = q.front();
	q.pop();
	q2.push(cur);
	sum2 += cur;
	sum -= cur;
	idx++;
	}
	}
	return -1;
	}

view raw .cpp hosted with ❤ by GitHub

 

개발환경: Programmers IDE

 

지적, 조언, 질문 환영합니다! 질문 남겨주세요~