CodeForces 662 C.Binary Table

Description

给出一个$n\times m$的$01$矩阵，每次操作可以反转某行或某列，问可以得到的$1$的最小值

Input

第一行两个整数$n,m$表示矩阵行列数，之后输入一个$n\times m$的$01$矩阵$(1\le n\le 20,1\le m\le 100000)$

Output

输出经过若干次操作后矩阵里$1$的数量最小值

Sample Input

3 4
0110
1010
0111

Sample Output

2

Solution

注意到行数比较小，考虑枚举行的反转状态$sta$，二进制第$i$位是$1$说明反转第$i$行，否则不反转，以$ans[sta]$表示行反转状态为$sta$可以得到的$1$数量最小值，将一列$01$看作一个整数，以$num[i]$表示原矩阵中值为$i$的列数，以$f[i]$表示$i$的二进制表示中$1$的个数，那么一个值为$i$的列不反转则有$f[i]$个$1$，反转后会得到$n-f[i]$个$1$，以$dp[i]=min(f[i],n-f[i])$表示一个值为$i$的列通过列反转可得到的$1$的数量最小值，注意到一个值为$i$的列经过行反转会变成一个值为$i$^$sta$的列，则有
$ans[sta]=\sum\limits_{j=0}^{2^n-1}num[j]\cdot dp[j$^$sta]=\sum\limits num[j]\cdot dp[k],j$^$k=sta$

用$FWT$求出序列$num$和$dp$的异或卷积即得到$ans$值，$min\{ans[i],0\le i\le 2^n-1\}$即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int maxn=(1<<20)+5;
int n,m,a[maxn],f[maxn];
ll num[maxn],dp[maxn];
char s[maxn];
void FWT(ll *a,int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
                {  
                    ll x=a[i+j],y=a[i+j+d];  
                    a[i+j]=x+y,a[i+j+d]=x-y;   
                } 
}   
void UFWT(ll *a,int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
            {  
                ll x=a[i+j],y=a[i+j+d];  
                a[i+j]=(x+y)/2,a[i+j+d]=(x-y)/2;    
            }  
} 
int main()
{
    scanf("%d%d",&n,&m);
    int N=1<<n;
    for(int i=0;i<n;i++)
    {
        scanf("%s",s+1);
        for(int j=1;j<=m;j++)
            if(s[j]=='1')a[j]^=(1<<i);
    }
    for(int i=1;i<=m;i++)num[a[i]]++;
    for(int i=1;i<N;i++)f[i]=f[i/2]+(i&1);
    for(int i=0;i<N;i++)dp[i]=min(f[i],n-f[i]);
    FWT(dp,N),FWT(num,N);
    for(int i=0;i<N;i++)dp[i]*=num[i];
    UFWT(dp,N);
    ll ans=dp[0];
    for(int i=1;i<N;i++)ans=min(ans,dp[i]);
    printf("%I64d\n",ans);
    return 0;
}