矩阵消除(二进制枚举,贪心)

题目传送门

文章目录

思路:

  1. 题目说我们选某几列,某几行,使得我们获得的权值最大。
  2. 那么我们二进制枚举行,例如有4*4的矩阵,k=4, 1010表示,1行和第4行需要选,剩下2个选列,对应行置为0。那么再取处理后的矩阵的权值和前2大的列即可

AC代码:

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int maxn = 200;
int a[maxn][maxn];
LL ans = -1e18;
int main()
{
   std::ios::sync_with_stdio(false);
   int n,m,k;
   cin>>n>>m>>k;
   LL all = 0;
   for(int i=1;i<=n;i++)
   {
       for(int j=1;j<=m;j++)
       {
           cin>>a[i][j];
           all+=a[i][j];
       }
   }
   if(k>=n||k>=m)
   {
       cout<<all<<endl;
       return 0;
   }
   int e = 1<<n;
   //进行二进制枚举行
   for(int i=0;i<e;i++)
   {
       int x = i;
       int b[maxn];
       memset(b,0,sizeof(b));
       int e2 = n;
       int cnt = 0;
       while(x)
       {
           b[e2--] = x&1;
           if(b[e2+1]) cnt++;
           x>>=1;
       }
       LL sum = 0;
       if(cnt>k) continue;
        int a2[maxn][maxn];
        memset(a2,0,sizeof(a2));
        for(int r=1;r<=n;r++)
        {
            for(int c= 1;c<=m;c++)
        {
            a2[r][c] = a[r][c];
        }

        }
        //处理要选的行
       for(int r=1;r<=n;r++)
       {
           if(b[r])
           {
               for(int c=1;c<=m;c++)
               {
                   sum+=a2[r][c];
                   a2[r][c] = 0;
               }
           }
       }
       //剩下的k-cnt都是枚举列
       LL col[maxn];//根据k的范围来确定大小,容易越界
       memset(col,0,sizeof(col));
       for(int j=1;j<=m;j++)
       {
           for(int k=1;k<=n;k++)
           {
               col[j]+= a2[k][j];
           }
       }
       //从大到小排列
       sort(col+1,col+m+1,greater<LL>());
       for(int j=1;j<=k-cnt;j++)
       {
            sum+=col[j];
       }

       ans = max(ans,sum);
   }
   cout<<ans<<endl;
   return 0;
}

全部评论

相关推荐

团子请爱我一次_十月...:不是戈门,干哪来了,这就是java嘛
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
正在热议
更多
# 你的mentor是什么样的人? #
3895次浏览 28人参与
# 你觉得mentor喜欢什么样的实习生 #
10253次浏览 284人参与
# 未岚大陆求职进展汇总 #
23818次浏览 112人参与
# 帮我看看,领导说这话什么意思? #
6110次浏览 25人参与
# 没有家庭托举的我是怎么找工作的 #
12292次浏览 158人参与
# 怎么给家人解释你的工作? #
1371次浏览 16人参与
# 平安产险科技校招 #
2404次浏览 0人参与
# 求职低谷期你是怎么度过的 #
5208次浏览 91人参与
# 26届秋招公司红黑榜 #
11934次浏览 41人参与
# 从哪些方向判断这个offer值不值得去? #
6545次浏览 93人参与
# 同bg的你秋招战况如何? #
158817次浏览 927人参与
# 度小满求职进展汇总 #
10078次浏览 51人参与
# 实习必须要去大厂吗? #
146625次浏览 1541人参与
# 校招泡的最久的公司是哪家? #
4531次浏览 22人参与
# 你有哪些缓解焦虑的方法? #
37173次浏览 835人参与
# 面试紧张时你会有什么表现? #
1694次浏览 20人参与
# 你喜欢工作还是上学 #
77572次浏览 860人参与
# 入职第一天,你准备什么时候下班 #
85471次浏览 467人参与
# 秋招想进国企该如何准备 #
97706次浏览 487人参与
# 简历无回复,你会继续海投还是优化再投? #
103582次浏览 819人参与
# 机械人的工作环境真的很差吗 #
25014次浏览 119人参与
# 独居后,你的生活是更好了还是更差了? #
28125次浏览 263人参与
牛客网
牛客网在线编程
牛客网题解
牛客企业服务