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

프로그래머스

---

성공 풀이

class Queue {
  private var new: [Int] = []
  private var old: [Int]
  var queue: [Int] { old.reversed()+new }
  
  init(_ input: [Int] ) { old = input.reversed() }
  
  func add(_ n: Int) {
    new.append(n)
  }
  
  func removeFirst() -> Int {
    guard !old.isEmpty || !new.isEmpty else { return 0 }
    if old.isEmpty {
      old = new.reversed()
      new.removeAll()
    }
    return old.removeLast()
  }
}

func solution(_ queue1:[Int], _ queue2:[Int]) -> Int {
  var q1 = Queue(queue1)
  var q2 = Queue(queue2)
  var sum1 = queue1.reduce(0,+)
  var sum2 = queue2.reduce(0,+)
  let goal = (sum1+sum2)/2
  var count = 0
  var result = -1
  
  while count < queue1.count*3 {
    if sum1 == sum2 { result = count; break }
    if sum1 > sum2 {
      let n = q1.removeFirst()
      if n == 0 { break }
      q2.add(n)
      sum2 += n
      sum1 -= n
    } else {
      let n = q2.removeFirst()
      if n == 0 { break }
      q1.add(n)
      sum1 += n
      sum2 -= n
    }
    count += 1
  }
  return result
}

풀이 키워드

- 투포인터로 배열의 index를 기억해가며 문제를 푼다면 더 간단하게 풀 수 있습니다.
- 하지만 queue를 직접 만들어 풀어보고 싶어서 풀이 할 수 있을 정도로만 queue를 만들어서 풀었습니다.
- 풀이의 핵심은 queue1의 합이 클 때는 queue1의 첫번째 값을 queue2로 보내고
queue2의 합이 클 때면 queue1로 보내면 최소 횟수를 구할 수 있습니다.

---