Programmers / 3단계 / 두 큐 합 같게 만들기 / python / 2022 KAKAO TECH INTERNSHIP

 

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

 

 

나의 풀이

from collections import deque 
def solution(queue1, queue2):
    q1 = deque(queue1)
    q2 = deque(queue2)
    
    s1 = sum(q1)
    s2 = sum(q2) 
    sumnum = s1 + s2
    
    if sumnum % 2 == 1:
        return -1
    
    cnt = 0
    while True:
        
        if s1 > s2 and len(q1) != 0:
            num = q1.popleft()
            q2.append(num)
            s1 -= num
            s2 += num
            cnt += 1 
        
        elif s2 > s1 and len(q2) != 0:
            num = q2.popleft()
            q1.append(num)
            s2 -= num
            s1 += num
            cnt += 1
        
        else:
            return cnt

 

 

테스트 3 시간 초과 - while 문 위에서 두 큐의 합이 홀수일 때 -1을 리턴하도록 처리하였으나, 두 큐의 합이 짝수임에도 똑같이 나눌 수 없는 경우를 처리하는 부분을 구현하지 못하였다.

 

모범 답안

from collections import deque 
def solution(queue1, queue2):
    q1 = deque(queue1)
    q2 = deque(queue2)
    
    limit = len(q1) * 4
    s1 = sum(q1)
    s2 = sum(q2) 
    
    sumnum = s1 + s2
    
    if sumnum % 2 == 1:
        return -1
    
    cnt = 0
    while True:
        
        if s1 > s2 and len(q1) != 0:
            num = q1.popleft()
            q2.append(num)
            s1 -= num
            s2 += num
            cnt += 1 
        
        elif s2 > s1 and len(q2) != 0:
            num = q2.popleft()
            q1.append(num)
            s2 -= num
            s1 += num
            cnt += 1
        
        else:
            break
            
        if cnt > limit:
            return -1
        
    return cnt

 반복문 제한을 두기 위해 limit로 조건을 걸었다. queue1과 queue2의 모든 원소를 바꾸면 queue1 길이의 2배만큼 횟수가 필요하고 다시 모든 원소를 바꿔 원래의 모습으로 만들면 queue1 길이의 2배만큼 필요해 총 len(q1) * 4 만큼 횟수가 필요하기 때문에 limit를 len(q1)로 두었다.

 

참조

 

https://a-littlecoding.tistory.com/123

[프로그래머스] 두 큐 합 같게 만들기_Python level2(스택/큐)