結果

問題 No.1912 Get together 2
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-09-25 02:11:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,072 ms / 2,000 ms
コード長 1,257 bytes
コンパイル時間 345 ms
コンパイル使用メモリ 82,336 KB
実行使用メモリ 274,208 KB
最終ジャッジ日時 2024-09-25 02:12:11
合計ジャッジ時間 21,134 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
53,420 KB
testcase_01 AC 37 ms
53,916 KB
testcase_02 AC 38 ms
52,648 KB
testcase_03 AC 47 ms
61,664 KB
testcase_04 AC 38 ms
53,244 KB
testcase_05 AC 38 ms
52,692 KB
testcase_06 AC 39 ms
53,208 KB
testcase_07 AC 48 ms
62,264 KB
testcase_08 AC 113 ms
80,228 KB
testcase_09 AC 98 ms
77,716 KB
testcase_10 AC 117 ms
81,368 KB
testcase_11 AC 109 ms
81,488 KB
testcase_12 AC 129 ms
83,860 KB
testcase_13 AC 1,052 ms
273,888 KB
testcase_14 AC 1,059 ms
274,072 KB
testcase_15 AC 1,072 ms
274,108 KB
testcase_16 AC 1,017 ms
274,016 KB
testcase_17 AC 1,017 ms
274,152 KB
testcase_18 AC 1,033 ms
273,956 KB
testcase_19 AC 984 ms
274,140 KB
testcase_20 AC 995 ms
273,884 KB
testcase_21 AC 1,052 ms
274,152 KB
testcase_22 AC 995 ms
274,140 KB
testcase_23 AC 954 ms
273,864 KB
testcase_24 AC 956 ms
273,888 KB
testcase_25 AC 970 ms
273,812 KB
testcase_26 AC 948 ms
273,956 KB
testcase_27 AC 974 ms
274,060 KB
testcase_28 AC 162 ms
89,536 KB
testcase_29 AC 138 ms
89,900 KB
testcase_30 AC 150 ms
89,928 KB
testcase_31 AC 139 ms
89,560 KB
testcase_32 AC 134 ms
89,604 KB
testcase_33 AC 956 ms
274,208 KB
testcase_34 AC 38 ms
52,336 KB
testcase_35 AC 946 ms
273,808 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1912


def main():
    N, M = map(int, input().split())
    V = list(map(int, input().split()))
    S = []
    for _ in range(N):
        S.append(input())

    # 各Nについて使える箱をbitmask化する
    bit_sums = [0] * (2 ** M)
    for i in range(N):
        bit = 0
        s = S[i]
        for m in range(M):
            if s[m] == "o":
                bit |= (1 << m)
        bit_sums[bit] += V[i]

    # 高速ゼータ変換で任意の箱の集合Sの中に収まる範囲でしか入らないスライムVの総和を計算
    zz = [[0] * (2 ** M) for _ in range(M + 1)]
    for i in range(2 ** M):
        zz[0][i] = bit_sums[i]
    for j in range(M):
        for i in range(2 ** M):
            if i & (1 << j) > 0:
                zz[j + 1][i] = zz[j][i - (1 << j)] + zz[j][i]
            else:
                zz[j + 1][i] = zz[j][i]

    # dp[選ばれていない箱S]についてdpを解く
    dp = [0] * (2 ** M)
    for s in range(2 ** M):
        for i in range(M):
            if s & (1 << i) > 0:
                s_ = s - (1 << i)
                dp[s] = max(dp[s], dp[s_] + (zz[-1][s] - zz[-1][s_]) ** 2)
    print(dp[-1])



        














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