해시는 Key-Value 쌍으로 데이터를 저장하는 자료구조입니다.
문제 설명
수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 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 |
입출력 예 설명
- 예제 #1
leo는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다. - 예제 #2
vinko는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다. - 예제 #3
mislav는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.
풀이 (Java)
풀이는 HashMap을 이용하거나 Arrays.sort()를 이용하는 방법이 있다.
아래는 Arrays.sort()를 이용한 것이다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
import java.util.*;
class Solution {
public String solution(String[] participant, String[] completion) {
String answer = "";
Arrays.sort(participant);
Arrays.sort(completion);
int i =0; //a
for(i =0; i<completion.length;i++){
if(!participant[i].equals(completion[i])){
return participant[i];
}
}
return participant[i];
}
}
|
cs |
참가자와 완주자 목록을 순차 정렬한 다음, 순서대로 비교하여 그 이름이 다를 때 그 참가자를 리턴한다. 그리고 완주자의 마지막까지 이름이 같았다면, 참가자의 마지막 이름이 완주하지 못한 사람이므로, 마지막 참가자를 리턴한다.
이것은 딱 한 명만 완주하지 못했기 때문에 쓸 수 있는 방법이다.
그리고 for문의 i를 바깥으로 빼내어 마지막 참가자를 리턴하게 한 것이 놀라웠다.
-----
아래는 HashMap을 이용한 것이다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
import java.util.*;
class Solution {
public String solution(String[] participant, String[] completion) {
Map<String, Integer> hash = new HashMap<>();
for (String arg : participant) hash.put(arg, hash.getOrDefault(arg, 0) + 1);
for (String arg : completion) hash.put(arg, hash.get(arg) - 1);
for (String key : hash.keySet()) {
if (hash.get(key) != 0) return key;
}
return null;
}
}
|
cs |
Key와 Value가 String, Integer 형인 HashMap hash를 만들었다.
첫번째 반복문 - 참가자를 차례로 꺼내 hash에 Key로 참가자를, Value로 (참가자(Key)에 대한 Value, 없으면 0)+1)를 넣는다.
두번째 반복문 - 완주자를 차례로 꺼내 hash에 Key로 완주자를, Value로 (hash의 완주자에 대한 Value - 1)를 넣는다.
세번째 반복문 - hash의 key를 차례로 꺼내 hash의 Key에 대한 Value가 0이 아니면 그 Key를 리턴한다.
즉, hashMap에 참가자를 Key로, 0+1을 Value로 넣는데,
두번째 반복문에서 완주자에 대해 Value로 -1를 한다. 그렇다면 참가자 목록에 있고 완주자 목록에도 있던 사람들의 Value는 0이 되고, 참가했지만 완주하지 않은 사람의 Value는 -1 그대로이다.
이렇게 마지막에 Value가 0이 아닌 사람들의 Key 값을 리턴하면 된다.
풀이 출처 https://sas-study.tistory.com/2
참고자료
- 생활코딩 바로실습 Java Collections Framework
- 맵은 맵인데, '해시'맵은 뭐지? '해시'는 뭐지??
* getOrDefault
찾는 키가 존재한다면 찾는 키의 값을 반환하고 없다면 기본 값을 반환한다.
'프로그래밍-학습기록 > 코딩테스트' 카테고리의 다른 글
백준 온라인 저지 | 10952 | while문: A+B - 5 (0) | 2020.07.15 |
---|---|
백준 온라인 저지 | 2439 | 별 찍기 - 2 (0) | 2020.07.14 |
백준 온라인 저지 | 15552 | 빠른 A+B (0) | 2020.07.10 |
프로그래머스 | 코딩테스트 연습 | 탐욕법 | 체육복 (0) | 2020.07.08 |
프로그래머스 | 코딩테스트 연습 | 완전탐색 | 모의고사 (0) | 2020.07.07 |