본문 바로가기
알고리즘/알고리즘

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

by 두둠칫 2022. 8. 1.

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

1. 첫코드 & 개선가능점

import java.util.HashMap;

class Solution {
    public int[] solution(int[] lottos, int[] win_nums) {
        int[] answer = new int[2];
        
        HashMap<Integer, Integer> map = new HashMap<>();
        
        for(int l : lottos){
            if(map.containsKey(l)){ // lottos에 중복없으니 필요없는 validation
                map.put(l, map.get(l)+1);   
            }
            else{
                map.put(l, 1);
            }
        }
        
        // 위 반복문에서 1~45만 map에 넣고 0이면 zeros에 카운팅
        
        int zeros = 0, wins = 0;
        
        // 여기선 wins만 카운팅
        for(HashMap.Entry<Integer, Integer> e : map.entrySet()){
            int k = e.getKey();
            int v = e.getValue();
            
            if(k==0)
                zeros += v;
            else{
                for(int n : win_nums){
                    if(k==n){
                        wins++;
                        break;
                    }
                }
            }
        }        
        
        answer[0] = (7 - (wins+zeros))<=6?(7 - (wins+zeros)):6;
        answer[1] = (7 - wins)<=6?(7 - wins):6;
        
        return answer;
    }
}