[Swift] 백준 2138 - 전구와 스위치
풀이
예를 들어 스위치 A, B, C가 있고 세 스위치는 모두 켜져있다고 하자.
A B C
0 0 0
이와 같은 상태일 때
- 스위치 A를 누르면
A B C
X X 0
-
스위치 B를 누르면
A B C
0 0 X -
스위치 C를 누르면
A B C
0 X 0
이와 같은 변화로 알 수 있는 것은
스위치를 누르고 난 후에 그 상태가 고정되는 것은 눌러진 스위치가 아니라
눌러진 스위치 바로 전의 스위치라는 것이다.
스위치 B의 상태는 B를 눌렀을 때 0 이었지만, C를 누르고 난 후에 X로 최종적으로 결정 된다.
따라서 i번째 스위치는
current[i-1] != future[i-1] 인 경우에 누르면 되는 것이다.
for i in 1..<N {
if current[i-1] != future[i-1] {
// 스위치 누름
count += 1
}
}
그러나 이렇게 하면 i가 0일 때인 가장 처음 스위치를 누르는 경우는 고려하지 못하게 된다.
0번째 스위치는 눌러질 수도 눌러지지 않을 수도 있기 때문에 누르는 경우, 누르지 않는 경우를 나누어서 두 번 연산해주면 된다.
제출한 코드
import Foundation
let N = Int(readLine()!)!
var current = Array(readLine()!).map{Int(String($0))!}
var future = Array(readLine()!).map{Int(String($0))!}
var result = Int.max
func isSame(_ current: [Int]) -> Bool {
var temp = current
var count = 0
for i in 1..<N {
if temp[i-1] != future[i-1] {
var changeRange = [i-1,i,i+1]
if i == (N-1) {
changeRange = [i-1, i]
}
for j in changeRange {
temp[j] = temp[j] == 0 ? 1 : 0
}
count += 1
}
}
let isSame = temp == future
result = isSame ? min(result, count) : result
return isSame
}
if isSame(current) {
print(result)
} else {
current[0] = current[0] == 0 ? 1 : 0
current[1] = current[1] == 0 ? 1 : 0
if isSame(current) {
print(result+1)
} else {
print(-1)
}
}
Author And Source
이 문제에 관하여([Swift] 백준 2138 - 전구와 스위치), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@sun02/Swift-백준-2138-전구와-스위치저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)