1. 程式人生 > >HDU 5325 Crazy Bobo (樹形DP+思維)*

HDU 5325 Crazy Bobo (樹形DP+思維)*

#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
const int  maxn =500005;
using namespace std;

/*
題目大意:給定一棵樹,
你可以在樹中找一個聯通快,
問其聯通快的最大大小是多少,
條件是:把聯通塊中的所有權重排序,
然後按序掃描的過程中的兩點之間的路徑的點
的權值均比出發的點小。

聯通塊的條件本質上可以理解為:
只能在已經走過的點進行擴充套件,
那麼手動模擬一會兒後就可以看到,
稍微簡單點的情況,一個點為中心擴充套件開來的,
每條分支上的數字都是遞增的,
這樣就是說每次跳躍都經過的是走過的點,
可以保證符合題意,但其實這種狀態可以擴充套件,
在每個分支上其實還可以有分叉,也是可以以一個最小的為中心,
就是以多個最小的為一條鏈。

這樣樹形DP的思維就稍微有點了,
對每個子節點,我們維護比當前節點權重值小的子節點的DP值來更新當前節點的DP值,
然後在比當前節點大的子節點中用之前計算好的DP值去向下更新。

*/

int val[maxn],n;
int dp[maxn];
int x,y;

///容器儲存邊
vector<int> g[maxn];
int ans=0;
void dfs(int u,int pre,int len)///當前的權重和維護出來的大小
{
    dp[u]=len+1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i],tmp=len;
        if(v==pre) continue;
        if(val[u]>val[v]) continue;
        dfs(v,u,0);
        dp[u]+=dp[v];
    }

    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==pre) continue;
        if(val[v]>val[u]) continue;
        dfs(v,u,dp[u]);
    }
}

int main()
{
    while(scanf("%d",&n)==1)
    {
        ///init();
        for(int i=1;i<=n;i++) scanf("%d",&val[i]),g[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        memset(dp,0,sizeof(dp));
        dfs(1,-1,0);
        ans=0;for(int i=1;i<=n;i++) ans=max(ans,dp[i]);
        printf("%d\n",ans);
    }
    return 0;
}