結果

問題 No.2423 Merge Stones
ユーザー t98slidert98slider
提出日時 2023-08-14 21:26:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 832 ms / 4,000 ms
コード長 1,058 bytes
コンパイル時間 1,943 ms
コンパイル使用メモリ 81,732 KB
実行使用メモリ 82,544 KB
最終ジャッジ日時 2024-05-02 09:31:40
合計ジャッジ時間 43,454 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
52,352 KB
testcase_01 AC 53 ms
61,056 KB
testcase_02 AC 52 ms
59,648 KB
testcase_03 AC 51 ms
60,032 KB
testcase_04 AC 41 ms
52,224 KB
testcase_05 AC 53 ms
59,264 KB
testcase_06 AC 52 ms
60,416 KB
testcase_07 AC 49 ms
59,264 KB
testcase_08 AC 48 ms
58,848 KB
testcase_09 AC 43 ms
52,864 KB
testcase_10 AC 55 ms
60,436 KB
testcase_11 AC 574 ms
82,136 KB
testcase_12 AC 769 ms
72,960 KB
testcase_13 AC 456 ms
81,936 KB
testcase_14 AC 770 ms
72,928 KB
testcase_15 AC 790 ms
73,088 KB
testcase_16 AC 690 ms
82,132 KB
testcase_17 AC 727 ms
72,832 KB
testcase_18 AC 694 ms
82,288 KB
testcase_19 AC 752 ms
72,832 KB
testcase_20 AC 757 ms
73,088 KB
testcase_21 AC 702 ms
82,048 KB
testcase_22 AC 765 ms
73,088 KB
testcase_23 AC 758 ms
73,216 KB
testcase_24 AC 766 ms
72,832 KB
testcase_25 AC 764 ms
72,832 KB
testcase_26 AC 832 ms
73,088 KB
testcase_27 AC 779 ms
73,600 KB
testcase_28 AC 679 ms
82,208 KB
testcase_29 AC 753 ms
82,048 KB
testcase_30 AC 748 ms
72,832 KB
testcase_31 AC 743 ms
73,088 KB
testcase_32 AC 674 ms
81,748 KB
testcase_33 AC 746 ms
73,088 KB
testcase_34 AC 747 ms
73,344 KB
testcase_35 AC 770 ms
73,344 KB
testcase_36 AC 736 ms
72,704 KB
testcase_37 AC 734 ms
73,088 KB
testcase_38 AC 744 ms
72,832 KB
testcase_39 AC 715 ms
82,024 KB
testcase_40 AC 709 ms
82,176 KB
testcase_41 AC 749 ms
73,216 KB
testcase_42 AC 662 ms
82,028 KB
testcase_43 AC 750 ms
72,704 KB
testcase_44 AC 691 ms
82,176 KB
testcase_45 AC 764 ms
73,088 KB
testcase_46 AC 753 ms
73,308 KB
testcase_47 AC 745 ms
72,960 KB
testcase_48 AC 732 ms
72,832 KB
testcase_49 AC 751 ms
72,832 KB
testcase_50 AC 754 ms
72,576 KB
testcase_51 AC 473 ms
82,092 KB
testcase_52 AC 472 ms
82,040 KB
testcase_53 AC 478 ms
82,224 KB
testcase_54 AC 463 ms
82,048 KB
testcase_55 AC 683 ms
82,248 KB
testcase_56 AC 630 ms
82,176 KB
testcase_57 AC 582 ms
82,304 KB
testcase_58 AC 433 ms
81,920 KB
testcase_59 AC 578 ms
82,048 KB
testcase_60 AC 437 ms
82,176 KB
testcase_61 AC 463 ms
82,048 KB
testcase_62 AC 479 ms
82,544 KB
testcase_63 AC 472 ms
82,176 KB
testcase_64 AC 482 ms
82,048 KB
testcase_65 AC 478 ms
82,036 KB
testcase_66 AC 457 ms
82,120 KB
testcase_67 AC 456 ms
82,048 KB
testcase_68 AC 458 ms
82,048 KB
testcase_69 AC 462 ms
82,064 KB
testcase_70 AC 457 ms
82,132 KB
testcase_71 AC 447 ms
82,048 KB
testcase_72 AC 456 ms
82,048 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import stdin
import itertools
N, K = map(int, stdin.readline().split())
A = list(map(int, stdin.readline().split()))
C = list(map(int, stdin.readline().split()))
dp = [[0] * (2 * N + 1) for _ in range(2 * N)]
dp2 = [[0] * (2 * N + 1) for _ in range(2 * N)]
s = [0] + list(itertools.accumulate(A + A))
ans = max(A)

for i in range(N):
    C[i] -= 1
    dp[i][i + 1] |= 1 << C[i]
    dp2[i][i + 1] = dp[i][i + 1]
    for j in range(1, K + 1):
        dp2[i][i + 1] |= dp[i][i + 1] >> j
        dp2[i][i + 1] |= dp[i][i + 1] << j
    dp[i + N][i + 1 + N] = dp[i][i + 1]
    dp2[i + N][i + 1 + N] = dp2[i][i + 1]

for Len in range(2, N + 1):
    for l in range(2 * N - Len):
        r = l + Len
        for m in range(l, r):
            dp[l][r] |= dp[l][m] & dp2[m][r]
            dp[l][r] |= dp2[l][m] & dp[m][r]
        
        if dp[l][r]:
            ans = max(ans, s[r] - s[l])
            dp2[l][r] = dp[l][r]
            for j in range(1, K + 1):
                dp2[l][r] |= dp[l][r] >> j
                dp2[l][r] |= dp[l][r] << j

print(ans)
0