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

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

코딩테스트 연습 - 로또의 최고 순위와 최저 순위

 

저는 로또 번호에 어떤 숫자가 나왔는지를 구하는 함수(init),

번호를 맞춘 개수와 0의 개수를 구하는 함수(calCorrrect),

맞춘 개수에 따른 순위를 구하는 함수(changeCntToGrade)로 나눠서 문제를 해결하였습니다.

 

 

import java.util.*;
class Solution {
    private static void init(int[] win_nums,Set<Integer> set){
        for(int num : win_nums){
            set.add(num);
        }
    }
    private static int[] calCorrect(int[] lottos,Set<Integer> numbers){
        int correctCnt = 0;
        int zeroCnt = 0;
        for(int lotto : lottos){
            if(numbers.contains(lotto)){
                correctCnt++;
            }else if(lotto == 0){
                zeroCnt++;
            }
        }
        return new int[] {correctCnt,correctCnt + zeroCnt};
    }
    private static int changeCntToGrade(int correctCnt){
        if(correctCnt == 6){
            return 1;
        }else if(correctCnt == 5){
            return 2;
        }else if(correctCnt == 4){
            return 3;
        }else if(correctCnt == 3){
            return 4;
        }else if(correctCnt == 2){
            return 5;
        }else{
            return 6;
        }
    }
    public int[] solution(int[] lottos, int[] win_nums) {
        int[] answer = new int[2];
        Set<Integer> numbers = new HashSet<>();
        init(win_nums,numbers);
        int[] result = calCorrect(lottos,numbers);
        answer[0] = changeCntToGrade(result[1]);
        answer[1] = changeCntToGrade(result[0]);
        return answer;
    }
}