結果

問題 No.115 遠足のおやつ
ユーザー しらっ亭
提出日時 2015-08-22 15:05:12
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 396 ms / 5,000 ms
コード長 1,050 bytes
コンパイル時間 601 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 51,840 KB
最終ジャッジ日時 2025-01-03 01:06:44
合計ジャッジ時間 5,350 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

    dp = [[[False for d in range(D+1)] for k in range(K + 2)] for i in range(N + 1)]
    # dp[i][k][d] = i 番目までで k 個買って d 円使う組み合わせを作れるか
    src = [[[None for d in range(D+1)] for k in range(K + 2)] for i in range(N + 1)]

    dp[0][0][0] = True

    for i in range(N):
        for k in range(K, -1, -1):
            for d in range(D + 1):
                if dp[i][k][d]:
                    dp[i + 1][k][d] = True
                    src[i + 1][k][d] = (i, k, d)
                    if d + i + 1 <= D:
                        dp[i + 1][k + 1][d + i + 1] = True
                        src[i + 1][k + 1][d + i + 1] = (i, k, d)

    if dp[N][K][D]:
        r = []
        i, k, d = N, K, D
        while k:
            pi, pk, pd = src[i][k][d]
            if pd != d:
                r.append(i)
            i, k, d = pi, pk, pd
        print(' '.join(str(i) for i in r[::-1]))
    else:
        print(-1)


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