結果

問題 No.1912 Get together 2
ユーザー ぷら
提出日時 2022-04-22 21:37:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 815 ms / 2,000 ms
コード長 1,087 bytes
コンパイル時間 2,382 ms
コンパイル使用メモリ 196,888 KB
最終ジャッジ日時 2025-01-28 20:07:19
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

long long dp[1 << 21],sum[21][1 << 21];

int main() {
    int N,M;
    cin >> N >> M;
    vector<int>V(N);
    for(int i = 0; i < N; i++) {
        cin >> V[i];
    }
    vector<string>S(N);
    for(int i = 0; i < N; i++) {
        cin >> S[i];
        int id = 0;
        for(int j = 0; j < M; j++) {
            if(S[i][j] == 'o') continue;
            id |= 1 << j;
        }
        for(int j = 0; j < M; j++) {
            if(S[i][j] == 'x') continue;
            sum[j][id] += V[i];
        }
    }
    for(int k = 0; k < M; k++) {
        for(int i = 0; i < M; i++) {
            for(int j = (1 << M)-1; j >= 0; j--) {
                if(1 & (j >> i)) {
                    sum[k][j^(1 << i)] += sum[k][j];
                }
            }
        }
    }
    for(int i = 0; i < (1 << M); i++) {
        for(int j = 0; j < M; j++) {
            if(1 & (i >> j)) {
                continue;
            }
            dp[i|(1 << j)] = max(dp[i|(1 << j)],dp[i]+sum[j][i]*sum[j][i]);
        }
    }
    cout << dp[(1 << M)-1] << endl;
}
0