下面我们直接从题目入手分析差分和前缀和在题目中的应用:
(该题出自牛客小白月赛47)
设: N=1010,M=1e6+10;
分析:如果矩阵中有两个点的颜色相同则以着两个点为对角线的一个小正方形内所有的点都被认为是被覆盖了。因为同一种颜色的点可能存在多个,所以我们要找到同一种颜色的点的最大横纵坐标的下标和最小横纵坐标的下标,那么我么就可以认为以最大和最小横纵坐标为对角线的正方形内的点都被覆盖了,我们可以分别用四个数组:xmin[M]保存最小的横坐标,xmax[M]保存最大的横坐标,ymin[M]保存最小的纵坐标,ymax[M]保存最大的纵坐标。然后我们构造以一个和输入矩阵大小相同的二维举证b[N][N], 然后对b进行差分,如果xmin[i]=ymax[i]&&ymin[i]=ymax[i]我们就认为这种颜色的点只有一个直接跳过即可,否则就对最大和最小下标围成的正方形内的数值都加一:
for(int i=1;i<=M;i++)
{
if(xmax[i]!=-1)
{
if(xmax[i]==xmin[i]&&ymin[i]==ymax[i]) continue;
b[xmin[i]][ymin[i]]+=1;
b[xmin[i]][ymax[i]+1]-=1;
b[xmax[i]+1][ymin[i]]-=1;
b[xmax[i]+1][ymax[i]+1]+=1;
}
}
然后再对b数组求一遍前缀和;
最后遍历一遍b数组如果该点的值是0则认为这个点没有被覆盖答案加一。
下面是完整代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1010,M=1e6+10;
int xmin[M];
int xmax[M];
int ymin[M];
int ymax[M];
int b[N][N];
int n,m;
int main()
{
scanf("%d%d",&n,&m);
memset(xmin,0x3f,sizeof xmin);
memset(xmax,-1,sizeof xmax);
memset(ymin,0x3f,sizeof ymin);
memset(ymax,-1,sizeof ymax);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
int x;
scanf("%d",&x);
xmin[x]=min(xmin[x],i);
xmax[x]=max(xmax[x],i);
ymin[x]=min(ymin[x],j);
ymax[x]=max(ymax[x],j);
}
for(int i=1;i<=M;i++)
{
if(xmax[i]!=-1)
{
if(xmax[i]==xmin[i]&&ymin[i]==ymax[i]) continue;
b[xmin[i]][ymin[i]]+=1;
b[xmin[i]][ymax[i]+1]-=1;
b[xmax[i]+1][ymin[i]]-=1;
b[xmax[i]+1][ymax[i]+1]+=1;
}
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
b[i][j]=b[i][j-1]+b[i-1][j]-b[i-1][j-1]+b[i][j];
int ans=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(b[i][j]==0) ans++;
cout<<ans<<endl;
return 0;
}