F - LIS on Tree(LIS&DFS)

F - LIS on Tree(LIS&DFS)

题目传送门

题意:给定一棵树,求所有结点到根结点的长度。

思路:显然根据的贪心思想,我们可以对其在树上进行操作,与普通的不同的是,一开始我们可以将存

的数组进行初始化为,这样每次只需要进行二分操作就行了,省去了直接添加到数组末尾的那一步。由于

不同的路径是不同的,所以每次搜索完一个结点就要回溯,还原到之前的数组。

这样问题就解决了。

时间复杂度:

AC代码:

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
struct edge{
    int to,nt;
}e[N<<1];
int ans[N],d[N],a[N],cnt=1,h[N],n;//ans[i]存放答案,d[i]表示LIS数组 
void add(int u,int v){
    e[cnt]={v,h[u]};
    h[u]=cnt++;
}
void dfs(int u,int fa){
    int p=lower_bound(d+1,d+n+1,a[u])-d;//基于LIS的贪心思想 
    int tmp=d[p];
    d[p]=a[u];
    ans[u]=max(ans[fa],p);//取最大值. 
    for(int i=h[u];i;i=e[i].nt)
    {
         int v=e[i].to;
         if(v==fa) continue;
         dfs(v,u);
    }
    d[p]=tmp;//回溯,因为不同子树的LIS路径不同. 
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1,u,v;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    memset(d,0x3f,sizeof d);
    dfs(1,0);
    for(int i=1;i<=n;i++)
        printf("%d\n",ans[i]);
    return 0;
}
全部评论

相关推荐

10-29 15:38
门头沟学院 Java
榕城小榕树:难道你简历里写了配送路径优化算法?
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务