https://www.acmicpc.net/problem/11723
11723번: 집합
첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다. 둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.
www.acmicpc.net
문제
비어있는 공집합 S가 주어졌을 때, 아래 연산을 수행하는 프로그램을 작성하시오.
- add x: S에 x를 추가한다. (1 ≤ x ≤ 20) S에 x가 이미 있는 경우에는 연산을 무시한다.
- remove x: S에서 x를 제거한다. (1 ≤ x ≤ 20) S에 x가 없는 경우에는 연산을 무시한다.
- check x: S에 x가 있으면 1을, 없으면 0을 출력한다. (1 ≤ x ≤ 20)
- toggle x: S에 x가 있으면 x를 제거하고, 없으면 x를 추가한다. (1 ≤ x ≤ 20)
- all: S를 {1, 2, ..., 20} 으로 바꾼다.
- empty: S를 공집합으로 바꾼다.
입력
첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.
둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.
출력
check 연산이 주어질때마다, 결과를 출력한다.
소스코드
풀이
★ 문제 그대로 파이썬의 집합 자료형을 활용하여 풀어야 하는 문제입니다. 처음에 집합을 사용하지 않고 풀었을 때, 시간초과 및 메모리 초과가 발생하였습니다.
★ 집합 자료형은 중복을 허용하지 않기 때문에 해당 수가 집합 안에 들어있는 지 아닌 지를 굳이 확인해주지 않아도 됩니다
★ sys.stdin.readline을 사용하여 문자열을 입력 받을 때는 줄바꿈 문자인 \n도 함께 출력되기 때문에 strip() 함수로 제거해주셔야 합니다
★ 입력받은 문자열의 길이를 기준으로 길이가 1일 때는 all,empty 2일 때는 나머지 경우로 나눠줍니다.
★ 리스트와 다르게 집합에 원소를 추가할 때는 'add'를 사용해줍니다. 제거하는 경우에는 'remove'와 'discard'를 사용할 수 있는데, remove 함수는 집합 안에 원소가 없을 경우 'Key Error'가 발생하지만 discard의 경우에는 에러 없이 동작하기에 discard를 사용해주었습니다.
★ remove, discard 참고자료
파이썬(Python) 집합(set) 타입, discard vs. remove 비교
파이썬의 데이터 타입 중에 집합(set) 타입이 있다. 집합 타입의 메소드 중에 특정 엘리먼트를 삭제하는 메소드가 두 개 있는데, 'discard()'와 'remove()'가 그것들이다. 이 두 메소드 모두 엘리먼트를
dev.plusblog.co.kr
'Algorithm > 백준' 카테고리의 다른 글
| [백준 알고리즘] 2108번: 통계학 (Python) (0) | 2023.04.26 |
|---|---|
| [백준 알고리즘] 15596번: 정수 N개의 합 (Python) (0) | 2023.04.22 |
| [백준 알고리즘] 2738번: 행렬 덧셈 (Python) (0) | 2023.04.21 |
| [백준 알고리즘] 7568번: 덩치 (Python) (1) | 2023.04.20 |
| [백준 알고리즘] 4796번: 캠핑 (Python) (0) | 2023.04.19 |