题目链接:https://vjudge.net/contest/213916#problem/L
这个并查集寻找连通块的操作很有必要学习一下,在这种图中可以更快地寻找满足条件的区域。
题意就是给出一个n*m大小的图,再给一个数组,对于每个数组中的数,寻找图中满足点权值大于这个数的区域的个数(满足条件的连通块的大小)。
枚举所有的数组中的值,复杂度O(T*n*m)=10^11,肯定不行,所以得想办法加速。将数组从大到小排序,再将所有的点,按照权值大小从大到小排序,那么枚举数组时,只寻找比它大的点,将他们用并查集保存为许多连通块;然后依次向较小的元素遍历时,相应的满足条件的点也会增加,这时就会增加并查集的大小或者数目,也就是将遍历过程变成了一个合并集合,新增集合的过程。具体细节看代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
#include <cmath>
using namespace std;
const int maxn=1010;
const int maxn1=1e5+10;
typedef long long ll;
ll mp[maxn][maxn];
ll T[maxn1];
ll ans[maxn1];
int fa[maxn*maxn];
int indice[maxn][maxn];//通过x,y寻找Point的index
const int dx[]={1,0,-1,0};
const int dy[]={0,1,0,-1};
int n,m;
struct Point
{
int x,y;
ll v;
Point(int _x=0,int _y=0,ll _v=0):x(_x),y(_y),v(_v){}
bool operator < (const Point &rhs)const
{
return v>rhs.v;
}
}P[maxn*maxn];
//并查集
int find(int x)
{
return fa[x]==x?fa[x]:fa[x]=find(fa[x]);
}
bool cmp(const int &a,const int &b)
{
return a>b;
}
vector<Point>v;
bool judge(int x,int y)
{
if(x>0&&x<=n&&y>0&&y<=m)
return true;
return false;
}
//获取原图中与p点上下左右相邻的点的编号
void getAdj(Point p,vector<Point>&v)
{
v.clear();
int x=p.x;
int y=p.y;
for(int i=0;i<4;i++)
{
int xx=x+dx[i];
int yy=y+dy[i];
if(judge(xx,yy))
{
int index=indice[xx][yy];
v.push_back(P[index]);
}
}
}
int main()
{
int Test;
scanf("%d",&Test);
while(Test--)
{
scanf("%d%d",&n,&m);
for(int i=0;i<maxn1;i++)ans[i]=0;
for(int i=0;i<maxn1;i++)T[i]=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
scanf("%lld",&mp[i][j]);
P[(i-1)*m+j]=Point(i,j,mp[i][j]);
}
}
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)scanf("%lld",&T[i]);
sort(P+1,P+n*m+1);//由大到小
sort(T+1,T+t+1,cmp);//由大到小排序
for(int i=1;i<=n*m;i++)
{
fa[i]=i;
indice[P[i].x][P[i].y]=i;
}
int tmp=1;
for(int i=1;i<=t;i++)
{
ans[i]=ans[i-1];
while(T[i]<P[tmp].v&&tmp<=n*m)
{
ans[i]++;
getAdj(P[tmp],v);
//遍历周围的点
for(int j=0;j<v.size();j++)
{
Point &pn=v[j];
if(pn.v<=T[i])continue;
int index=indice[pn.x][pn.y];
int pnpa=find(index);//寻找对应的连通块
if(pnpa==find(tmp))continue;//如果是同一连通块,也就是已经被标记了,不管
//如果未被标记根节点,合并连通块
fa[pnpa]=tmp;
ans[i]--;
}
tmp++;
}
}
for(int i=t;i>=1;i--)
printf("%d ",ans[i]);
cout<<endl;
}
return 0;
}