結果

問題 No.1360 [Zelkova 4th Tune] 協和音
コンテスト
ユーザー monnu
提出日時 2021-01-23 01:13:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 24 ms / 2,000 ms
コード長 1,053 bytes
コンパイル時間 2,002 ms
コンパイル使用メモリ 173,612 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-12-29 08:00:36
合計ジャッジ時間 5,781 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MOD 998244353
#define INF 1000000000000000000
#define MAX 2000

int main(){
  int N;
  cin>>N;
  vector<ll> A(N);
  vector<vector<ll>> B(N,vector<ll>(N));
  for(int i=0;i<N;i++){
    cin>>A[i];
  }
  for(int i=0;i<N;i++){
    for(int j=0;j<N;j++){
      cin>>B[i][j];
    }
  }
  vector<ll> dp(1<<N,-INF);
  dp[0]=0;
  for(int i=1;i<(1<<N);i++){
    int bit=i;
    int k=0;
    while(bit!=1){
      bit>>=1;
      k++;
    }
    bit=i-(1<<k);
    dp[i]=dp[bit]+A[k];
    for(int j=0;j<k;j++){
      if((bit&1)==1){
        dp[i]+=B[j][k];
      }
      bit>>=1;
    }
  }

  ll score=0;
  int bit=0;
  for(int i=0;i<(1<<N);i++){
    if(dp[i]>=score){
      bit=i;
      score=dp[i];
    }
  }
  vector<int> ans;
  for(int i=0;i<N;i++){
    if(((bit>>i)&1)==1){
      ans.push_back(i+1);
    }
  }
  cout<<score<<endl;
  for(int i=0;i<ans.size();i++){
    cout<<ans[i]<<" ";
  }
  cout<<endl;
}
0