zoj 2684 ChonSu

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=1684

题目大意：

给定一个树的N（1~1000）个树叶间的距离（从左到右），一次询问，询问树叶s和t的距离。

ps：这个树是一个除了树叶结点，其余结点均有2个儿子。

题目思路：

树叶距离为2的两节点的必然是某个结点的左右儿子。

所以我们从左往右扫描初始间距数列，遇到第一个2就将其取出合成一个新的结点，然后这个位置两边的值减1，不断重复，知道数列空掉，这个用链表写起来非常方便，而且高效。

如（黑的表示间距，红色表示节点编号）

 13223345536

遇到第一个间距2，数列变为 127245536

代码：

#pragma comment(linker, "/STACK:102400000,102400000")
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<ctype.h>
#include<iostream>
#include<algorithm>
#include<stack>
#include<queue>
#include<list>
#include<map>
#include<set>
#include<vector>
#include<string>
using namespace std;
#define ll long long
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n))
#define clr_all(x,c) memset(x,c,sizeof(x))
#define IT iterator
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle l+r>>1
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define eps (1e-8)
#define PI 3.1415926535897932384626433832795
#define E 2.7182818284590452353602874713527
template <class T> T _min(T a,T b){return a<b? a:b;}
template <class T> T _max(T a,T b){return a>b? a:b;}
template <class T> T _abs(T a){return a>0? a:-a;}
template <class T> T _mod(T a,T m){return a<m? (a<0? (a%m+m)%m:a):a%m;}
template <class T> T _gcd(T a,T b){while(b){T t=b;b=a%b;a=t;}return a;}
template <class T> void _swap(T &a,T &b){T t=b;b=a;a=t;}
template <class T> void getmax(T &a,T b){a= a>b? a:b;}
template <class T> void getmin(T &a,T b){a= (a!=-1 && a<b)? a:b;}
int TS,cas=1;
const int M=2000+5;
int n;
int rt,tot;
struct node{
	int dis,l,r;
	node(int d=0,int _l=0,int _r=0){dis=d,l=_l,r=_r;}
};
list<node>l;
int fa[M],vis[M];
int dis(int s,int t){
	clr_all(vis,-1);
	int i,cnt;
	for(i=s,cnt=0;i!=-1;i=fa[i],cnt++) vis[i]=cnt;
	for(i=t,cnt=0;vis[i]==-1;i=fa[i]) cnt++;
	return vis[i]+cnt;
}

void run(){
	int i,j;
	scanf("%d",&n);
	l.clear();
	for(i=1;i<n;i++){
		scanf("%d",&j);
		l.push_back(node(j,i,i+1));
	}
	clr_all(fa,-1);
	rt=n,tot=0;
	list<node>::IT it=l.begin(),las,tmp;
	while(true){
		if(it->dis==2){
			++rt;
			fa[it->l]=rt,fa[it->r]=rt;
			if(it!=l.begin()){
				tmp=it,tmp--;
				tmp->dis--,tmp->r=rt;
			}
			tmp=it,tmp++;
			if(tmp!=l.end()) tmp->dis--,tmp->l=rt;
			if(it==l.begin()) las=it,las++;
			else las=it,las--;
			l.erase(it);
			if(l.empty()) break;
			it=las;
		}else it++;
	}
	int s,t;
	scanf("%d%d",&s,&t);
	printf("%d\n",dis(s,t));
}

void presof(){
}

int main(){
	//freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
	presof();
	//run();
	//while(~scanf("%d",&n)) run();
	for(scanf("%d",&TS),cas=1;cas<=TS;cas++) run();
	return 0;
}