15. [알고리즘] 정다면체

내 생각

defaultdict를 사용하는게 핵심. 해시를 생각했다.

구현

import collections

n,m = map(int,input().split())
di = collections.defaultdict(int)

for i in range(1,n+1):
    for j in range(1,m+1):
        di[i+j] += 1

max_value = max(di.values())
res = []
for d in di.items():
    if d[1] == max_value:
        res.append(d[0])
res = list(map(str,res))
print(" ".join(res))

다른사람 생각

리스트를 사용했다. 배열 크기를 잘 정하는게 핵심인듯

구현

n,m = map(int,input().split())
cnt = [0] * (n+m+3)
max_value = float('-inf')
for i in range(1,n+1):
    for j in range(1,m+1):
        cnt[i+j] += 1
for i in range(n+m+1):
    if max_value < cnt[i]:
        max_value = cnt[i]
for i in range(n+m+1):
    if cnt[i] == max_value:
        print(i,end=" ")