백준 - 빌런호석(Swift)

백준 - 빌런호석(Swift)

 

 

 문제 설명

 

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

22251번: 빌런 호석

 

 나의 풀이

 

문제를 보고 어떻게 풀어야할지 감이 잘 안왔다.

딱 문제만 이해한 느낌.. 

그래서 다른 사람의 풀이를 보면서 감을 찾았다.

 

먼저 K가 6까지기 때문에 N이 표현될 수 있는 수는 999,999까지이다. 즉 NlogN까지는 가능하다.

 

그리고 주어진 숫자에서 특정 층수를 갈 수 있는지를 확인해야하기 떄문에 1층부터 N층까지 먼저 전부 탐색을한다.

이후 일의자리부터 K자리까지 비교를 하는데 그 안에서 또 LED가 같은지 다른지를 판별을 해서 다르다면 cnt를 올려준다.

이렇게 함으로써 특정 숫자가 특정 층수가 되기위해 변환되어야 할 LED숫자를 알 수 있게되고 이게 P보다 작거나 같다면 결과값으로 넣을 수 있게 된다.

 

나머지연산자를 통해서 일의자리를 구하는 것과 나누기를 통해서 십의자리 백의자리또한 구하는 걸 배웠다.

 

 

import Foundation

let ch: [[Bool]] = [
    [true, true, true, false, true, true, true], //0
    [false, false, true, false, false, false, true], //1
    [false, true, true, true, true, true, false], //2
    [false, true, true, true, false, true, true], //3
    [true, false, true, true, false, false, true], //4
    [true, true, false, true, false, true, true], //5
    [true, true, false, true, true, true, true], //6
    [false, true, true, false, false, false, true], //7
    [true, true, true, true, true, true, true, true], //8
    [true, true, true, true, false, true, true] //9
]

let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
let N = input[0], K = input[1], P = input[2], X = input[3]
var result = 0
for i in 1..<N+1 {
    if i==X { // 같은 숫자라면 넘어간다.
        continue
    }
    var cnt = 0 // 바뀐 LED의 숫자
    var from = X, to = i
    for k in 0..<K { // 자릿수만큼 진행. K가 3이라면 999가 최대값이니 3번 진행
        for j in 0..<7 { // 숫자 하나당 디스플레이 개수
            if ch[from%10][j] != ch[to%10][j] { cnt += 1 } //LED가 다르다면 전환
        }
        from /= 10 
        to /= 10
    }
    if cnt <= P { // 바뀐 LED의 개수가 최대값보다 작다면
        result += 1
    }
}
print(result)