【二叉树、堆】15轻院校赛-J-堆

原题：http://acm.zzuli.edu.cn/problem.php?cid=1099&pid=9

【描述】

【输入】

【输出】

Sample Input

3

1

10

3

10 5 3

1 2

1 3

5

1 2 3 4 5

3 1

2 1

2 4

2 5

Sample Output

Yes

No

Yes

 

【代码】

转自：http://www.cnblogs.com/chenchengxun/p/4443560.html

#include<stdio.h>
#include<iostream>
#include<stack>
#include<queue>
#include<math.h>
#include<stdlib.h>
#include<cstring>
#include<algorithm>
using namespace std;
#define Max(a,b) (a>b?a:b)
#define Min(a,b) (a<b?a:b)
#define INF 0xfffffff
#define maxn 410
 
int Tree[maxn][maxn], n, flag, Value[maxn], vis[maxn];
 
void DFS(int k)
{
 
    vis[k] = 1;
    for(int i=1; i<=n; i++)
    {
        if(Tree[k][i] && !vis[i])
        {
            if(Value[k] <= Value[i])
                DFS(i);
            else
                flag = 1;
        }
    }
}
 
int main()
{
    int T, i, a, b;
 
    cin >> T;
 
    while(T--)
    {
        cin >> n;
        flag = 0;
        memset(Tree, 0, sizeof(Tree));
        memset(vis, 0, sizeof(vis));
        for(i=1; i<=n; i++)
            cin >> Value[i];
 
        for(i=1; i<n; i++)
        {
            cin >> a >> b;
            Tree[a][b] = 1;
        }
 
        DFS(1);
        if(flag)
            printf("No
");
        else
            printf("Yes
");
    }
    return 0;
}