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.7182818284590452353602874713527template <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;}