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

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

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

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

 

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());

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

        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);
            }
            
            else if(k.equals("upper_bound")){
                
                int a = Integer.parseInt(st.nextToken());
                
                if(m.higher(a) == null){
                    sb.append("None\n");
                }
                else sb.append(m.higher(a) + "\n");
            }

            else if(k.equals("lower_bound")){
                 int a = Integer.parseInt(st.nextToken());
               
                if(m.ceiling(a) == null){
                    sb.append("None\n");
                }
                else sb.append(m.ceiling(a) + "\n");
            }
            else if(k.equals("largest")){
                 
                  if(m.isEmpty()){
                    sb.append("None\n");
                   }
                
                  else sb.append(m.last() + "\n");
            }
            else if(k.equals("smallest")){
               
                 if(m.isEmpty()){
                    sb.append("None\n");
                    continue;
                }
                
                else sb.append(m.first() + "\n");
            }
        }

        System.out.println(sb);


    }
}