study record

[프로그래머스-자바] 해시 Hash 본문

알고리즘

[프로그래머스-자바] 해시 Hash

asong 2021. 5. 3. 11:15

완주하지못한 선수

  • HashMap을 이용한 문제 풀이
  • hash.put(key, value), hash.get(key)를 이용한 문제풀이를 진행했다.
import java.util.HashMap;

public class Hash {
    public static void main(String[] args) {
        String[] participant = {"marina", "josipa", "nikola", "vinko", "filipa"};
        String[] completion = {"josipa", "filipa", "marina", "nikola"};
        System.out.println(solution(participant, completion));
    }

    public static String solution(String[] participant, String[] completion) {
        String answer = "";

        HashMap<String, Integer> hash = new HashMap<>();
        for (String player : participant) {
            if (hash.containsKey(player)) {
                hash.put(player, hash.get(player) + 1);
            } else {
                hash.put(player, 1);
            }
        }
        for (String winner : completion) {
            hash.put(winner, hash.get(winner) - 1);
        }
        for (String player : participant) {
            if (hash.get(player) != 0) {
                answer = player;
            }
        }
        return answer;
    }
}

 

전화번호목록

  • HashMap을 이용해 hashMap.containsKey()를 이용해 찾는 방법
import java.util.HashMap;

public class PhoneNumberList {
    public boolean solution(String[] phone_book) {
        boolean answer = true;

        HashMap<String, String> hashMap = new HashMap<String, String>();

        for(int i=0; i< phone_book.length;i++){
            hashMap.put(phone_book[i], "prefix");
        }

        for(String phone: phone_book){
            for(int idx = 0; idx < phone.length();idx++){
                if(hashMap.containsKey(phone.substring(0,idx))){
                    answer = false;
                    break;
                }
            }
        }
        return answer;
    }
}

 

위장

  • HashMap, hashMap.getOrDefault() 사용
  • hashMap키들의 세트는 hashMap.keySet !
import java.util.HashMap;

public class Disguise {
    public int solution(String[][] clothes) {
        int answer = 1;
        HashMap<String, Integer> hashMap = new HashMap<>();

        for (String[] cloth : clothes) {
            hashMap.put(cloth[1], hashMap.getOrDefault(cloth[1], 0) + 1);
        }

        for (String key : hashMap.keySet()) {
            answer *= hashMap.get(key) + 1;
        }

        return answer - 1;
    }
}

 

베스트앨범

  • 해시맵을 통해 장르별 재생수를 더하여 저장
  • Collections.sort()와 compareTo를 이용해 장르별 재생수 내림차순으로 정렬
  • 차례차례 정렬된 장르 중 최대 재생 값 두 개를 찾는다.
  • ArrayList를 answer[]배열로 바꾸면 끝
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;

public class BestAlbum {
    public int[] solution(String[] genres, int[] plays) {
        HashMap<String, Integer> hashMap = new HashMap<>();
        for(int i=0;i<genres.length;i++){
            hashMap.put(genres[i],hashMap.getOrDefault(genres[i],0)+plays[i]);
        }
        ArrayList<String> glist = new ArrayList<String>(hashMap.keySet());
        Collections.sort(glist, (o1, o2) -> (hashMap.get(o2).compareTo(hashMap.get(o1))));

        ArrayList<Integer> answers = new ArrayList<>();


        for(int i=0;i<glist.size();i++){
            int max=0;
            int index1=0;
            int index2=0;
            for(int j=0;j<genres.length;j++){
                if(glist.get(i).equals(genres[j])){

                    if(max<plays[j]){
                        max = plays[j];
                        index1 = j;
                    }
                }
            }

            max = 0;
            for(int j=0;j<genres.length;j++){
                if(glist.get(i).equals(genres[j])){

                    if(max<plays[j] && j!=index1){
                        max = plays[j];
                        index2 = j;
                    }
                }
            }

            answers.add(index1);
            if(max != 0){
                answers.add(index2);
            }
        }

        int[] answer = new int[answers.size()];
        for(int i=0;i<answers.size();i++){
            answer[i] = answers.get(i);
        }

        return answer;
    }
}