한 걸음 두 걸음
백준 11723 집합 ] HashSet Collection 사용 본문
반응형
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.*;
public class Main {
public static void main(String[] args) throws NumberFormatException, IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringBuilder sb = new StringBuilder("");
int testCase = Integer.parseInt(br.readLine());
Set<String> hs1 = new HashSet<>();
for(int i = 0 ; i < testCase; i++) {
StringTokenizer st = new StringTokenizer(br.readLine());
switch(st.nextToken()) {
case "add":
hs1.add(st.nextToken());
break;
case "remove":
hs1.remove(st.nextToken());
break;
case "check":
if( hs1.contains(st.nextToken())) sb.append("1 ");
else sb.append("0 ");
break;
case "toggle":
String toggleStr = st.nextToken();
if(hs1.contains(toggleStr)) hs1.remove(toggleStr);
else hs1.add(toggleStr);
break;
case "all":
hs1.clear();
for(int j = 1 ; j <= 20; j++)
hs1.add(j+"");
break;
case "empty":
hs1.clear();
break;
}
}
System.out.println(sb.toString());
}
}
참고 > 시간초과 났던 코드
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int testCase = sc.nextInt();
sc.nextLine();
HashSet<String> hs1 = new HashSet<>();
HashSet<String> allSet = new HashSet<>();
for(int i = 1 ; i <= 20; i++)
allSet.add(i+"");
for(int i = 0 ; i < testCase; i++) {
String str = sc.next();
switch(str) {
case "add":
hs1.add(sc.next());
break;
case "remove":
hs1.remove(sc.next());
break;
case "check":
if( hs1.contains(sc.next())) System.out.println("1");
else System.out.println("0");
break;
case "toggle":
String toggleStr = sc.next();
if(hs1.contains(toggleStr)) hs1.remove(toggleStr);
else hs1.add(toggleStr);
break;
case "all":
hs1.clear();
hs1 = (HashSet)allSet.clone();
break;
case "empty":
hs1.clear();
break;
}
}
}
}
반응형
'CSE > baekjoon & swexpert' 카테고리의 다른 글
백준 1676 팩토리얼 0의 개수 / java (0) | 2019.04.27 |
---|---|
Jaccard Index 구하는 JAVA 소스코드 (0) | 2019.04.16 |
백준 아호코라식 문제 다음에 풀어보자!! 개념공부가 좀 필요해요~ (0) | 2019.04.13 |
백준 14425 문자열집합 ] 자료구조 HashSet 이용! (Set) (0) | 2019.04.13 |
백준 1931 회의실배정 (0) | 2019.04.08 |