[백준] 11723 집합

https://www.acmicpc.net/problem/11723

11723번: 집합

 

add, remove 등 메소드를 직접 구현해도 되지만, 자료구조를 이용해서 푸는 게 훨씬 좋아보인다. 메소드별로 조건을 자세히보면 집합 S 내 특정원소의 존재여부만 체크하면 되는 조건들로 이루어져 있으므로 선형구조보다는 Set이나 Map 등을 이용해서 해결하는 것이 효율적이라고 생각했고, 나는 HashSet을 이용해서 간단하게 문제를 해결할 수 있었다.

---

문제 및 입출력

---

코드

import java.util.HashSet;
import java.util.Scanner;

public class Main
{
    public static void main(String[] args)
    {
        Scanner sc = new Scanner(System.in);
        HashSet<String> set = new HashSet<String>();
        StringBuilder sb = new StringBuilder();
        int m = sc.nextInt();
        String temp = sc.nextLine();
        
        for(int i=0; i<m; i++)
        {
            String[] input = sc.nextLine().split(" ");
            
            if(input[0].equals("add"))
            {
                set.add(input[1] + "");
            }
            else if(input[0].equals("remove"))
            {
                if(set.contains(input[1]))
                {
                    set.remove(input[1]);
                }
            }
            else if(input[0].equals("check"))
            {
                if(set.contains(input[1]))
                {
                    sb.append("1" + "\n");
                }
                else
                {
                    sb.append("0" + "\n");
                }
            }
            else if(input[0].equals("toggle"))
            {
                if(set.contains(input[1]))
                {
                    set.remove(input[1]);
                }
                else
                {
                    set.add(input[1]);
                }
            }
            else if(input[0].equals("all"))
            {
                set.clear();
                
                for(int j=1; j<=20; j++)
                {
                    set.add(j + "");
                }
            }
            else if(input[0].equals("empty"))
            {
                set.clear();
            }
        }
        
        System.out.println(sb.toString());
        sc.close();
    }
}