这道题目的解法的(向量法)
大神不愧是大神,竟然想到这样的方法。
设置一个数组pre ,保存 第i元素的直接前驱 pre【i】
同时设置一个数组offset,保存偏移量。
根据题目
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
那么d-1 表示偏移量
d=1时,d-1=0 ,表示同类
d=2时,d-1=1,表示x吃y。
d=0是,d-1+3=2 表示y吃x。
有了这个假设,我们就可以在并查集中完成任意两个元素之间的关系转换了。
不妨继续假设,x的当前集合根节点rootx,y的当前集合根节点rooty,x->y的偏移值为d-1(题中给出的询问已知条件)
(1)如果rootx和rooty不相同,那么我们把rooty合并到rootx上,并且更新relation关系域的值(注意:p[i].relation表示i的根结点到i的偏移量!!!!(向量方向性一定不能搞错))
此时 rootx->rooty = rootx->x + x->y + y->rooty,这一步就是大牛独创的向量思维模式
上式进一步转化为:rootx->rooty = (relation[x]+d-1+3-relation[y])%3 = relation[rooty],(模3是保证偏移量取值始终在[0,2]间)
这道题目是输入量有K句话(0 <= K <= 100,000),要用scanf 读取,用cin的时候超时了,用scanf 采用了266ms。
还有一定要注意向量的方向。
#include <iostream>
#include <stdio.h>
using namespace std;
const int N=50010;
int pre[N],offset[N];
int n,m;
void make_set(int n)
{
for(int i=0;i<n;i++)
{
pre[i]=i;
offset[i]=0 ;//自己与自己同类 ,所以偏移为0;
}
}
int find(int x)
{
if(x==pre[x])
return x;
int fx=find(pre[x]);
offset[x]=(offset[x]+offset[pre[x]])%3; //状态转移方程,important
return pre[x]=fx;
}
void union_set(int x,int y,int t)
{
int fx=find(x);
int fy=find(y);
pre[fy]=fx;
offset[fy]=(offset[x]+ t-1 -offset[y]+3)%3;
}
int main()
{
cin>>n>>m;
int d,x,y;
int ans=0;
make_set(n);
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&d,&x,&y);
if(x>n||y>n||(x==y&&d==2))
{
ans++;
continue;
}
if(find(x)==find(y))
{
if( (offset[y]-offset[x]+3)%3!=d-1)
ans++;
}
else
{
union_set(x,y,d);
}
}
cout<<ans<<endl;
}