結果

問題 No.1345 Beautiful BINGO
ユーザー shiomusubi496
提出日時 2021-01-16 13:57:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 2,271 bytes
コンパイル時間 2,025 ms
コンパイル使用メモリ 175,136 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-28 11:55:19
合計ジャッジ時間 5,731 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#define int long long
using namespace std;
int N,M,A[16][16],ans=1e18;
signed main(){
    cin>>N>>M;
    for(int i=0;i<N;i++)for(int j=0;j<N;j++)cin>>A[i][j];
    for(int i=0;i<(1<<N);i++){
        if(__builtin_popcount(i)+N<M)continue;
        int cnt=0,m=0;
        for(int j=0;j<N;j++)if((i>>j)&1){
            m++;
            for(int k=0;k<N;k++)cnt+=A[j][k];
        }
        vector<int>v(N);
        for(int j=0;j<N;j++)if(!((i>>j)&1)){
            for(int k=0;k<N;k++)v[k]+=A[j][k];
        }
        sort(v.begin(),v.end());
        for(int j=0;j<M-m;j++)cnt+=v[j];
        ans=min(ans,cnt);
    }
    for(int i=0;i<(1<<N);i++){
        if(__builtin_popcount(i)+N+1<M)continue;
        int cnt=0,m=1;
        for(int j=0;j<N;j++)cnt+=A[j][j];
        for(int j=0;j<N;j++)if((i>>j)&1){
            m++;
            for(int k=0;k<N;k++)if(j!=k)cnt+=A[j][k];
        }
        vector<int>v(N);
        for(int j=0;j<N;j++)if(!((i>>j)&1)){
            for(int k=0;k<N;k++)if(j!=k)v[k]+=A[j][k];
        }
        sort(v.begin(),v.end());
        for(int j=0;j<M-m;j++)cnt+=v[j];
        ans=min(ans,cnt);
    }
    for(int i=0;i<(1<<N);i++){
        if(__builtin_popcount(i)+N+1<M)continue;
        int cnt=0,m=1;
        for(int j=0;j<N;j++)cnt+=A[j][N-1-j];
        for(int j=0;j<N;j++)if((i>>j)&1){
            m++;
            for(int k=0;k<N;k++)if(j+k+1!=N)cnt+=A[j][k];
        }
        vector<int>v(N);
        for(int j=0;j<N;j++)if(!((i>>j)&1)){
            for(int k=0;k<N;k++)if(j+k+1!=N)v[k]+=A[j][k];
        }
        sort(v.begin(),v.end());
        for(int j=0;j<M-m;j++)cnt+=v[j];
        ans=min(ans,cnt);
    }
    for(int i=0;i<(1<<N);i++){
        if(__builtin_popcount(i)+N+2<M)continue;
        int cnt=0,m=2;
        for(int j=0;j<N;j++)cnt+=A[j][j];
        for(int j=0;j<N;j++)if(j!=N-1-j)cnt+=A[j][N-1-j];
        for(int j=0;j<N;j++)if((i>>j)&1){
            m++;
            for(int k=0;k<N;k++)if(j!=k&&j+k+1!=N)cnt+=A[j][k];
        }
        vector<int>v(N);
        for(int j=0;j<N;j++)if(!((i>>j)&1)){
            for(int k=0;k<N;k++)if(j!=k&&j+k+1!=N)v[k]+=A[j][k];
        }
        sort(v.begin(),v.end());
        for(int j=0;j<M-m;j++)cnt+=v[j];
        ans=min(ans,cnt);
    }
    cout<<ans<<endl;
}
0