로또의 최고 순위와 최저 순위 - 프로그래머스(Swift)

로또의 최고 순위와 최저 순위 - 프로그래머스(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/77484

프로그래머스

 

 

 나의 풀이

 

0을 얼마나 담고있는지 카운트를 하고 당첨번호가 몇개인지를 카운트해서 switch문으로 문제를 해결했다.

 

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    
    var zeroBeforeLotto = win_nums.filter{ lottos.contains($0) }.count
    let zeroCount = lottos.filter{ $0 == 0 }.count
    
    var bestCount = zeroBeforeLotto + zeroCount
    var worstNum = 0
    
    var result:[Int] = []
    switch bestCount {
        case 6: 
        if zeroCount == 6 {
            result.append(1)
            result.append(0+zeroCount)   
        } else {
            result.append(1)
            result.append(1+zeroCount)
        }
        case 5:
        result.append(2)
        result.append(2+zeroCount)
        case 4:
        result.append(3)
        result.append(3+zeroCount)
        case 3:
        result.append(4)
        result.append(4+zeroCount)
        case 2:
        result.append(5)
        result.append(5+zeroCount)
        default:
        result.append(6)
        result.append(6)
    }
    
    return result
}

 

(switch문의 안에서의 하드코딩.. 보이시나요..?)

case6에서 if문을 넣은 것은 6개가 0일때  1+6을 하게되면 7위가 되기 때문이다. 

이런 절망적인 풀이 속에서 다른 사람의 풀이를 보고 내 풀이를 개선했다! 

 

 

 다른 사람의 풀이

 

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    
    var winCount = win_nums.filter{ lottos.contains($0) }.count
    let zeroCount = lottos.filter{ $0 == 0 }.count
    
    return [min(7-winCount-zeroCount, 6), min(7-winCount, 6)]
}

 

아름다운 풀이다. 당첨개수에 따라서 순위가 변동되는 것을 7-winCount로 표현했고(당첨된 번호의 개수가 5개라면 2등이기 때문), 당첨개수도 없고 0의 개수도 없을때 7이되는데 이를 min함수를 통해 6개로 제한한 점이 인상깊다.

 

 

 다른 사람의 풀이 - Set자료구조

 

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    
    let mine = Set(lottos)
    let wins = Set(win_nums)
    let score = mine.intersection(wins).count
    let joker = lottos.filter{ $0 == 0 }.count
    let ranking = [6, 6, 5, 4, 3, 2, 1]
    
    return [ranking[score+joker], ranking[score]]
    
}

 

ranking 배열을 통해서 1개일때와 2개일때의 등수를 넣을 수 있고 이를 통해 직관적인 풀이가 가능했다.

(역시 다른 사람들의 풀이를 봐야 생각이 확장되는 것 같다)

 

 

 피드백

 

1. 등수를 체크할 때 0위는 없기 때문에 7에서 빼주고 min함수로 최소값을 정해주는 개념을 숙지하자.

2. intersection함수는 Set메서드에서 사용가능하다.