矩阵消除游戏(状压+dfs）

矩阵消除游戏

•  比赛主页
•  我的提交

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述 

牛妹在玩一个名为矩阵消除的游戏，矩阵的大小是nn行mm列，第ii行第jj列的单元格的权值为ai,jai,j​，牛妹可以进行kk个回合的游戏，在每个回合，牛妹可以选择一行或者选择一列，然后将这一行或者这一列的所有单元格中的权值变为00，同时牛妹的分数会加上这一行或者这一列中的所有单元格的权值的和。

牛妹想最大化她的得分，球球你帮帮她吧！

输入描述:

第一行三个整数n,m,kn,m,k
接下来nn行每行mm个整数表示矩阵中各个单元格的权值。

输出描述:

输出一个整数表示牛妹能获得的最大分数。

示例1

输入

复制

3 3 2
101 1 102
1 202 1
100 8 100

输出

复制

414

备注:

1≤n,m≤151≤n,m≤15

1≤ai,j≤1e61≤ai,j​≤1e6

1≤k≤n∗m1≤k≤n∗m

int n,m,k;
int a[20][20];
int l[20];
int ans=0;
int get_cnt(int x){//状压储存选中的行
    int res=0;
    while (x) {
        res+=x&1;
        x>>=1;
    }
    return res;
}
void solve(int id,int cnt,int x){ //id是正在处理的行 cnt是已经选中的行 x是需要的行
    int now=get_cnt(cnt);
    int t=cnt;
    if(n-id+1<x-now) return;
    if(now==x){
        int res=0;
        for(int i=1;i<=n;i++,t>>=1){
            if(t&1){
                for(int j=1;j<=m;j++){
                    res+=a[i][j];
                }
            }
        }
        for(int j=1;j<=m;j++){
            t=cnt;
            l[j]=0;
            for(int i=1;i<=n;i++,t>>=1){
                if(!(t&1)){
                    l[j]+=a[i][j];
                }
            }
        }
        sort(l+1,l+1+m,greater<int>());
        for(int i=1;i<=k-x;i++){
            res+=l[i];
        }
        ans=max(ans,res);
        return;
    }
    solve(id+1,cnt,x);
    solve(id+1,cnt|(1<<(id-1)),x);
}
int main(){
    cin>>n>>m>>k;
    int sum=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>a[i][j];
            sum+=a[i][j];
        }
    }
    if(k>=n||k>=m){
        cout<<sum<<endl;
        return 0;
    }
    for(int x=0;x<=k;x++){ //x是需要的行 k-x就是需要的列
        solve(1,0,x);
    }
    cout<<ans;
}