백준 - 스위치 켜고 끄기(Swift)

백준 - 스위치 켜고 끄기(Swift)

 

 

 문제 설명

 

https://www.acmicpc.net/problem/1244

1244번: 스위치 켜고 끄기

 

 나의 풀이

 

스위치의 개수와 학생의 수가 100이하 이므로 완전탐색을 할 수 있고 이로인해 구현문제인 것을 알 수 있다.

모든 구현문제가 그렇듯이 항상 한 문장을 제대로 읽지 않아서 틀리는 경우가 많다.

나의 경우에는 구간에 속한 스위치의 개수는 홀수라는 문장을 간과했었는데, 여자가 3을 뽑았다면 3 또한 뒤집어줘야하기 때문에 이를 고려했어야 했다.

 

import Foundation

let n = Int(readLine()!)!
var states = readLine()!.components(separatedBy:" ").map{Int($0)!} // 스위치 상태
let total = states.count
states.insert(0, at: 0) // 1번 인덱스부터 편하게 하기위해 삽입
let num = Int(readLine()!)! // 학생 수

for _ in 0..<num {
    let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
    let type = input[0], card = input[1] // 성별, 받은 숫자
    var ran = 1
    switch type {
    case 1: // 남자일 때
        let share = total / card // 배수가 되는 개수
        for i in 1..<share+1 {
            let idx = card*i
            toggle(idx)
        }
    case 2: // 여자일 때
        while true {
            if check(card, ran) { // 범위안 값들이 같다면 toggle후 더 깊게 탐색
                toggle(card-ran)
                toggle(card+ran)
            } else {
                toggle(card)
                break
            }
            ran += 1
        }
    default:
        break
    }
}

states.removeFirst() // 처음 삽입한 것을 지운다.
if states.count < 21 {
    print(states.map{String($0)}.joined(separator:" "))
} else {
    let p = total / 20
    let rest = total % 20
    var idx = -20
    for _ in 0..<p {
        idx += 20
        for j in 0..<20 {
            print(states[idx+j], terminator: " ")
        }
        print()
    }
    if rest != 0 {
        idx += 20
        for j in 0..<rest {
            print(states[idx+j], terminator: " ")
        }
    }
    
}


// states[idx]의 상태를 바꿔준다.
func toggle(_ idx: Int) {
    if states[idx] == 0 {
        states[idx] = 1
    } else {
        states[idx] = 0
    }
}

// 여자일 때 인덱스 범위를 확인해서 가능하면 true 아니면 false
func check(_ idx: Int, _ range: Int) -> Bool {
    if idx-range < 1 || idx+range > total { // 인덱스의 범위가 넘어가면 false
        return false
    }
    return states[idx-range] == states[idx+range] ? true : false
}

 

 다른 사람의 풀이

 

20개씩 끊어야 하는 부분을 몫과 나머지를 구해서 처리했었는데 다른 사람의 풀이를 보니 더욱 간편한 방법이 있었다.

인덱스가 20번째일때는 출력하고 줄넘김만 하면 20보다 작을때도 처리할 수 있게 된다.

 

for (i, x) in states.enumerated() {
    if (i+1) % 20 == 0 {
        print(x, terminator: "\n")
    } else { print(x, terminator: " ")}
}