ZZULIOJ 1734 堆【Dfs】水题

1734: 堆

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 325   Solved: 170

Submit Status Web Board

Description

Input

Output

Sample Input

3110310 5 31 21 351 2 3 4 53 12 12 42 5

Sample Output

YesNoYes

思路：

直接按照题意Dfs即可，维护每个点其子树的最大价值。

如果a【u】>这个最大价值，那么结果就是No

Ac代码：

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
int a[150000];
int flag;
vector<int >mp[150000];
int Dfs(int u,int from)
{
    int tmp=0;
    int maxn=0;
    for(int i=0;i<mp[u].size();i++)
    {
        int v=mp[u][i];
        if(v==from)continue;
        tmp=1;
        maxn=max(maxn,Dfs(v,u));
    }
    if(tmp==1)
    {
        if(a[u]>maxn)flag=1;
        return maxn;
    }
    else return a[u];
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]),mp[i].clear();
        for(int i=0;i<n-1;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            mp[x].push_back(y);
            mp[y].push_back(x);
        }
        flag=0;
        Dfs(1,-1);
        if(flag==0)printf("Yes\n");
        else printf("No\n");
    }
}