백준 - 올림픽(Swift)

백준 - 올림픽(Swift)

 

 

 문제 설명

 

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

8979번: 올림픽

 

 나의 풀이

 

이 문제가 구현으로 분류된 문제가 아니였다면 단순 구현 문제라는 것을 어떻게 알았을까?

문제는 단순히 금, 은, 동만을 비교해서 등수를 정한다. 문제의 입력조건도 작고 최적을 찾는 문제는 아니기 때문에 단순히 구현하는 능력을 보는 문제라는 것을 알 수 있을 것이다.

 

문제가 재밌던 점은 여러개의 속성을 넣은 프로퍼티와 ==함수를 만들면서 실제로 개발하는 느낌이 들었다. 마지막으로 id를 넣음으로써 원하고자하는 등수를 얻을 수 있게 되었다.

 

import Foundation

struct Country {
    var id: Int
    var gold: Int
    var silver: Int
    var bronze: Int
    var prize: Int
    
    static func ==(_ lhs: Country, _ rhs: Country) -> Bool {
        return (lhs.gold == rhs.gold) && (lhs.silver == rhs.silver) && (lhs.bronze == rhs.bronze) 
    }
}

let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
let N = input[0], Fnum = input[1]
var countries: [Country] = []
for i in 0..<N {
    let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
    countries.append(Country(id: input[0], gold: input[1], silver: input[2], bronze: input[3], prize: 0))
}
countries.sort(by: {$0.gold != $1.gold 
                    ? $0.gold > $1.gold 
                    : ($0.silver == $1.silver 
                       ? $0.bronze > $1.bronze : $0.silver > $1.silver)})
var cur = countries[0]
var prize = 1
var cnt = 1
countries[0].prize = 1
for i in 1..<countries.count {
    cnt += 1
    if countries[i] == cur {
        countries[i].prize = prize
    } else {
        cur = countries[i]
        prize = cnt
        countries[i].prize = prize
    }
}

print(countries.filter{$0.id == Fnum}[0].prize)