結果

問題 No.1912 Get together 2
ユーザー shotoyoo
提出日時 2021-09-21 00:19:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 747 ms / 2,000 ms
コード長 968 bytes
コンパイル時間 184 ms
コンパイル使用メモリ 82,364 KB
実行使用メモリ 107,308 KB
最終ジャッジ日時 2024-06-24 01:21:41
合計ジャッジ時間 14,513 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))


def zeta_sub(val, n):
    """
    given: そのbitの出席パターン回数
    compute: 各bitのsubsetについて、countの総和
    """
    # len(val)==2^n
    out = val[:]
    for i in range(n):
        for j in range(1<<n):
            if j>>i&1:
                out[j] += out[j^(1<<i)]
    return out

n,m = list(map(int, input().split()))
vs = list(map(int, input().split()))
vals = [0]*(1<<m)
for i in range(n):
    s = input()
    b = int(s.replace("o", "1").replace("x", "0")[::-1], 2)
    vals[b] += vs[i]
size = zeta_sub(vals, m)
dp = [0]*(1<<m)
for b in range(1<<m):
    val = dp[b]
    for i in range(m):
        if b>>i&1:
            continue
        nb = b|(1<<i)
        dp[nb] = max(dp[nb], val + (size[nb] - size[b])**2)
ans = dp[(1<<m)-1]
print(ans)
0