AcWing 刷题计划:0x40 数据结构进阶——AcWing 240. 食物链

1.题目描述:


2.题意:
略。

3.思路:
a)带权并查集。 这是一道经典的带权并查集的题了。【最近刚学,拿来练手。】 关系之间给出来了:吃,被吃,同类。那么我们只需要按关系分配权值就可以了。我们这里设定:
0:同类。1:吃祖先。2:被祖先吃。 为啥这么设计呢?首先由于我们初始化的时候默认为0,自己和自己自然是同类啦,而1和2刚好是关于3互补的。
设计好权值,我们就可以按照带权并查集的做法去做了。

4.代码:

//AcWing 240. 食物链
//#include
//#pragma GCC optimize(3,"Ofast","inline")
#include
#include
#include
#include
#include
#include
#include
//#include
#include
#include
#include
#include
#include
#include
#define FAST ios::sync_with_stdio(false)
#define DEV_RND ((int)rand()*RAND_MAX+rand())
#define RND(L,R) (DEV_RND%((R)-(L)+1)+(L))
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i
#define per(i,n,a) for(int i=n-1;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define li inline
#define re register
using namespace std;
//typedef uniform_int_distribution RNDI;
typedef pair<int,int> PII;
typedef vector<int> VI;
typedef double db;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int Hash = 131;//13331
const int maxn = 1e5+5;
const int maxm = 100000+5;
const int mod = 1e9+7;
const int inf = 0x3f3f3f3f;
const double eps = 1e-7;
const double pi = acos(-1);
//int dir[4][2]={
     {-1,0},{1,0},{0,-1},{0,1}};
//li int f(int x){return x==par[x]?par[x]:par[x]=f(par[x]);}
//mt19937 eng(time(0));
//li int RND(int L,int R){RNDI rnd(L,R);return rnd(eng);}
li ll lowbit(ll x){
     return x&(-x);}
ll gcd(ll a,ll b){
     return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){
     if(!b){
     d=a,x=1,y=0;}else{
     ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
li ll qpow(ll a,ll b,ll MOD=mod){
     ll res=1;a%=MOD;while(b>0){
     if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res%MOD;}
li ll qmul(ll a,ll b,ll MOD=mod){
     return (a*b-(ll)((long double)a/MOD*b)*MOD+MOD)%MOD;}
li ll inv(ll x,ll p){
     return qpow(x,p-2,p);}
li ll jos(ll n,ll k,ll s=1){
     ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
db f(db x){
     return x;}
li db sim(db l,db r){
     return (f(l)+4.*f((l+r)/2.)+f(r))*(r-l)/6.;}
db asr(db l,db r,db ans,db eps){
     db m=l+(r-l)/2.,L=sim(l,m),R=sim(m,r);return fabs(L+R-ans)<=15.*eps?L+R+(L+R-ans)/15.:asr(l,m,L,eps/2)+asr(m,r,R,eps/2);}
db asr(db l,db r,db eps){
     return asr(l,r,sim(l,r),eps);}
namespace IO
{
     
	li ll read()
	{
     
		ll x=0,sign=1;char c=getchar();
		while(c>'9'||c<'0') {
     if(c=='-') sign=-1;c=getchar();}
		while('0'<=c&&c<='9') x=x*10+c-'0',c=getchar();
		return x*sign;
	}
	template<typename T>
	li void write(T x,char t='\n')
	{
     
		if(x<0){
     x=-x;putchar('-');};
		static int sta[25];int top=0;
		do{
     sta[top++]=x%10;}while(x/=10);
		while(top) putchar(sta[--top]+'0');
		putchar(t);
	}
}
using namespace IO;
/*-------------head-------------*/
//
int n,k;
int par[maxn],val[maxn];
int f(int x)
{
     
	if(x==par[x]) return x;
	int t=f(par[x]);
	val[x]=(val[x]+val[par[x]])%3;
	return par[x]=t;
}
li void solve()
{
     
	rep(i,1,n+1) par[i]=i,val[i]=0;
	k=read();
	int res=0;
	while(k--)
	{
     
		int v=read()-1,x=read(),y=read();
		if(x>n||y>n||(x==y&&v==1)) {
     res++;continue;}
		int fx=f(x),fy=f(y);
		if(fx==fy)
		{
     
			if((val[y]-val[x]+3)%3!=v) res++;
		}
		else
		{
     
			par[fx]=fy;
			val[fx]=(-val[x]+val[y]+v)%3;
		}
	}
	write(res);
	//puts("");
}
int main()
{
     
	//srand(time(0));
	//freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
	//for(int QwQ=read();QwQ;QwQ--) solve();
	while(~scanf("%d",&n)) solve();
	return 0;
}

你可能感兴趣的:(AcWing)