알고리즘(Algorithm)/프로그래머스(Programmers)

[프로그래머스] 완주하지 못한 선수 - JAVA

나아가는중 2021. 10. 21. 23:12
반응형

프로그래머스 완주하지 못한 선수 - JAVA


문제 설명

  1. 마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 retrun.

제한 사항

  1. 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
  2. completion의 길이는 participant의 길이보다 1 작습니다.
  3. 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
  4. 참가자 중에는 동명이인이 있을 수 있습니다.

문제 풀이

  1. 참가자 중 완주자를 빠르고 효율적으로 찾기 위해 Hash를 사용.
  2. 동명이인이 존재하기 때문에 HashMap을 사용하여 동명이인의 수를 저장.
  3. 참가자 이름과 숫자(동명이인 수)를 HashMap에 저장.
    • getOrDefault를 사용하여 HashMap에 존재하는지 검사하고 존재하면 1을 증가, 존재하지 않을시 1을 저장.
  4. 완주자 이름을 HashMap에서 찾아 수가 1인 경우 제거, 여럿 존재하는 경우 1을 감소.
  5. 4의 작업이 완료되면 1명의 미완주자만 남게되어 answer에 저장 후 return.

소스 코드

import java.util.*;

class Solution {
    public String solution(String[] participant, String[] completion) {        
        Map<String, Integer> map = new HashMap<>();

        for (String s : participant) { // 3
            map.put(s, map.getOrDefault(s, 0) + 1);
        }

        for (String s : completion) { // 4
            if (map.get(s) == 1) {
                map.remove(s);
            } else {
                map.put(s, map.get(s) - 1);
            }
        }

        String answer = ""; // 5

        for (String s : map.keySet()) {
            answer = s;
        }

        return answer;
    }
}
반응형