結果

問題 No.1912 Get together 2
コンテスト
ユーザー chineristAC
提出日時 2021-10-06 18:33:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 928 ms / 2,000 ms
コード長 956 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 178 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 109,072 KB
最終ジャッジ日時 2024-06-24 01:22:08
合計ジャッジ時間 18,174 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def zeta_to_small(N,A):
    res = [a for a in A]
    for i in range(N):
        t = 2**i
        for j in range(1<<N):
            if j&t:
                res[j-t] += res[j]
    return res

def solve(N,M,V,S):
    box = [0 for i in range(1<<M)]

    for i in range(N):
        t = 0
        for j in range(M):
            if S[i][j]=="x":
                t += 1<<j
        box[t] += V[i]

    zeta_box = zeta_to_small(M,box)

    dp = [0 for S in range(2**M)]
    ALL = 2**M-1
    for S in range(1,2**M):
        done = ALL - S
        for j in range(M):
            if S>>j & 1:
                tmp = zeta_box[done] - zeta_box[done + 2**j]
                dp[S] = max(dp[S],dp[S-2**j]+tmp*tmp)

    return dp[-1]

import sys

input = lambda :sys.stdin.readline().rstrip()
mi = lambda :map(int,input().split())
li = lambda :list(mi())

N,M = map(int,input().split())
V = list(map(int,input().split()))
S = [input() for i in range(N)]
print(solve(N,M,V,S))
0