BOJ: #1759 - 암호 만들기

23:52

처음에는 모음의 개수만 체크했는데, 자음의 개수도 체크해야 한다는 걸 보고.. 수정했더니 통과.

• 중복이 없어야 하니까 visited 변수로 방문 여부를 확인
• 암호에서 증가하는 순서대로만 나올 수 있다고 해서 ascii 값으로 확인

문제

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

내가 작성한 코드

Swift

let x = readLine()!.split(separator: " ").map { Int(String($0))! }
let l = x[0]
let c = x[1]
let alpha = readLine()!.split(separator: " ").map { String($0) }

var picked = [String]()
var results = [String]()
var visited = Array(repeating: false, count: c)
func pick(toPick: Int) {
    if toPick == 0 {
        results.append(picked.joined())
        return
    }

    for i in 0..<c {
        if visited[i] {
            continue
        }
        if let last = picked.last {
            if Character(last).asciiValue! > Character(alpha[i]).asciiValue! {
                continue
            }
        }
        visited[i] = true
        picked.append(alpha[i])
        pick(toPick: toPick - 1)
        picked.removeLast()
        visited[i] = false
    }
}

pick(toPick: l)
results.sort()
let vowels = Set(["a", "e", "i", "o", "u"])
for r in results {
    var count = 0
    var countC = 0
    for c in r {
        if vowels.contains(String(c)) {
            count += 1
        } else {
            countC += 1
        }
    }
    if count >= 1 && countC >= 2 {
        print(r)
    }
}