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

문제 설명

• 수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
• 마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한 사항

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

입출력 예

입출력 예 설명

예제 #1

• leo는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #2

• vinko는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #3

• mislav는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

---

문제 풀이

나의 풀이

import java.util.*;

class Solution {
public String solution(String[] participant, String[] completion) {
        Map<String, Integer> containList = new HashMap<String, Integer>();
        for (String s : participant) {
            if (containList.containsKey(s)) {
                containList.put(s, containList.get(s).intValue() + 1);
            } else {
                containList.put(s, 1);
            }
        }

        for (String s : completion) {
            int a = containList.get(s);
            if (a == 1) {
                containList.remove(s);
            } else {
                containList.put(s, containList.get(s) - 1);
            }
        }
        return new ArrayList<String>(containList.keySet()).get(0);
    }
}

 

보완한 나의 풀이

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

    for (String s : participant) {
      containList.compute(s, (key, value) -> value == null ? 1 : ++value);
    }

    for (String s : completion) {
      containList.compute(s, (key, value) -> value == 1 ? null : --value);
    }
    return containList.keySet().iterator().next();
}

• Map의 compute 메서드 사용법을 알게 되었다.
  • 써보니 정말 편하다. 진짜 정말 편하다.. 표현이 안되네

인상적인 답안

public static String solution(String[] participant, String[] completion) {
    HashMap<String, Integer> map = new HashMap<>();
    for (int i = 0; i < participant.length; i++) {
      map.compute(participant[i], (k, v) -> v != null ? null : 1);
      if (i < completion.length) {
        map.compute(completion[i], (k, v) -> v != null ? null : 1);
      }
    }
    return map.keySet().iterator().next();
 }

• 동명이인이 있어도 이름이 participant 와 completion 존재하는 수가 홀수일때 완주를 못한 선수로 간주를 했다.
• 그렇기 때문에 participant에 있든 completion에 있든 map에 이미 있는 이름이라면 없애는 식의 계산을 한다.
• 정말 심플하다.

---

참고

코딩테스트 연습 - 완주하지 못한 선수 | 프로그래머스

---

끝으로

이 글이 도움이 되었다면, 하단의 Google 광고 👎👎👎 한번씩 클릭 부탁 드립니다. 🙏🙏🙏

광고 클릭은 많은 힘이 됩니다!