[파이썬 | BOJ | 11652] 카드

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

11652번: 카드

문제

준규는 숫자 카드 N장을 가지고 있다. 숫자 카드에는 정수가 하나 적혀있는데, 적혀있는 수는 -2^62보다 크거나 같고, 2^62보다 작거나 같다.

준규가 가지고 있는 카드가 주어졌을 때, 가장 많이 가지고 있는 정수를 구하는 프로그램을 작성하시오. 만약, 가장 많이 가지고 있는 정수가 여러 가지라면, 작은 것을 출력한다.

입력

첫째 줄에 준규가 가지고 있는 숫자 카드의 개수 N (1 <= N <= 1000000)이 주어진다. 둘째 줄부터 N개 줄에는 숫자 카드에 적혀있는 정수가 주어진다.

출력

첫째 줄에 준규가 가장 많이 가지고 있는 정수를 출력한다.

풀이

알고리즘 분류 : 수학, sorting

N개 list를 모두 탐색하면서, d[arr[n]]값을 하나씩 증가시키는 방법을 쓸 수 없다. 2^62 * 2 개의 배열을 만들 수 없기 때문이다.

따라서 N개 list를 정렬하고, 같은 값의 개수만큼 인덱스를 증가시키면서, 제일 개수가 많은 최소값을 찾는다.

 

import sys
input = lambda : sys.stdin.readline().rstrip()

N = int(input())
arr = list()
for i in range(N):
    arr.append(int(input()))
arr.append(arr[N-1]//2+10)
arr.sort()
ans = [[1 , arr[0]]]
idx = 0

while idx < N:
    ans.append([arr.count(arr[idx]), arr[idx]])
    idx += arr.count(arr[idx])
print(max(ans, key=lambda item: (item[0], -item[1]))[1])

-> 시간초과

 

이 문제를 해결하기 위해서는 O(n) 시간복잡도로 해결해야하는데,

while문은 O(n)이지만, count를 사용하면서 O(n^2)이 되었다.

 

따라서 그냥 모든 n을 탐색하면서 바로 인접한 list의 값이 달라질때를 찾는것으로 알고리즘을 바꾼다.

 

N = int(input())
arr = list()

for i in range(N):
    arr.append(int(input()))
arr.append(arr[N-1]//2+10)
arr.sort()
ans = [[1 , arr[0]]]
idx = 0

for i in range(N):
    if(arr[i] != arr[i+1]):
        ans.append([i-idx+1, arr[i]])
        idx = i + 1
print(max(ans, key=lambda item: (item[0], -item[1]))[1])