結果

問題 No.115 遠足のおやつ
ユーザー tcltk
提出日時 2021-01-16 17:19:12
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,505 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 233,216 KB
最終ジャッジ日時 2024-11-27 21:26:05
合計ジャッジ時間 68,264 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#region Header
#!/usr/bin/env python3
# from typing import *

import sys
import io
import math
import collections
import decimal
import itertools
from queue import PriorityQueue
import bisect
import heapq

def input():
    return sys.stdin.readline()[:-1]

sys.setrecursionlimit(1000000)
#endregion

# _INPUT = """10 9 3
# """
# sys.stdin = io.StringIO(_INPUT)


def main():
    N, D, K = map(int, input().split())

    dp = [[[False for _ in range(K+1)] for _ in range(D+1)] for _ in range(N+1)]
    dp[0][0][0] = True
    for n in range(0, N):
        for d in range(0, D+1):
            for k in range(0, K+1):
                if d + (n+1) <= D and k + 1 <= K:
                    dp[n+1][d+(n+1)][k+1] = dp[n][d+(n+1)][k+1] or dp[n][d][k]
                dp[n+1][d][k] = dp[n+1][d][k] or dp[n][d][k]

    result = []
    if dp[N][D][K]:
        q = collections.deque()
        q.appendleft((N, D, K, []))
        while q:
            p = q.pop()
            n = p[0]
            d = p[1]
            k = p[2]
            L = p[3]
            if n == 0 and d == 0 and k == 0:
                result.append(L)
            elif n > 0:
                if dp[n-1][d][k]:
                    q.appendleft((n-1, d, k, L))
                if d-n >= 0 and dp[n-1][d-n][k-1]:
                    q.appendleft((n-1, d-n, k-1, [n]+L))
        result.sort()
        print(*result[0])
    else:
        print(-1)


if __name__ == '__main__':
    main()
0