結果

問題 No.3014 岩井満足性問題
コンテスト
ユーザー ytsutsu
提出日時 2025-02-02 15:51:33
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,938 bytes
コンパイル時間 649 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 280,688 KB
最終ジャッジ日時 2025-02-02 15:51:42
合計ジャッジ時間 7,078 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 MLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
from ast import Call, parse, unparse, walk
from inspect import currentframe, getsourcelines
from sys import maxsize, stdin

_tokens = (y for x in stdin for y in x.split())
def read(): return next(_tokens)
def iread(): return int(next(_tokens))


def dprint(*args, pretty=True):
    def _inner(v):
        def _format_3d(v): return '\n' + '\n'.join(['\n'.join([' '.join([str(z) for z in y]) for y in x]) + '\n' for x in v]).rstrip('\n')
        def _format_2d(v): return '\n' + '\n'.join([' '.join([str(y) for y in x]) for x in v])
        def _dim(v): return (1 + min(_dim(x) for x in v) if v else 1) if isinstance(v, (list, tuple)) else 1 if isinstance(v, str) and len(v) > 1 else 0
        dim = _dim(v) if pretty else -1
        return _format_3d(v) if dim == 3 else _format_2d(v) if dim == 2 else str(v)
    frame = currentframe().f_back
    source_lines, start_line = getsourcelines(frame)
    tree = parse(source_lines[frame.f_lineno - max(1, start_line)].strip())
    call_node = next(node for node in walk(tree) if isinstance(node, Call) and node.func.id == 'dprint')
    arg_names = [unparse(arg) for arg in call_node.args]
    print(', '.join([f'\033[4;35m{name}:\033[0m {_inner(value)}' for name, value in zip(arg_names, args)]))


def main():
    n, d, k = iread(), iread(), iread()
    A = [iread() for _ in range(n)]
    C = [iread() for _ in range(n)]
    INF = maxsize
    dp = [[-INF for _ in range(k + 1)] for _ in range(d + 1)]
    dp[0][0] = 0
    for a, c in zip(A, C):
        p = [x[:] for x in dp]
        dp, p = p, dp
        for i in range(d):
            ni = i + 1
            for j in range(k + 1):
                if p[i][j] == -INF:
                    continue
                nj = min(k, j + c)
                dp[ni][nj] = max(dp[ni][nj], p[i][j] + a)
    ans = dp[d][k]
    if ans == -INF:
        print('No')
    else:
        print(ans)


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