二进制枚举
颓废了几个月没打代码了,今天开始要认真的代码了
记录一下学习的二进制枚举
二进制枚举就是用01串来表示一个状态,选或者不选。
列如我们有五瓶矿泉水,我们就可以用0 1来选择要不要。
就是从00000~11111之间的枚举,也就是[0,2^n-1]之间的枚举。
如此就有
for(int i=0;i<(1<<n);i++)
我们还需知道哪些位置选了,如此就有
for(int j = 0; j < n; j++)
if(i & (1 << j))//判断第j位选没选
printf(" %d ",a[j]);
<<左移符号,相对于数值乘2,
&与运算,1和1为1,1和0为0, 0和0为0
下面一道例题:
矩阵消除游戏
牛妹在玩一个名为矩阵消除的游戏,矩阵的大小是n行m列,第i行第j列的单元格的权值为aij,牛妹可以进行k个回合的游戏,在每个回合,牛妹可以选择一行或者选择一列,然后将这一行或者这一列的所有单元格中的权值变为0,同时牛妹的分数会加上这一行或者这一列中的所有单元格的权值的和。
牛妹想最大化她的得分,球球你帮帮她吧!
输入描述:
第一行三个整数n,m,k
接下来n行每行m个整数表示矩阵中各个单元格的权值。
输出描述:
输出一个整数表示牛妹能获得的最大分数。
输入
3 3 2
101 1 102
1 202 1
100 8 100
输出
414
思路:一开始我以为只选行或者只选列,结果直接错了。
正确思路应该是:因为先选行后选列和先选列后选行,结果一样。所以枚举选行或者枚举选列都行。
我是枚举选行,后在选列
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=1e5+5;
ll s[20][20];
ll a[20];//每行和
ll b[20];//每列和
bool cmp(ll a,ll b){
return a>b;
}
int main() {
ll n,m,k;
cin>>n>>m>>k;
ll sum=0;
for(ll i=0;i<n;i++){
for(ll j=0;j<m;j++){
cin>>s[i][j];
a[i]+=s[i][j];
sum+=s[i][j];
}
}
if(k>=min(n,m)) cout<<sum<<endl;
else{
bool vis[20];
ll ans=0;
for(ll i=0;i<(1<<n);i++){
memset(vis,0,sizeof(vis));
memset(b,0,sizeof(b));
ll cnt=0;//记录枚举的行数
ll x=0;
for(ll j=0;j<n;j++){
if(i&(1<<j)){
vis[j]=1;
cnt++;
x+=a[j];//选中的行加上
}
}
ll cnt2=k-cnt;//还能选的列数
if(cnt2<0) continue;
for(ll j=0;j<m;j++){
for(ll ki=0;ki<n;ki++){
if(!vis[ki]) b[j]+=s[ki][j];
}
}
sort(b,b+m,cmp);
for(ll j=0;j<cnt2;j++) x+=b[j];
ans=max(ans,x);
}
cout<<ans<<endl;
}
return 0;
}