AtCoder 2381 Nuske vs Phantom Thnook

题面

题意

给出一个01矩阵，任意两个1之间的路径（不能经过0）之多只有1条，多次询问每个子矩形中由1构成的联通块的个数。

做法

这题的关键是要抓住任意两个1之间的路径（不能经过0）之多只有1条这个条件，这就意味着1构成的联通块都是树，而树的点数-边数=1，因此只要用二维前缀和维护子矩阵中1的个数和1-1这样的边的个数，相减即为联通块的个数。

代码

#include<iostream>
#include<cstdio>
#define N 2010
using namespace std;

int n,m,Q,qz[N][N],ha[N][N],lie[N][N];
bool mm[N][N];
char str[N];

int main()
{
    int i,j,a,b,c,d;
    cin>>m>>n>>Q;
    for(i=1;i<=m;i++)
    {
        scanf("%s",str+1);
        for(j=1;j<=n;j++)
        {
            mm[i][j]=(str[j]=='1');
        }
        for(j=1;j<n;j++) if(mm[i][j]==mm[i][j+1]&&mm[i][j]) ha[i][j]=1;
        if(i>=2)
            for(j=1;j<=n;j++) if(mm[i][j]==mm[i-1][j]&&mm[i][j]) lie[i-1][j]=1;
    }
    for(i=1;i<=m;i++)
    {
        for(j=1;j<=n;j++)
        {
            qz[i][j]=qz[i-1][j]+qz[i][j-1]-qz[i-1][j-1]+mm[i][j];
            ha[i][j]+=ha[i-1][j]+ha[i][j-1]-ha[i-1][j-1];
            lie[i][j]+=lie[i-1][j]+lie[i][j-1]-lie[i-1][j-1];
        }
    }
    for(i=1;i<=Q;i++)
    {
        scanf("%d%d%d%d",&a,&b,&c,&d);
        printf("%d\n",
        qz[c][d]-qz[c][b-1]-qz[a-1][d]+qz[a-1][b-1]-
        (ha[c][d-1]-ha[c][b-1]-ha[a-1][d-1]+ha[a-1][b-1])-
        (lie[c-1][d]-lie[c-1][b-1]-lie[a-1][d]+lie[a-1][b-1])
        );
    }
}