[ 프로그래머스 ] 로또의 최고 순위와 최저 순위

 

 

🍄 코딩테스트

코딩테스트 문제 풀이

✍🏻 Github

문제 풀이 github 링크

 

문제 설명

로또 6/45(이하 '로또'로 표기)는 1부터 45까지의 숫자 중 6개를 찍어서 맞히는 대표적인 복권입니다. 아래는 로또의 순위를 정하는 방식입니다.

순위 당첨 내용

1	6개 번호가 모두 일치
2	5개 번호가 일치
3	4개 번호가 일치
4	3개 번호가 일치
5	2개 번호가 일치
6(낙첨)	그 외

로또를 구매한 민우는 당첨 번호 발표일을 학수고대하고 있었습니다. 하지만, 민우의 동생이 로또에 낙서를 하여, 일부 번호를 알아볼 수 없게 되었습니다. 당첨 번호 발표 후, 민우는 자신이 구매했던 로또로 당첨이 가능했던 최고 순위와 최저 순위를 알아보고 싶어 졌습니다.

 

알아볼 수 없는 번호를 0으로 표기하기로 하고, 민우가 구매한 로또 번호 6개가 44, 1, 0, 0, 31 25라고 가정해보겠습니다. 당첨 번호 6개가 31, 10, 45, 1, 6, 19라면, 당첨 가능한 최고 순위와 최저 순위의 한 예는 아래와 같습니다.

당첨 번호	31	10	45	1	6	19	결과
최고 순위 번호	31	0→10	44	1	0→6	25	4개 번호 일치, 3등
최저 순위 번호	31	0→11	44	1	0→7	25	2개 번호 일치, 5등

• 순서와 상관없이, 구매한 로또에 당첨 번호와 일치하는 번호가 있으면 맞힌 걸로 인정됩니다.
• 알아볼 수 없는 두 개의 번호를 각각 10, 6이라고 가정하면 3등에 당첨될 수 있습니다.
  • 3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다.
• 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.
  • 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.

 

민우가 구매한 로또 번호를 담은 배열 lottos, 당첨 번호를 담은 배열 win_nums가 매개변수로 주어집니다. 이때, 당첨 가능한 최고 순위와 최저 순위를 차례대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

 

 

제한사항

• lottos는 길이 6인 정수 배열입니다.
• lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
  • 0은 알아볼 수 없는 숫자를 의미합니다.
  • 0을 제외한 다른 숫자들은 lottos에 2개 이상 담겨있지 않습니다.
  • lottos의 원소들은 정렬되어 있지 않을 수도 있습니다.
• win_nums은 길이 6인 정수 배열입니다.
• win_nums의 모든 원소는 1 이상 45 이하인 정수입니다.
  • win_nums에는 같은 숫자가 2개 이상 담겨있지 않습니다.
  • win_nums의 원소들은 정렬되어 있지 않을 수도 있습니다.

 

 

입출력 예

lottos	win_nums	result
[44, 1, 0, 0, 31, 25]	[31, 10, 45, 1, 6, 19]	[3, 5]
[0, 0, 0, 0, 0, 0]	[38, 19, 20, 40, 15, 25]	[1, 6]
[45, 4, 35, 20, 3, 9]	[20, 9, 3, 45, 4, 35]	[1, 1]

 

 

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

알아볼 수 없는 번호들이 아래와 같았다면, 1등과 6등에 당첨될 수 있습니다.

당첨 번호 38 19 20 40 15 25 결과

최고 순위 번호	0→38	0→19	0→20	0→40	0→15	0→25	6개 번호 일치, 1등
최저 순위 번호	0→21	0→22	0→23	0→24	0→26	0→27	0개 번호 일치, 6등

입출력 예 #3

민우가 구매한 로또의 번호와 당첨 번호가 모두 일치하므로, 최고 순위와 최저 순위는 모두 1등입니다.

 

---

 

 

문제 풀이

 

나의 풀이

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    
    // 0의 개수 구하기
    let number0 = lottos.filter{ $0 == 0 }.count
    
    // 현재 순위
    var rank = 7
    
    // 일치하는 값
    for i in win_nums {
        if lottos.contains(i) {
            rank -= 1
        }
    }
    
    let best = rank - number0
    
    return [ best == 7 ? 6 : best , rank == 7 ? 6 : rank ]
}

 

생각보다 간단했다.

0의 개수와 로또번호와 내번호 중 일치하는 값을 찾으면 되었는데, 순위를 어떻게 정하냐가 문제였던거 같다.

 

나는 생각을 조금 바꿔 현재 순위를 정해두고, 로또 번호와 일치하는 개수를 빼주는 방식으로 최하점을 정했다.

하지만 rank 가 7일 때가 문제였다.

그래서 그냥 삼항연산자를 이용해 해결해주었다.

 

여기서 나의 로또 번호 중 0을 제외하면 반복문에서 더 효율이 나오지 않을까라는 생각이 있었는데,

반복문의 개수가 크지 않고, 0을 제외하는 과정의 실행시간도 고려해보아서 뺐다.

 

 

다른 사람 풀이

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {

    let zeroCount = lottos.filter { $0 == 0}.count
    let winCount: Int = win_nums.filter { lottos.contains($0) }.count

    return [min(7-winCount-zeroCount,6), min(7-winCount,6)]
}

 

이 분은 lottos의 값과 winCount의 값을 찾을 때 고차함수를 사용했다.

내가 고민하였던 순위가 7인 부분을 min 함수를 이용해 푸셨다.

 

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

이 분은 새로운 관점에서 접근을 하셔서 신기했다.

두 가지의 배열을 집합에 넣어 교집합으로 푸셨다.

또한 순위가 7인 부분을 새로운 ranking 배열을 사용해서 해결하였다.

집합으로 풀었을 때와 실행시간을 비교해보고 싶어서 위의 코드를 살짝 바꾸어 주었다.

 

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    let mine = Set(lottos)
    let winner = Set(win_nums)
    let score = mine.intersection(winner).count
    let joker = lottos.filter { ($0 == 0) }.count
    
    let best = 7 - score - joker
    let worst = 7 - score
    return [ best == 7 ? 6 : best, worst == 7 ? 6 : worst ]
}

// 집합으로 푼 실행시간
테스트 1 〉	통과 (0.07ms, 16.4MB)
테스트 2 〉	통과 (0.07ms, 16.3MB)
테스트 3 〉	통과 (0.08ms, 16.3MB)
테스트 4 〉	통과 (0.07ms, 16.2MB)
테스트 5 〉	통과 (0.07ms, 16.3MB)
테스트 6 〉	통과 (0.09ms, 16.3MB)
테스트 7 〉	통과 (0.06ms, 16.5MB)
테스트 8 〉	통과 (0.05ms, 16.4MB)
테스트 9 〉	통과 (0.07ms, 16.4MB)
테스트 10 〉	통과 (0.05ms, 16.3MB)
테스트 11 〉	통과 (0.07ms, 16.2MB)
테스트 12 〉	통과 (0.04ms, 16.4MB)
테스트 13 〉	통과 (0.05ms, 16.4MB)
테스트 14 〉	통과 (0.06ms, 16.3MB)
테스트 15 〉	

// 반복문과 contains를 사용해 푼 실행시간
테스트 1 〉	통과 (0.05ms, 16.1MB)
테스트 2 〉	통과 (0.04ms, 16.3MB)
테스트 3 〉	통과 (0.04ms, 16.4MB)
테스트 4 〉	통과 (0.05ms, 16.4MB)
테스트 5 〉	통과 (0.08ms, 16.3MB)
테스트 6 〉	통과 (0.05ms, 16.4MB)
테스트 7 〉	통과 (0.04ms, 16.5MB)
테스트 8 〉	통과 (0.05ms, 16.2MB)
테스트 9 〉	통과 (0.04ms, 16.1MB)
테스트 10 〉	통과 (0.04ms, 16.3MB)
테스트 11 〉	통과 (0.04ms, 16.6MB)
테스트 12 〉	통과 (0.06ms, 16.4MB)
테스트 13 〉	통과 (0.05ms, 16MB)
테스트 14 〉	통과 (0.04ms, 16.3MB)
테스트 15 〉	통과 (0.05ms, 16.4MB)

 

집합보다 반복문과 contain가 조금 더 빠른 것 같다.