[백준 / 실버4] 1920 수 찾기 (Java)

문제 보기

사용한 것

• contains()를 O(1)로 사용하기 위한 HashSet

풀이 방법

• 입력 값 -> set
• 입력 값이 set에 존재하면 1, 안하면 0으로 변환 후 출력

코드

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        br.readLine(); // not use
        Set<Integer> set = Arrays.stream(br.readLine().split(" "))
            .map(Integer::parseInt)
            .collect(Collectors.toSet());
        br.readLine(); // not use
        Arrays.stream(br.readLine().split(" "))
            .mapToInt(Integer::parseInt)
            .map(i -> set.contains(i) ? 1 : 0)
            .forEach(System.out::println);
    }
}