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;
}