CF662C Binary Table FWT
有一个 $n$ 行 $m$ 列的表格,每个元素都是 0/1,每次操作可以选择一行或一列,将 0/1 翻转,即 0 换为 1,1 换为 0.
求:经过若干次操作后,表格中最多有多少个 1.
数据范围:$1\leqslant n \leqslant 20,1\leqslant m \leqslant 10^5$
我们发现,如果只允许操作行/列的话这道题就非常好做了.
但是由于行翻转和列翻转是有交叉部分的,所以我们就很难去计算.
然后你发现一个特别好的性质就是虽然 $m$ 很大,但是 $n$ 很小.
所以我们可以考虑从 $n$ 入手.
一个最暴力的做法就是 $2^n$ 枚举每个行翻不翻转,然后依次枚举每个列,每个列的贡献就是0和1数量较小值.
这个时间复杂度是 $O(2^nm)$ 的.
我们将每一列的 0/1 状态都压成二进制 $sta[i]$,那么假如说行的翻转状态时 $p$ 的话第 $i$ 列的状态就是 $sta[i]\bigoplus p$
( $\bigoplus $ 表示异或符号)
这个时候答案就是 $ans[p]=\sum_{i \bigoplus j=p}cnt[i]\times val[j]$
这个 $val[j]$ 表示 $j$ 表示的二进制中 0 和 1 数量的较小值,$cnt[i]$ 表示初始状态下状态为 $i$ 的列的个数.
用 FWT 优化即可.
#include <cstdio>
#include <algorithm>
#define N 21
#define ll long long
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
char S[1<<N];
int lim,cnt[1<<N],g[1<<N];
ll a[1<<N],b[1<<N],ans=100000000;
int lowbit(int t) { return t&(-t); }
void FWT(ll *a,int opt)
{
int i,j,k;
for(i=1;i<lim;i<<=1)
{
for(j=0;j<lim;j+=i<<1)
{
for(k=0;k<i;++k)
{
ll x=a[j+k],y=a[j+k+i];
a[j+k]=x+y,a[j+k+i]=x-y;
if(opt==-1) a[j+k]/=2,a[j+k+i]/=2;
}
}
}
}
int main()
{
// setIO("input");
int i,j,n,m;
scanf("%d%d",&n,&m),lim=1<<n;
for(i=0;i<n;++i)
{
scanf("%s",S);
for(j=0;j<m;++j)
{
if(S[j]=='1') g[j]|=(1<<i);
}
}
for(i=0;i<m;++i) ++a[g[i]];
for(i=1;i<lim;++i) cnt[i]=cnt[i>>1]+(i&1);
for(i=0;i<lim;++i) b[i]=min(cnt[i],n-cnt[i]);
FWT(a,1),FWT(b,1);
for(i=0;i<lim;++i) a[i]*=b[i];
FWT(a,-1);
for(i=0;i<lim;++i) ans=min(ans,a[i]);
printf("%lld\n",ans);
return 0;
}

浙公网安备 33010602011771号