首页 > 代码库 > 【思路】Gym - 101173F - Free Figurines

【思路】Gym - 101173F - Free Figurines

套娃形成一些链形结构,给你套娃的初始状态和目标状态,问你需要几步(将最外层套娃打开,以及将一整套套娃塞进一个空套娃都算一步)才能达到。

容易发现,只有每条链链尾的匹配段可以不拆,其他的都得拆开。

#include<cstdio>
using namespace std;
int n,ra[100010],rb[100010],fa[100010],fb[100010],sa[100010],sb[100010];
int ans1,ans2;
void df1(int rt,int U){
	ra[U]=rt;
	if(sa[U]){
		df1(rt,sa[U]);
	}
}
void df2(int rt,int U){
	rb[U]=rt;
	if(sb[U]){
		df2(rt,sb[U]);
	}
}
int main(){
	int x;
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%d",&x);
		if(x){
			fa[i]=x;
			sa[x]=i;
			++ans1;
		}
	}
	for(int i=1;i<=n;++i){
		if(!fa[i]){
			df1(i,i);
		}
	}
	for(int i=1;i<=n;++i){
		scanf("%d",&x);
		if(x){
			++ans2;
			fb[i]=x;
			sb[x]=i;
		}
	}
	for(int i=1;i<=n;++i){
		if(!fb[i]){
			df2(i,i);
		}
	}
	for(int i=1;i<=n;++i){
		if(!sa[i] && !sb[i]){
			int U1=fa[i],U2=fb[i];
			while(U1 && U2 && U1==U2){
				--ans1;
				--ans2;
				U1=fa[U1];
				U2=fb[U2];
			}
		}
	}
	printf("%d\n",ans1+ans2);
	return 0;
}

【思路】Gym - 101173F - Free Figurines