코드트리 [자바 java] HashSet 기본

https://www.codetree.ai/missions/8/problems/hashset-basic?&utm_source=clipboard&utm_medium=text 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

이 문제는 제목 그대로 HashSet 기본 함수를 이용해보는 문제이다.

 

https://ethereal-coder.tistory.com/153

[알고리즘 개념] Map, Set, Queue

이와 관련된 개념은 위에 정리해두었다.

 

import java.io.*;
import java.util.*;
import java.math.*;


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

        HashSet<Integer> m = new HashSet<>();

        StringTokenizer st;
        StringBuilder sb = new StringBuilder();

        for(int i = 0;i<n;i++){
            st = new StringTokenizer(br.readLine());
            String k = st.nextToken();
            
            if(k.equals("add")){
                int a = Integer.parseInt(st.nextToken());
                m.add(a);
            }

            else if(k.equals("find")){
                int a = Integer.parseInt(st.nextToken());
                if(m.contains(a)){
                    sb.append("true" + "\n");
                }
                else sb.append("false\n");
            }

            else if(k.equals("remove")){
                int a = Integer.parseInt(st.nextToken());
                m.remove(a);
            }
            
        }

        System.out.println(sb);


    }
}