https://www.acmicpc.net/problem/2910
문제 위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.
창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.
만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.
이렇게 정렬하는 방법을 빈도 정렬이라고 한다.
수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.
맨처음에는 조건을 제대로 보지 않고 카운팅스타로 문제를 해결할려고 했습니다. 빈도, 어느게 먼저 왔는지, 출력할 숫자 이렇게 3가지 정보를 가지고 정렬을 했습니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
n,m = map(int,input().split())
num_list = list(map(int,input().split()))
count = [0]*(m+1)
where = [0]*(m+1)
for i in range(n):
count[num_list[i]]+=1
if where[num_list[i]] == 0:
where[num_list[i]] = i+1
answer = []
for i in range(0,m+1):
if count[i]!=0:
answer.append([count[i],where[i],i])
answer.sort(key=lambda x:(-x[0],x[1]))
for i in range(len(answer)):
print((str(answer[i][2])+" ")*answer[i][0], end='')
문제는 시간초과가 아니라 메모리 초과가 나왔습니다.
입력 첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)
원소의 최대값이 100억인데 그게 3개를 가지고 문제를 풀려고하니 메모리에서 바로 문제가 나왔습니다. 이래서는 문제를 해결할 수 없습니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
n,m = map(int,input().split())
num_list = list(map(int,input().split()))
number = []
where = []
count = []
for i in range(n):
if num_list[i] not in number:
number.append(num_list[i])
count.append(1)
where.append(i)
else:
temp = number.index(num_list[i])
count[temp]+=1
answer = []
for i in range(len(number)):
answer.append([count[i],where[i],number[i]])
answer.sort(key=lambda x:(-x[0],x[1]))
for i in range(len(answer)):
print((str(answer[i][2])+" ")*answer[i][0], end='')
number where, count를 따로 관리해서 마지막에 집어넣고 출력했습니다.
Comments powered by Disqus.