틀렸습니다! 모음집 - 2021년 04월 14일(수)

나름대로 풀었는데, 틀렸거나 시간 초과인 코드 모음집

BOJ: #2870 - 수학숙제

입력이 들어오는 문자열에서 숫자는 배열에 따로 저장해서 처리했다. 모든 값이 0인 경우에는 allSatisfy()를 이용해서 처리했고, 그렇지 않은 경우에는 모든 0을 정규식으로 처리했다. 그리고 정렬은 swift string number sort로 구글링해서 나온 걸 사용했다. 근데 틀렸음… 왜 틀렸는지 모르겠당

문제

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

내가 작성한 코드

Swift

import Foundation
let n = Int(readLine()!)!
var nums = [String]()
for _ in 0..<n {
    let input = readLine()!
    var num = [Character]()
    for char in input {
        if char.isNumber {
            num.append(char)
        } else {
            if !num.isEmpty {
                if num.allSatisfy({$0 == "0"}) {
                    nums.append("0")
                } else {
                    let trimmed = String(num).replacingOccurrences(of: "^0+", with: "", options: .regularExpression)
                    nums.append(trimmed)
                    num = []
                }
            }
        }
    }
 
    if !num.isEmpty {
        if num.allSatisfy({$0 == "0"}) {
            nums.append("0")
        } else {
            let trimmed = String(num).replacingOccurrences(of: "^0+", with: "", options: .regularExpression)
            nums.append(trimmed)
            num = []
        }
    }
}
 
// 정렬 부분 고치기
nums.sort(by: { $0.localizedStandardCompare($1) == .orderedAscending})
//nums.sort(by: { $0.compare($1, options: .numeric) == .orderedAscending})
for num in nums {
    print(num)
}

BOJ: #2852 - NBA 농구

대회에서 출제됐던 문제라 대회 홈페이지에서 테스트 케이스를 봤다. 중간에 문제에 대한 이해를 잘못했다는 걸 느꼈는데 어떻게 고쳐야할 지 모르겠어서 보류.

문제

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

내가 작성한 코드

Swift

import Foundation
let n = Int(readLine()!)!
var info = [[String]]()
 
for _ in 0..<n {
    let victory = readLine()!.split(separator: " ").map { String($0) }
    info.append(victory)
}
 
var prevWin = info[0][0]
var prevTime = info[0][1].split(separator: ":").map { Int(String($0))! }
var firstTeam = [0, 0]
var secondTeam = [0, 0]
 
for i in 1..<n {
    if prevWin != info[i][0] { // 이기고 있던 시간
        let time = info[i][1].split(separator: ":").map { Int(String($0))! }
        var minute = time[0] - prevTime[0]
        var second = time[1] - prevTime[1]
        if second < 0 {
            minute -= 1
            second += 60
        }
 
        if prevWin == "1" { // 1번 팀이 이기고 있던 시간
            firstTeam[0] += minute
            firstTeam[1] += second
        } else { // 2번 팀이 이기고 있던 시간
            secondTeam[0] += minute
            secondTeam[1] += second
        }
 
        prevWin = info[i][0]
        prevTime = time
    }
}
 
if prevWin == info[n-1][0] { // 마지막까지 이기고 있었으면
    let time = info[n-1][1].split(separator: ":").map { Int(String($0))! }
    var overMinute = 48 - time[0]
    var overSecond = -time[1]
    if overSecond < 0 {
        overMinute -= 1
        overSecond += 60
    }
 
    if prevWin == "1" { // 1번 팀이 이기고 있던 시간
        firstTeam[0] += overMinute
        firstTeam[1] += overSecond
    } else { // 2번 팀이 이기고 있던 시간
        secondTeam[0] += overMinute
        secondTeam[1] += overSecond
    }
}
 
print(String(format: "%02d", firstTeam[0]) + ":" + String(format: "%02d", firstTeam[1]))
print(String(format: "%02d", secondTeam[0]) + ":" + String(format: "%02d", secondTeam[1]))

BOJ: #1436 - 영화감독 숌

단순하게 생각해서 666부터 n개의 666이 들어가는 숫자 배열까지 순회해서 삽입했는데, 시간 초과. 하긴, Xcode에서 9999를 할 때 느리긴 했다. 시간을 더 어떻게 줄여야할 지 모르겠어서 패스.

문제

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

내가 작성한 코드

Swift

import Foundation
let n = Int(readLine()!)!
var nums = [String]()
var i = 666
while nums.count != n {
    if String(i).contains("666") {
        nums.append(String(i))
    }
    i += 1
}
 
print(nums[n-1])

BOJ: #14620 - 꽃길

탐색 문제처럼 풀었는데, 어디가 잘못된 거야…? 난 모르겠네 /_\

문제

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

내가 작성한 코드

Swift

let n = Int(readLine()!)!
var price = [[Int]]()
for _ in 0..<n {
    let line = readLine()!.split(separator: " ").map { Int(String($0))! }
    price.append(line)
}
 
var seeds = 3
var board = Array(repeating: Array(repeating: 0, count: n), count: n)
var ans = 0
while seeds != 0 {
    var dict = [[Int]:Int]()
    for i in 0..<n {
        for j in 0..<n {
            // 꽃이 온전히 피어야함
            if j - 1 < 0 || j + 1 >= n || i - 1 < 0 || i + 1 >= n {
                continue
            }
 
            // 이미 꽃을 심은 곳은 안됨
            if board[i][j] == 1 || board[i][j-1] == 1 || board[i][j+1] == 1 || board[i+1][j] == 1 || board[i-1][j] == 1 {
                continue
            }
 
            // 일단 가격과 위치를 저장해두자
            let sum = price[i][j] + price[i][j+1] + price[i][j-1] + price[i+1][j] + price[i-1][j]
            dict[[i, j]] = sum
        }
    }
 
 
    if let minPos = dict.min(by: { $0.value <= $1.value }) {
        ans += minPos.value
        board[minPos.key[0]][minPos.key[1]] = 1
        board[minPos.key[0]][minPos.key[1] + 1] = 1
        board[minPos.key[0]][minPos.key[1] - 1] = 1
        board[minPos.key[0] + 1][minPos.key[1]] = 1
        board[minPos.key[0] - 1][minPos.key[1]] = 1
        seeds -= 1
    }
}
 
print(ans)

BOJ: #4195 - 친구 네트워크

BFS/DFS 유형의 문제라고 생각해서 풀었는데, 31% 쯤에서 시간 초과. 질문 게시판을 보니까 이런식으로 푸는 거 같진 않은데 다른 방식은 생각이 안 나서 패스.

문제

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

내가 작성한 코드

Swift

let t = Int(readLine()!)!
for _ in 0..<t {
    let f = Int(readLine()!)!
    var graph = [String: [String]]()
    for _ in 0..<f {
        let friend = readLine()!.split(separator: " ").map { String($0) }
        // 양방향 친구 관계
        graph[friend[0], default: []].append(friend[1])
        graph[friend[1], default: []].append(friend[0])
        // 친구 관계가 생길 때마다 네트워크 탐색하기
        var visited: Set = [friend[0]]
        var queue: Set = [friend[0]]
        while !queue.isEmpty {
            let v = queue.removeFirst()
 
            for w in graph[v]! {
                if !visited.contains(w) {
                    visited.insert(w)
                    queue.insert(w)
                }
            }
        }
        print(visited.count)
    }
}