結果

問題 No.1912 Get together 2
コンテスト
ユーザー SSRS
提出日時 2022-04-22 22:02:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 231 ms / 2,000 ms
コード長 989 bytes
コンパイル時間 2,194 ms
コンパイル使用メモリ 199,764 KB
最終ジャッジ日時 2025-01-28 20:16:06
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
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<int> A(N, 0);
  for (int i = 0; i < N; i++){
    for (int j = 0; j < M; j++){
      if (S[i][j] == 'o'){
        A[i] |= 1 << j;
      }
    }
  }
  vector<long long> zeta(1 << M, 0);
  for (int i = 0; i < N; i++){
    zeta[A[i]] += V[i];
  }
  for (int i = 0; i < M; i++){
    for (int j = 0; j < (1 << M); j++){
      if ((j >> i & 1) == 1){
        zeta[j] += zeta[j - (1 << i)];
      }
    }
  }
  vector<long long> dp(1 << M, 0);
  for (int i = 0; i < (1 << M); i++){
    for (int j = 0; j < M; j++){
      if ((i >> j & 1) == 0){
        int i2 = i | (1 << j);
        int S = zeta[(1 << M) - 1 - i] - zeta[(1 << M) - 1 - i2];
        dp[i2] = max(dp[i2], dp[i] + (long long) S * S);
      }
    }
  }
  cout << dp[(1 << M) - 1] << endl;
}
0