hdu 5444 Elven Postman(二叉搜索树)

Elven Postman

Time Limit: 1500/1000 MS (Java/Others)    Memory Limit: 131072/131072 K (Java/Others)
Total Submission(s): 1223    Accepted Submission(s): 694

Problem Description

Elves are very peculiar creatures. As we all know, they can live for a very long time and their magical prowess are not something to be taken lightly. Also, they live on trees. However, there is something about them you may not know. Although delivering stuffs through magical teleportation is extremely convenient (much like emails). They still sometimes prefer other more “traditional” methods.

So, as a elven postman, it is crucial to understand how to deliver the mail to the correct room of the tree. The elven tree always branches into no more than two paths upon intersection, either in the east direction or the west. It coincidentally looks awfully like a binary tree we human computer scientist know. Not only that, when numbering the rooms, they always number the room number from the east-most position to the west. For rooms in the east are usually more preferable and more expensive due to they having the privilege to see the sunrise, which matters a lot in elven culture.

Anyways, the elves usually wrote down all the rooms in a sequence at the root of the tree so that the postman may know how to deliver the mail. The sequence is written as follows, it will go straight to visit the east-most room and write down every room it encountered along the way. After the first room is reached, it will then go to the next unvisited east-most room, writing down every unvisited room on the way as well until all rooms are visited.

Your task is to determine how to reach a certain room given the sequence written on the root.

For instance, the sequence 2, 1, 4, 3 would be written on the root of the following tree.

 

Input

First you are given an integer  T(T≤10)  indicating the number of test cases.

For each test case, there is a number  n(n≤1000)  on a line representing the number of rooms in this tree.  n  integers representing the sequence written at the root follow, respectively  a1,...,an  where  a1,...,an∈{1,...,n} .

On the next line, there is a number  q  representing the number of mails to be sent. After that, there will be  q  integers  x1,...,xq  indicating the destination room number of each mail.

 

Output

For each query, output a sequence of move ( E  or  W ) the postman needs to make to deliver the mail. For that  E  means that the postman should move up the eastern branch and  W  the western one. If the destination is on the root, just output a blank line would suffice.

Note that for simplicity, we assume the postman always starts from the root regardless of the room he had just visited.

 

Sample Input

  
  

   
   2
4
2 1 4 3
3
1 2 3
6
6 5 4 3 2 1
1
1
  
  

 

Sample Output

  
  

   
   E

WE
EEEEE
  
  

 

solution：

按照二叉搜索树建树，然后比它小的在它的东面，比它大的在它的西面，我直接在查找时候就输出W和E，二叉搜索树是按照《挑战程序设计竞赛》上的板码的，好像有点丑，但是能1A我也心满意足了。

#include<cstdio>
#include<algorithm>
using namespace std;
struct node{
    int val;
    node* l, *r;
};
node *insert(node *p, int x)
{
    if (p == NULL)
    {
        node *q = new node;
        q->val = x;
        q->l = q->r = NULL;
        return q;
    }
    else
    {
        if (x < p->val)p->l = insert(p->l, x);
        else p->r = insert(p->r, x);
        return p;
    }
}
void find(node *p, int x)
{
    if (x == p->val||p==NULL)return;
    else if (x < p->val)
    {
        printf("E"); return find(p->l, x);
    }
    else {
        printf("W"); return find(p->r, x);
    }
}
int main()
{
    int t, n,x,m;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%d", &n);
        node *head = new node;
        head->l = head->r = NULL;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &x);
            if (i == 0)head->val = x;
            else insert(head, x);
        }
        scanf("%d", &m);
        while (m--)
        {
            scanf("%d", &x);
            find(head, x);
            printf("\n");
        }
    }
    return 0;
}