Vijos P1531 食物链

没看懂偏移向量和拆点

用的带权并查集水过去的

注意先判断 是否大于n

可能有 1 n+1 n+1

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 50000+10;

int fa[MAXN],fv[MAXN];
int find(int x)
{
	if(fa[x]!=x) 
	{
		int t=fa[x];
		fa[x]=find(fa[x]);
		fv[x]+=fv[t];
	}
	return fa[x];
}
void link(int x,int y,int opt)
{
	int fx=find(x);
	int fxv=fv[x];
	int fy=find(y);
	int fyv=fv[y];

	int v=fxv-fyv+opt;
	v=(v%3+3)%3;
	fa[fy]=fx;
	fv[fy]=v;
}int n,k;
int query(int x,int y,int opt)
{if(x>n||y>n) return 0;
	if(x==y) 
	{
		if(opt==0) return 2;
		return 0;
	}
	
	if(find(x)!=find(y)) return 1;
	if(((fv[x]-fv[y])%3+3)%3==(3-opt)%3) return 2;
	else return 0;
}
int main()
{
	 cin>>n>>k;
	for(int i=1;i<=n;i++) fa[i]=i;
		int cnt=0;
	for(int i=1;i<=k;i++)
	{
		int opt,x,y;
		scanf("%d %d %d",&opt,&x,&y);
		opt--;
		int t=query(x,y,opt);
		if(t==1) link(x,y,opt);
		if(t==0) cnt++;
	}
	cout<<cnt;
	
	return 0;
}