[이것이 코딩 테스트다 with Python] 부품 찾기(Java)

Goal

“이것이 코딩 테스트다 with Python” 교재의 문제를 분석하고 코드와 함께 이해해보기 위한 글입니다.

문제 분석

자바에 내장된 이진 탐색 트리인 TreeSet을 이용해 구현하면 됩니다.

코드 구현

public class FindComponent {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        TreeSet<Integer> treeSet = new TreeSet<>();

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            treeSet.add(Integer.valueOf(st.nextToken()));
        }

        int m = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        for (int i = 0; i < m; i++) {
            if (treeSet.contains(Integer.parseInt(st.nextToken()))) {
                bw.write("yes ");
            } else bw.write("no ");
        }
        bw.close();
        br.close();
    }
}

​