[파이썬, Java] 로또의 최고 순위와 최저 순위

문제 설명

로또 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등에 당첨될 수 있습니다.
  • 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등에 당첨될 수 있습니다.

당첨 번호381920401525결과

최고 순위 번호	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등입니다.

---

1. 실제로 사용되는 로또 순위의 결정 방식과는 약간 다르지만, 이 문제에서는 지문에 명시된 대로 로또 순위를 결정하도록 합니다.  ↩

 

 

🐝 나의 풀이

def solution(lottos, win_nums):
    answer = [0]*2
    cnt=0
    unknown=0
    rank={6:1, 5:2, 4:3, 3:4, 2:5}
    
    for lotto in lottos :
        if lotto in win_nums :  # 당첨
            cnt+=1
        elif lotto == 0 :   # 모르는 번호
            unknown+=1
    
    if cnt+unknown >= 1 :
        answer[0]=rank[cnt+unknown]
    else :
        answer[0]=6
        
    if cnt <= 1 :    
        answer[1]=6
    else :
        answer[1]=rank[cnt]
    
    return answer   # [최고, 최저]

cnt는 맞춘 번호의 갯수다.

unknown은 0의 갯수다.

최고 순위가 될려면 현재 맞춘 번호의 갯수 cnt+unknown 모두 맞추기

최저 순위는 현재 맞춘 번호의 갯수다.

rank 딕셔너리에서 key는 맞춰야 하는 번호의 갯수, 값은 순위다.

순위가 6위인 경우는 넣지 않았기 때문에 조건문으로 처리했다.

 

 

import java.util.*;

class Solution {
    public int[] solution(int[] lottos, int[] win_nums) {
        int[] answer = new int[2];
        int cnt=0;
        int unknown=0;
        int[] rank={6, 6, 5, 4, 3, 2, 1};
        
        for(int i : lottos){
            if(Arrays.stream(win_nums).anyMatch(j -> j==i)){
                cnt++;
            }else if(i == 0){
                unknown++;
            }
        }
        
        answer[0]=rank[cnt+unknown];
        answer[1]=rank[cnt];

        return answer;
    }
}

다른 사람의 파이썬 풀이를 자바로 구현해봤다.

if(Arrays.asList(win_nums).contains(lottos[i]))

그리고 처음에는 이렇게 했는데 안된다 ㅠ

true가 나와야 할 것이 false가 나오는...

 

 

🐍 다른 사람 풀이

def solution(lottos, win_nums) :
    rank=[6, 6, 5, 4, 3, 2, 1]
    cnt_0=lottos.count(0)
    ans=0
    for x in win_nums :
        if x in lottos :
            ans+=1
    return rank[cnt_0+ans], rank[ans]

맞춘 번호의 갯수를 인덱스로 rank에서 값을 꺼내 반환한다.

푸는 방법은 나랑 같지만 인덱스, 값을 잘 활용해서 깔끔한 풀이다.

 

 

def solution(lottos, win_nums) :
    rank={
        0:6,
        1:6,
        2:5,
        3:4,
        4:3,
        5:2,
        6:1
    }
    return [rank[len(set(lottos)&set(win_nums))+lottos.count(0)], rank[len(set(lottos)&set(win_nums))]]

집합을 사용했다.

and 연산으로 당첨된 번호의 집합을 알아낸다.

 

 

import java.util.*;
import java.util.stream.LongStream;

class Solution {
    public int[] solution(int[] lottos, int[] winNums) {
        return LongStream.of(
                (lottos.length+1)-Arrays.stream(lottos).filter(l -> Arrays.stream(winNums).anyMatch(w -> w==l) || l==0).count(),
                (lottos.length+1)-Arrays.stream(lottos).filter(l -> Arrays.stream(winNums).anyMatch(w -> w==l)).count()
        )
                .mapToInt(op -> (int)(op > 6 ? op-1 : op))
                .toArray();
    }
}

stream을 잘 활용했다.

 

 

import java.util.*;

class Solution {
    public int[] solution(int[] lottos, int[] win_nums) {
        Map<Integer, Boolean> map=new HashMap<Integer, Boolean>();
        int zeroCount=0;
        
        for(int lotto : lottos){
            if(lotto == 0){
                zeroCount++;
                continue;
            }
            map.put(lotto, true);
        }
        
        int sameCount=0;
        for(int winNum : win_nums){
            if(map.containsKey(winNum)) sameCount++;
        }
        
        int maxRank=7-(sameCount+zeroCount);
        int minRank=7-sameCount;
        if(maxRank > 6) maxRank=6;
        if(minRank > 6) minRank=6;
        
        return new int[]{maxRank, minRank};
    }
}

lotto 배열에서 0은 zeroCount를 1 증가시키고, 나머지 값들은 HashMap에 넣는다.

win_nums 배열을 반복하면서 HashMap에 있으면 sameCount를 1 증가시킨다.

 

 

 

문제 출처 👉 프로그래머스