본문 바로가기

알고리즘/카카오

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

반응형

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


public static int solution(int[] queue1, int[] queue2) {
        long sum1 = 0;
        long sum2 = 0;
        long sum  = 0;
        
        Deque<Integer> q1 = new ArrayDeque<>();
        Deque<Integer> q2 = new ArrayDeque<>();
        
        for(int i=0; i<queue1.length; i++) {
        	q1.add(queue1[i]);
        	q2.add(queue2[i]);
        	
        	sum1 += queue1[i];
        	sum2 += queue2[i];
        }
        
        if(sum % 2 != 0) return -1;
        sum = (sum1+sum2)/2;
        
        int p1 = 0; int p2 = 0;
        int n = queue1.length * 2;
        
        while(p1 <= n || p2 <= n) {
        	if(sum1 == sum && sum2 == sum) {
        		return p1+p2; 
        	}
        	else if(sum1 < sum) {
        		sum1 += q2.peek();
        		sum2 -= q2.peek();
        		q1.add(q2.poll());
        		p2++;
        		
        	} else if(sum2 < sum) {
        		sum2 += q1.peek();
        		sum1 -= q1.peek();
        		q2.add(q1.poll());
        		p1++;
        	}
        }
        
        return -1;
    }
반응형