結果

問題 No.1912 Get together 2
ユーザー hirayuu_yc
提出日時 2025-01-11 09:06:00
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 902 ms / 2,000 ms
コード長 1,395 bytes
コンパイル時間 1,966 ms
コンパイル使用メモリ 119,568 KB
実行使用メモリ 224,512 KB
最終ジャッジ日時 2025-01-11 09:06:26
合計ジャッジ時間 18,335 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <cmath>
using namespace std;

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];
    }

    vector<vector<long long>> zeta(1 << M, vector<long long>(M, 0));

    for (int i = 0; i < N; i++) {
        int num = 0;
        for (int j = 0; j < M; j++) {
            if (S[i][j] == 'o') {
                num += (1 << j);
            }
        }
        for (int j = 0; j < M; j++) {
            if (S[i][j] == 'o') {
                zeta[num][j] += V[i];
            }
        }
    }

    for (int i = 0; i < M; i++) {
        for (int j = 0; j < (1 << M); j++) {
            if ((j & (1 << i)) == 0) {
                for (int k = 0; k < M; k++) {
                    zeta[j ^ (1 << i)][k] += zeta[j][k];
                }
            }
        }
    }

    vector<long long> dp(1 << M, -(1LL << 60));
    dp[0] = 0;

    for (int i = 0; i < (1 << M); i++) {
        int pos = ((1 << M) - 1) ^ i;
        for (int j = 0; j < M; j++) {
            if ((pos >> j) & 1) {
                dp[i | (1 << j)] = max(dp[i | (1 << j)], dp[i] + zeta[pos][j] * zeta[pos][j]);
            }
        }
    }

    cout << dp[(1 << M) - 1] << endl;

    return 0;
}
0