Codeforces 838A - Binary Blocks(二维前缀和+容斥)

838A - Binary Blocks

思路：求一下前缀和，然后就能很快算出每一小正方块中1的个数了，0的个数等于k*k减去1的个数，两个的最小值就是要加进答案的值。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset((a),(b),sizeof(a))
const int INF=0x3f3f3f3f;
const int N=;
char mp[N][N];
int Mp[N][N]={};

int main()
{
    ios::sync_with_stdio(false);
    cin.tie();
    int n,m;
    cin>>n>>m;
    for(int i=;i<=n;i++)cin>>mp[i]+;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        Mp[i][j]=mp[i][j]-'';
    }
    for(int i=;i<N;i++)
    {
        for(int j=;j<N;j++)
        {
            Mp[i][j]+=Mp[i-][j]+Mp[i][j-]-Mp[i-][j-];
        }
    }
    int minn=min(n,m);
    int ans=INF;

    for(int i=;i<=minn;i++)
    {
        bool flag=true;
        int x=n/i;
        int y=m/i;
        if(n%i)x++;
        if(m%i)y++;
        int cnt=;
        for(int j=;j<=x;j++)
        {
            for(int k=;k<=y;k++)
            {
                if(j==&&k==)
                {
                    int t=Mp[j*i][k*i];
                    cnt+=min(i*i-t,t);
                }
                else
                {
                    int t=Mp[j*i][k*i]+Mp[j*i-i][k*i-i]-Mp[j*i][k*i-i]-Mp[j*i-i][k*i];
                    cnt+=min(i*i-t,t);
                }
            }
        }
        ans=min(ans,cnt);
    }
    printf("%d\n",ans);
    return ;
}