（2015年郑州轻工业学院ACM校赛题） J 堆

判断是否是一个堆，把树构造好遍历一遍就OK了

 

#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;
　　　　　　　Tree[b][a] = 1;//谢谢提醒！
        }
 
        DFS(1);
        if(flag)
            printf("No\n");
        else
            printf("Yes\n");
    }
    return 0;
}