結果
問題 |
No.115 遠足のおやつ
|
ユーザー |
|
提出日時 | 2021-01-16 17:17:45 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,505 bytes |
コンパイル時間 | 257 ms |
コンパイル使用メモリ | 82,376 KB |
実行使用メモリ | 886,912 KB |
最終ジャッジ日時 | 2024-11-27 21:23:03 |
合計ジャッジ時間 | 69,401 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 TLE * 5 MLE * 5 |
ソースコード
#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()