并查集的拓展:
并查集最开始的使用 , 是用于判断一个图是否是连通图 , 由于并查集查询特点的特点(查询复杂度为O(1))所以用得很广。
并查集的变形一般是和向量偏移(类别偏移)一起结合 。
下面就通过这个题目来讨论向量偏移:
在并查集中 , 如果两个点时属于同一个集合 , 那么这两个点肯定就有一定的关系 , 所以我们认为只要a、b之间有关系 , 那么我们就让a、b的根节点一样 , 这样 , a、b间的关系就只有同一团伙和不是同一团伙。
然后我们再用一个数组来标记a、b节点分别和根节点的关系 , 因为我们只要知道a、b节点和根节点的关系 , 我们就能确定a、b的关系(确定中间点) , 因此我们只需要随着根节点的改变 , 来更新每个节点的偏移量就行了 , 这样我们就能在O(1)的时间复杂度内确定a、b节点的关系。
因此并查集的偏移向量 , 就是通过确定每个点和根节点的关系 , 然后就能确定任意点之间的关系
代码:
//0表示是同一个团体 , 1表示不是同一个团体
#include
#include
#include
using namespace std;
int n , m;
int p[100010] , f[100010];
int find(int x)
{
if(p[x] == x) return x;
int fa = p[x]; //记录根节点
p[x] = find(p[x]);
f[x] = (f[x]+f[fa])%2; //f[x]表示是相对以前根节点的状态 , f[fa]表示以前根节点相对当前根节点的状态 , 而相加就是表示点x相对根节点的状态
return p[x];
}
void fun(int x , int y)
{
int g = find(x) , h = find(y);
if(g == h) return ; //根节点都是一样时就直接跳出
else
{
p[g] = h;
f[g] = (f[x] + f[y] + 1)%2; //f[x]+1 == f[g] , f[g]+f[y] 就是点g相对于根节点h的关系
}
}
int main()
{
int t;
cin>>t;
while(t--)
{
cin>>n>>m;
getchar();
int i , j , x , y;
for(i = 1; i <= n; i++)
p[i] = i , f[i] = 0; //相对自己是0
char c;
for(i = 1; i <= m; i++)
{
scanf("%c %d %d" , &c , &x , &y);
getchar();
if(c == 'D')
{
fun(x , y);
}
else
{
if(n == 2)//题目说每个团伙至少有一个人
printf("In different gangs.\n");
else
{
int g = find(x) , h = find(y);
if(g != h)
printf("Not sure yet.\n");
else
{
if(f[x] == f[y])
printf("In the same gang.\n");
else
printf("In different gangs.\n");
}
}
}
}
}
return 0;
}
poj 1703 并查集的拓展
最新推荐文章于 2023-05-21 11:50:43 发布