728x90
문제 설명
수많은 마라톤 선수들이 마라톤에 참여하였습니다.
단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.
제한사항
- 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
- completion의 길이는 participant의 길이보다 1 작습니다.
- 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
- 참가자 중에는 동명이인이 있을 수 있습니다.
입출력 예
participant | completion | return |
---|---|---|
["leo", "kiki", "eden"] | ["eden", "kiki"] | "leo" |
["marina", "josipa", "nikola", "vinko", "filipa"] | ["josipa", "filipa", "marina", "nikola"] | "vinko" |
["mislav", "stanko", "mislav", "ana"] | ["stanko", "ana", "mislav"] | "mislav" |
내 풀이
public static String solution(String[] participant, String[] completion) {
String answer = "";
HashMap<String, Integer> partList = new HashMap();
int cnt;
for(String part : participant){
cnt = 1;
if(partList.containsKey(part)){
cnt += partList.get(part);
}
partList.put(part, cnt);
}
for(String comp : completion){
cnt = partList.get(comp) - 1;
if(cnt <= 0){
partList.remove(comp);
}else{
partList.put(comp, cnt);
}
}
return partList.keySet().iterator().next();
}
아쉬운 점
- participant 데이터 HashMap 데이터 변경
- 선수들 데이터를 넣기 위해 HashMap으로 변경 시 선수들 인원을 세팅하기 위해 HashMap을 사용하였지만 for문 안에 containKey를 사용하여 한 번의 조건을 주었다 하지만 HashMap의 "getOrDefault"을 사용하면 데이터가 없을때의 초기 값을 설정할 수 있어 코드가 더욱 간결 해 질 수 있다..
- partList.put(part, partList.getOrDefault(part, 0) + 1);
- 람다 사용
- for문을 사용하기 보다 java 람다를 사용하여 코드를 간결하게 만들 수 있다.
개선해야 할 점
HashMap의 getOrDefault()와 같이 기존의 지원하는 메소드를 좀 더 사용하여 필요시 언제든지 사용할 수 있게 해야 한다.
Git Hub
728x90
'Coding Test' 카테고리의 다른 글
기능 개발 (0) | 2023.04.15 |
---|---|
베스트앨범 (0) | 2023.04.15 |
위장 (0) | 2023.04.15 |
전화 번호부 접두사 (0) | 2023.04.15 |