2015 多校第三场 Crazy Bobo

2900+ms过的

# pragma comment(linker, "/STACK:1024000000,1024000000") //扩栈
# include <stdio.h>
# include <string.h>
# include <algorithm>
# include <queue>
# include <vector>
using namespace std;
const int MAXN=5*1e5+100;
int cnt,head[MAXN],n,w[MAXN],vis[MAXN],num;
struct Edge
{
    int to,next;
}edge[MAXN];
/*
给定n个点的有权无向图，每个顶点的权值不同。要求选择尽可能多的点，使这些点连通，并且将这些点按照权值排序后，
相邻顶点的路径上的点权都要小于这两个顶点的权值。
问题转化为：对于无向图的每一条边，由权值小的点向权值大的点连有向边，然后求每个点能到达的点有多少个，取大值，
即包含顶点个数最多的连通分量。由于各个点权值不同，不存在环，但图不一定连通。
注意：此算法会爆栈，因此需要拓展栈，C++有效。*/

/*根据本题题意，所给连接关系一定是一棵树，所以在dfs中可以不必判断走重复点的情况，加判断会TLE*/

void addedge (int u,int v)
{
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}

int dfs (int u) //求一棵树的节点个数
{
    vis[u]=1;
    int sum=0;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].to;
        sum=sum+dfs(v);
    }
    return sum+1;
}
int main ()
{
    int u,v,ans;
    while(scanf("%d",&n)!=EOF)
    {
        cnt=0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&w[i]);
        }
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            if(w[u]<w[v]){ vis[v]=1;addedge(u,v);}
            else {vis[u]=1;addedge(v,u);}
        }
        ans=-1;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i])
            {
               ans=max(ans,dfs(i));
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}