Codeforces 838A. Binary Blocks （区间筛+二维前缀和, IndiaHacks 2nd Elimination 2017 ）

Problem

$N\times M$ 的 01 矩阵，要求将其严格划分为 $k\times k$ 的若干块（不足部分将原矩阵扩展，扩展处补 0 ）。要求将每个 $k\times k$ 的块中同为 1 或同为 0，若不满足，将任意一个 $1\to 0$ ，$0\to 1$ 的代价为 1 ，求选择一个可行的 k 使得代价最小。

Limit

$2\le n,m\le 2500$

$k\gt 1$

Idea

依次枚举所有可能的 k ，再统计每个 k 的最小代价，复杂度将达到 $mathcal{O}(N^3)$

考虑优化统计每个 k 的最小代价。由于 $N\times M$ 划分为 $k\times k$ 的块，则划分块数为 $\lceil \frac{N}{k} \rceil \times \lceil \frac{M}{k}\rceil$ 。可近似看作是 $\frac{N\times M}{k\times k}$ 。枚举 K 的总代价总小于 $N\times M \times \ln{(N\times M)}$ 。利用二维前缀和维护前缀块中 1 的个数，对于每个枚举块求最小值的复杂度仅 $\mathcal{O(1)}$ 的查询。

Code

#include<bits/stdc++.h>
using namespace std;
const int N =2500 + 10;
char mp[N][N];
int n, m, pre[N][N], ans;
int getOne(int l, int u, int r, int d)
{
    if(r > m)   r = m;
    if(d > n)   d = n;
    return pre[d][r] - pre[u-1][r] - pre[d][l-1] + pre[u-1][l-1];
}
int main()
{
    scanf("%d %d", &n, &m);
    ans = n*m;
    for(int i=1;i<=n;i++)
        scanf(" %s", mp[i]+1);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
        pre[i][j] = pre[i][j-1] + (mp[i][j]=='1'?1:0) + pre[i-1][j] - pre[i-1][j-1];
    for(int k=2;k<=max(n, m);k++)
    {
        int cnt = 0, dig;
        for(int i=1;i<=n;i+=k)
        for(int j=1;j<=m;j+=k)
        {
            dig = getOne(j, i, j+k-1, i+k-1);
            cnt += min(dig, k*k - dig);
        }
        ans = min(ans, cnt);
    }
    printf("%d\n", ans);
}