본문 바로가기

파이썬 알고리즘 (인프런)/코드 구현력 기르기

5. 정다면체

# 정다면체

def run(N, M):
    a = [0] * (N + M + 1)
    for i in range(1, N + 1):
        for j in range(1, M + 1):
            a[i+j] += 1

    max_cnt, res = -1e9, []
    for i, cnt in enumerate(a):
        if cnt > max_cnt:  # 최대일때 숫자들
            max_cnt = cnt
            res = [i]
        elif cnt == max_cnt:
            res.append(i)

    return res


N, M = map(int, input().split())
ans = run(N, M)
[print(x, end=" ") for x in ans]

'파이썬 알고리즘 (인프런) > 코드 구현력 기르기' 카테고리의 다른 글

7. 소수 (에라토스테네스 체)  (0) 2021.01.06
6. 자릿수의 합  (0) 2021.01.06
4. 대표값  (0) 2021.01.06
3. K번째 큰 수  (0) 2021.01.06
2. K번째 수  (0) 2021.01.06