[programmers/Lv.2] 두 큐 합 같게 만들기

❓  문제

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

 

🛎️  아이디어

- 1차 아이디어
  queue이기 때문에 둘다 list -> queue 형태로 만들어서 popleft() 사용

- 2차 아이디어
  queue1, queue2를 연결해서 하나의 리스트로 만든 후 그 안에서 규칙을 찾는다.
  -> 예외 내용으로 인해 정규식이 나오지 않음.

- 3차 아이디어
  블로그 참고해서 합이 큰 큐 -> 합이 작은 큐로 원소를 하나씩 이동시키는 아이디어를 얻음
  + while True로 시간초과 걸려서 찾아본 결과 300000로 제한 걸거나 중간에 count를 넣어서하는 방법이 있음.

 

💻  풀이

from collections import deque

def solution(queue1, queue2):
  q1, q2 = deque(queue1), deque(queue2)
  sum1, sum2 = sum(q1), sum(q2)

  for i in range(300000):
    if sum1 == sum2:
      return i
    elif sum1 > sum2:
      num = q1.popleft()
      q2.append(num)
      sum1 -= num
      sum2 += num
    else:
      num = q2.popleft()
      q1.append(num)
      sum2 -= num
      sum1 += num
  return -1

print(solution([3, 2, 7, 2], [4, 6, 5, 1]))