https://www.acmicpc.net/problem/1764
김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.
듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.
출력
듣보잡의 수와 그 명단을 사전순으로 출력한다.
예제 입력 1
3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton
예제 출력 1
2
baesangwook
ohhenrie
package silver4;
import java.util.*;
public class B1764 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
//숫자 입력
int noListen = sc.nextInt();
int noSeen = sc.nextInt();
//사람 이름에 해당하는 사람이 몇명인지 검사
//듣과 보에 중복은 없으므로 중복 우려는 없음
Set<String> s = new HashSet<>();
//사람 입력
for (int i = 0; i < noListen; i++) { //못들은사람 넣기
String name = sc.next();
s.add(name);
}
List<String> res = new ArrayList<String>();
for (int i = 0; i < noSeen; i++) {
String name = sc.next();
if(s.contains(name)) { //듣보 존재?
res.add(name);
}
}
Collections.sort(res);
System.out.println(res.size());
for (String name : res) {
System.out.println(name);
}
}
}
Set 객체 s를 만들어 못들은 사람을 입력받을 때마다 s에 넣고
보지 못한 사람을 입력받을 때마다 s에 존재하는지 검사한다(O(1)
)
그리고 존재하면 결과를 출력할 리스트인 res에 저장한다.
저장 후에는 res를 정렬하여 출력한다.
처음에는 Set을 쓰지 않고 전부 입력받은 후 하나하나 검색하려 해서 시간초과가 났다.