[백준 15734] 명장 남정훈 (Swift)

15734번: 명장 남정훈

왼발잡이 선수, 오른발잡이 선수, 양발잡이 선수의 수가 각각 주어진다.

왼발잡이 선수와 오른발잡이 선수의 수를 같게 만들고, 그 수를 구하는 문제.

이때 양발잡이 선수는 왼발잡이 또는 오른발잡이로 취급할 수 있다.

왼발잡이와 오른발잡이 선수의 수를 같게 만들면서 최대가 되게 해야 한다.

풀이 (1)

나의 풀이.

왼발 ↔ 오른발 은 교환이 되지 않으며, 오직 양발잡이만 왼발/오른발잡이로 배치할 수 있다.

따라서 남아있는 양발잡이의 수만큼 반복문을 돌며 확인한다.

• 만약 왼발잡이가 오른발잡이보다 많다면, 양발잡이 한명을 왼발잡이로 보낸다.
• 만약 오른발잡이가 왼발잡이보다 많다면, 양발잡이 한명을 오른발잡이로 보낸다.
• 만약 왼발잡이와 오른발잡이의 수가 같고, 양발잡이가 2명 이상이라면, 양발잡이 두명을 왼발잡이와 오른발잡이에 한명씩 보낸다.
• 만약 왼발잡이의 오른발잡이의 수가 같고, 양발잡이가 1명이라면, 반복문을 끝낸다.
  • 기본적으로는 반복문의 조건식에 의해, 양발잡이의 수가 0일 때 반복문을 탈출한다. 하지만 이 경우 양발잡이의 수가 0이 아닌데도 더 이상 왼발잡이/오른발잡이에 배치할 수 없다. 따라서 이 경우도 탈출하게 한다.

반복문을 통해 양발잡이들을 모두 배치한 다음에도 왼발잡이와 오른발잡이의 수가 같지 않을 수 있다.

따라서 답은 왼발잡이와 오른발잡이 중 작은 쪽을 2배로 한 값으로 구했다.

let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
var left = inputArr[0]
var right = inputArr[1]
var both = inputArr[2]
while both > 0 {
    if left > right {
        both -= 1
        right += 1
        continue
    }
    if left < right {
        both -= 1
        left += 1
        continue
    }
    if left == right && both >= 2 {
        both -= 2
        left += 1
        right += 1
        continue
    }
    if left == right && both == 1 {
        break
    }
}
print(min(left, right) * 2)

풀이 (2)

다른 풀이를 보다 보니 더 간결한 풀이가 있어서 가져왔다.

나는 왼발잡이와 오른발잡이가 같은 경우를 2가지로 나누어 처리했는데 이 풀이는 왼발이나 오른발 중 한쪽의 같거나 큰 경우에 포함해 처리했다.

이렇게 처리해도 되는 이유는 왼발/오른발 중 한쪽에 +1을 하면, 다음 시행 때 어차피 반대쪽이 맞춰지기 때문이다.

시간복잡도는 처음 풀이와 동일하다. (8ms)

let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
var left = inputArr[0]
var right = inputArr[1]
var both = inputArr[2]
while both != 0 {
    if left <= right {
        left += 1
        both -= 1
        continue
    }
    if left > right {
        right += 1
        both -= 1
        continue
    }
}
print(min(left, right) * 2)