結果

問題 No.1912 Get together 2
ユーザー shotoyooshotoyoo
提出日時 2021-09-25 19:44:46
言語 PyPy3
(7.3.15)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,079 bytes
コンパイル時間 188 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 89,740 KB
最終ジャッジ日時 2024-06-24 01:21:46
合計ジャッジ時間 4,594 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 RE * 17
権限があれば一括ダウンロードができます

ソースコード

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()))
assert 1<=n<=10**5 and 1<=m<=15
vs = list(map(int, input().split()))
assert len(vs)==n
vals = [0]*(1<<m)
for i in range(n):
    assert 1<=vs[i]<=10**4
    s = input()
    assert len(s)==m and "o" in s
    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