結果

問題 No.1912 Get together 2
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-09-25 01:08:44
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,147 bytes
コンパイル時間 626 ms
コンパイル使用メモリ 82,260 KB
実行使用メモリ 90,052 KB
最終ジャッジ日時 2024-09-25 01:08:58
合計ジャッジ時間 13,029 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 WA * 30
権限があれば一括ダウンロードができます

ソースコード

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化する
    usable_box_set = []
    for i in range(N):
        bit = 0
        s = S[i]
        for m in range(M):
            if s[m] == "o":
                bit |= (1 << m)
        usable_box_set.append(bit)

    # 箱に入れた時の価値を最大にしてく
    used = 0    
    answer = 0
    for _ in range(M):
        values = []
        for i in range(M):
            if used & (1 << i) > 0:
                continue

            total_value = 0
            for j in range(N):
                bit = usable_box_set[j]
                v = V[j]
                if bit & (1 << i) > 0 and bit & used == 0:
                    total_value += v
            values.append((i, total_value))
        values.sort(key=lambda x: x[1],reverse=True)
        index, v = values[0]
        used |= (1 << index)
        answer += v ** 2
    print(answer)
        














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