Codeforce 490E(树形dp)

来源：互联网发布：淘宝如何做一件代发编辑：程序博客网时间：2024/05/21 07:55

链接：点击打开链接

题意：求树上的最长上升子序列长度

代码：

#include <vector>#include <stdio.h>#include <stdlib.h>#include <string.h>#include <iostream>#include <algorithm>using namespace std;const int INF=0x3f3f3f3f;vector<int> G[6005];int n,dp[6005],ans[6005],val[6005];void dfs(int s,int fa,int rt){    int i,cp,pos,tmp;    for(i=0;i<G[s].size();i++){        tmp=G[s][i];        if(tmp==fa)        continue;        pos=lower_bound(dp,dp+n,val[tmp])-dp;        cp=dp[pos];        dp[pos]=val[tmp];        ans[rt]=max(ans[rt],lower_bound(dp,dp+n,INF)-dp);        dfs(tmp,s,rt);        dp[pos]=cp;                             //将更改过的点回溯回去    }}int main(){    int i,j,u,v,res;    while(scanf("%d",&n)!=EOF){        for(i=0;i<=n;i++)        G[i].clear();        for(i=1;i<=n;i++)        scanf("%d",&val[i]);        for(i=1;i<n;i++){            scanf("%d%d",&u,&v);            G[u].push_back(v);            G[v].push_back(u);        }        memset(ans,0,sizeof(ans));        for(i=1;i<=n;i++){                      //就是枚举根节点，每次在树上进行lis            memset(dp,INF,sizeof(dp));            dp[0]=val[i];            dfs(i,0,i);        }        res=1;        for(i=1;i<=n;i++)        res=max(res,ans[i]);        printf("%d\n",res);    }    return 0;}

0 0