HDU 2545 树上战争（求树的高度）

树上战争

Time Limit: 10000/4000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1051 Accepted Submission(s): 594

Problem Description

给一棵树，如果树上的某个节点被某个人占据，则它的所有儿子都被占据，lxh和pfz初始时分别站在两个节点上，谁当前所在的点被另一个人占据，他就输了比赛，问谁能获胜

Input

输入包含多组数据
每组第一行包含两个数N，M（N，M<=100000），N表示树的节点数，M表示询问数，N=M=0表示输入结束。节点的编号为1到N。
接下来N-1行，每行2个整数A，B(1<=A,B<=N)，表示编号为A的节点是编号为B的节点的父亲
接下来M行，每行有2个数，表示lxh和pfz的初始位置的编号X，Y(1<=X,Y<=N,X!=Y)，lxh总是先移动

Output

对于每次询问，输出一行，输出获胜者的名字

Sample Input

2 1
1 2
1 2
5 2
1 2
1 3
3 4
3 5
4 2
4 5
0 0

Sample Output

lxh
pfz
lxh

提示：

本题输入、输出都很多，请使用scanf和printf代替cin、cout。

Source

UESTC 6th Programming Contest Online

分析

　　该题目的题意跟清楚，看谁先占领对方的点。双方肯定都是向树根方向走的，只有那样才能占据更多的点。所以谁离树根最近，则谁就能赢。所以该题目是一个求树高的问题。
　　用一个数组存储树，数组比链表好在可以快速找到该点的位置。用一个index数组记录该点在数组中的位置，就可以快速查找到。
　　定义一个树节点的结构体，其中包括该点名称，父节点位置。

代码如下

#include <stdio.h>#include <string.h>typedef struct{    int a;    int pre;}node;int index[100005];node ans[100005];int main(){    int n,m;    int i,j;    int a,b;    while(scanf("%d %d",&n,&m),n||m)    {        memset(index,0,sizeof(index));        int count=1;        node temp;        for(i=0;i<n-1;i++)        {            scanf("%d %d",&a,&b);            if(index[a]==0)            {                temp.a = a;                temp.pre=-1;                index[a] = count;                ans[count++] = temp;            }            if(index[b]==0)            {                temp.a=b;                temp.pre = index[a];                index[b] = count;                ans[count++] = temp;            }        }        for(i=0;i<m;i++)        {            scanf("%d %d",&a,&b);            int height1=0;            int height2=0;            while(ans[index[a]].pre!=-1)            {                a = ans[ans[index[a]].pre].a;                height1++;            }            while(ans[index[b]].pre!=-1)            {                b = ans[ans[index[b]].pre].a;                height2++;            }            if(height1<=height2)                printf("lxh\n");            else                printf("pfz\n");        }    }    return 0;}

优化后代码

#include <stdio.h>#include <string.h>int fetch[100005];int main(){    int n,m;    int i,j;    int a,b;    while(scanf("%d %d",&n,&m),n||m)    {        memset(fetch,0,sizeof(fetch));        for(i=0;i<n-1;i++)        {            scanf("%d %d",&a,&b);            fetch[b]=a;        }        for(i=0;i<m;i++)        {            scanf("%d %d",&a,&b);            int height1=0;            int height2=0;            while(fetch[a]!=0)            {                a=fetch[a];                height1++;            }            while(fetch[b]!=0)            {                b=fetch[b];                height2++;            }            if(height1<=height2)                printf("lxh\n");            else                printf("pfz\n");        }    }    return 0;}