Kotlin / 로또의 최고 순위와 최저 순위

<문제>

문제 설명

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

순위	당첨 내용
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등에 당첨될 수 있습니다.
• 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.

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

제한사항

• lottos는 길이 6인 정수 배열입니다.
• lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
• win_nums은 길이 6인 정수 배열입니다.
• win_nums의 모든 원소는 1 이상 45 이하인 정수입니다.

<내 풀이>

class Solution {
    fun solution(lottos: IntArray, winNums: IntArray): IntArray {
        var cnt = 0
        var cnt_1 = 7 // 최저
        val answer = IntArray(2)

        for (lotto in lottos) {
            if (lotto == 0) {
                cnt++
                println("cnt $cnt")
            } else {
                for (winNum in winNums) {
                    if (lotto == winNum) {
                        cnt_1--
                        println("cnt_1 $cnt_1")
                    }
                }
            }
        }

        if (cnt_1 == 7 && cnt == 0) {
            answer[0] = 6
            answer[1] = 6
        } else {
            if (cnt_1 == 7) {
                answer[1] = 6
            } else {
                answer[1] = cnt_1 // 최저 등수
            }
            if (cnt == 0) {
                answer[0] = cnt_1 // 최고 등수
            } else {
                answer[0] = cnt_1 - cnt // 최고 등수
            }
        }

        return answer
    }
}

 

1. lottos를 반복하며 lotto가 0이면 cnt를 증가시킨다. (선택하지 않은 번호)

2. 0이 아닌 경우 winNums와 비교하여 일치하는 번호가 있으면 cnt_1 감소 

• cnt_1이 7이고 cnt가 0이면, 모든 번호가 맞아 최고 순위와 최저 순위가 모두 6등
• 그 외의 경우, cnt_1과 cnt에 따라 최고 순위와 최저 순위를 계산

 

<다른사람 풀이>

class Solution {
    fun solution(lottos: IntArray, winNums: IntArray): IntArray {
        return intArrayOf(
                (lottos.size.plus(1)) - lottos.filter { winNums.contains(it) || it == 0 }.size,
                (lottos.size.plus(1)) - lottos.filter(winNums::contains).size
        ).map { if (it > 6) it - 1 else it }.toIntArray()
    }
}

 

lotto를 filter로 사전에 정리하고, map을 이용하여 원하는 값 만들어내기

 

___