코딩테스트#004 완주하지 못한 선수 (해시)

A Kind Dev·2022년 7월 8일
0

코딩테스트

목록 보기
4/19

문제 설명

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

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

나의 풀이

import java.util.HashMap;
import java.util.Map;

class Solution {
    public String solution(String[] participant, String[] completion) {
        
        String answer = "";
        
        Map<String, Integer> playerMap = new HashMap<String, Integer>();
        
        // 전체 선수 셋팅
        for (String player : participant) {
            playerMap.put(player, playerMap.getOrDefault(player, 0) + 1);
        }
        
        // 완주한 선수 셋팅
        for (String player : completion) {
            playerMap.put(player, playerMap.get(player) -1);
        }
        
        for (String key : playerMap.keySet()) {
            if (playerMap.get(key) != 0) {
                answer = key;
                break;
            }
        }
        
        return answer;
    }
}

풀이포인트

HashMap.getOrDefault(Object key, V DefaultValue)

찾는 키가 존재한다면 찾는 키의 값을 반환하고 없다면 기본값을 반환한다.

  • key : Map에서 찾을 key값
  • defaultValue : 찾는 key가 없다면 반환할 값
  • return : 찾는 key가 존재한다면 key의 value를 반환, 없다면 기본값 반환

위 문제는 다음과 같은 순서로 접근이 가능하다.

  1. 전체 선수의 이름을 Map에 key로 셋팅
  2. 동명이인이 존재할 수 있기 때문에 HashMap.getOrDefault(Object key, V DefaultValue) 를 사용하여 Map에 같은 이름으로 셋팅된 선수가 있는지 탐색한 후 있다면 기존값에 + 1을 한 값을 value로 셋팅
  3. Map에서 완주한 선수를 찾아 value에서 -1
  4. value가 0이 아닌 선수가 완주하지 못한 선수

HashMap의 API만 알고 있다면 쉽게 풀 수 있는 문제이다.

profile
친절한 개발자

0개의 댓글