[Swift] 프로그래머스 위클리 챌린지 6주차 복서 정렬하기

 

Problem

 

코딩테스트 연습 - 6주차

---

Solution

 

1. 복서가 이기고 진 전적과 몸무게가 더 많이 나가는 선수를 이긴 횟수를 세준다.

 

N은 싸우지 않았을 경우이므로 생략해주고 W일 경우나 L일 경우만 세줍니다.

 

그 중 만약 몸무게가 더 적다면 count를 +1 해줍니다.

 

func setFightsHistory(moreWeightsWinCount:inout [Int],_ head2head:[String],weights:[Int]) -> [[String]] {
    var fights:[[String]] = Array(repeating: [String](), count: weights.count)
    for (i,head) in head2head.enumerated() {
       let map = head.map{String($0)}
        for (j,m) in map.enumerated() {
            if m == "W" {
                fights[i].append("W")
                if weights[i] < weights[j] {
                    moreWeightsWinCount[i] += 1
                }
            }
            if m == "L" {
                fights[i].append("L")
            }
        }
    }
    return fights
}

 

2. Boxer 객체를 만들고 필요한 정보를 넣어준다.

 

index,몸무게,승률,몸무게가 더 많이 나가는 선수 이긴 횟수를 속성으로 세팅해줍니다.

 

struct Boxer {
    let index:Int
    let weight:Int
    let winRate:Double
    let moreWeightsWinCount:Int
}

 

그리곤 각 정보를 담아 Boxer를 만들어줍니다.

 

index는 선수들 번호가 1번부터 시작이므로 +1을 해주고

 

승률은 싸운횟수/이긴횟수로 세팅해줍니다.

 

func makeBoxers(weights:[Int],fights:[[String]],moreWeightsWinCount:[Int]) -> [Boxer] {
    var boxers:[Boxer] = []
    for (i,w) in weights.enumerated() {
        var winRate:Double = 0
        if fights[i].filter({$0 == "W"}).count > 0 && fights[i].count > 0 {
            winRate = Double(fights[i].filter{$0 == "W"}.count)/Double(fights[i].count)
        }
        let boxer = Boxer(index:i+1,weight: w, winRate: winRate, moreWeightsWinCount: moreWeightsWinCount[i])
        boxers.append(boxer)
    }
    return boxers
}

 

3. 복서들을 조건에 맞게 정렬해주고 index를 반환해줍니다.

 

func sortBoxersIndex(boxers:[Boxer]) -> [Int] {
    return boxers.sorted{
        if $0.winRate == $1.winRate {
            if $0.moreWeightsWinCount == $1.moreWeightsWinCount {
                if $0.weight ==  $1.weight {
                    return $0.index < $1.index
                }else {
                    return $0.weight > $1.weight
                }
            }else {
                return $0.moreWeightsWinCount > $1.moreWeightsWinCount
            }
        }else {
            return $0.winRate > $1.winRate
        }
    }.map{$0.index}
}

---

Source Code

 

---

P.S

 

1단계치곤 좀 어려웠던 문제같다.. 

 

처음에 승률을 구할 때 10/싸운횟수 * 이긴횟수로 계산을 했었는데 계속 7번,9번,10번 테스트 케이스가 틀렸다...

 

그냥 싸운횟수/이긴횟수로 했으면 됐는데 ;;; 

 

근데 왜 10/싸운횟수 * 이긴횟수로 하면 안되는거지?