[Python] 3052번 나머지 - 백준

나머지

3052번: 나머지

1차원 배열

 

문제

두 자연수 A와 B가 있을 때, \( A \% B \)는 A를 B로 나눈 나머지이다. 예를 들어, 7, 14, 27, 38을 3으로 나눈 나머지는 1, 2, 0, 2이다.

수 10개를 입력 받은 뒤, 이를 42로 나눈 나머지를 구한다. 그 다음 서로 다른 값이 몇 개 있는지 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄부터 열 번째 줄까지 숫자가 한 줄에 하나씩 주어진다. 이 숫자는 1,000보다 작거나 같고, 음이 아닌 정수이다.

출력

첫째 줄에, 42로 나누었을 때, 서로 다른 나머지가 몇 개 있는지 출력한다.

예제 입력 1

1
2
3
4
5
6
7
8
9
10

예제 출력 1

10

각 수를 42로 나눈 나머지는 1, 2, 3, 4, 5, 6, 7, 8, 9, 10이다.

예제 입력 2

42
84
252
420
840
126
42
84
420
126

예제 출력 2

1

모든 수를 42로 나눈 나머지는 0이다.

예제 입력 3

39
40
41
42
43
44
82
83
84
85

예제 출력 3

6

각 수를 42로 나눈 나머지는 39, 40, 41, 0, 1, 2, 40, 41, 0, 1이다. 서로 다른 값은 6개가 있다.

---

 

다양한 Python 풀이

풀이 1

로그인

print(len(set((int(input()) % 42 for _ in range(10)))))

 

Generator expression을 set으로 변환해서 중복이 없는 unique한 원소만 남긴다. 이 집합의 길이를 구해서 출력해준다.

 

풀이 2

로그인

 

[풀이 2]는 set comprehension을 사용한 풀이다.

print(len({int(input()) % 42 for _ in range(10)}))

 

위 [풀이 1]에서는 generator를 set을 변환해주었지만 이번엔 처음부터 set에 값을 추가한다. 처음 방법은 한번의 탐색에서 값을 비교한다. 하지만 이 방법은 매번 기존에 들어간 원소와 새로운 값을 반복하며 비교하기에 이번 문제에서는 시간이 좀 더 걸린다. 비교 정렬하는 과정에서 시간이 걸린다고 생각하면 이해가 쉬울 것 같다.