2021牛客暑期多校训练营7补题记录

xay loves trees

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long LL;
const int maxn=5e6+10;
int n,m,r;
int e,h[maxn],ne[maxn],ver[maxn];
int tot,h1[maxn],ne1[maxn],ver1[maxn];
int a[maxn*4],laz[maxn*4];
int son[maxn],id[maxn],fa[maxn],cnt,dep[maxn],siz[maxn],top[maxn]; 
int res=0,ans=0;
int dist[maxn];
void add1(int x,int y){
    ver1[++tot]=y;
    ne1[tot]=h1[x];
    h1[x]=tot;
}
void add(int x,int y){
    ver[++e]=y;
    ne[e]=h[x];
    h[x]=e;
}
void pushdown(int rt){
    laz[rt*2]=laz[rt];
    laz[rt*2+1]=laz[rt];
    a[rt*2]=laz[rt];
    a[rt*2+1]=laz[rt];
    laz[rt]=0;
}
void build(int rt,int l,int r){

    if(l==r){
        a[rt]=0;
        return;
    }
    int mid=(l+r)>>1;
    build(rt*2,l,mid);
    build(rt*2+1,mid+1,r);
    a[rt]=max(a[rt<<1],a[rt<<1|1]);
}
int query(int rt,int l,int r,int L,int R){
  //  cout<<l<<"uu"<<r<<" "<<L<<" "<<R<<" "<<rt<<endl;
    int ans=0;
    if(L<=l&&r<=R){
        return a[rt];
    }
    if(laz[rt])pushdown(rt);
    int mid=(l+r)>>1;
    if(L<=mid)ans=max(ans,query(rt*2,l,mid,L,R));
    if(R>mid)ans=max(ans,query(rt*2+1,mid+1,r,L,R));
    return ans;
}
void update(int rt,int l,int r,int pos,int k){
    //cout<<l<<" "<<r<<" "<<L<<" "<<R<<" "<<k<<endl;
   // cout<<l<<"SS"<<r<<endl;
    if(l==r){
        laz[rt]=k;
        a[rt]=k;
        return;
    }
    int mid=(l+r)>>1;
   // if(laz[rt])pushdown(rt);
    if(pos<=mid)update(rt*2,l,mid,pos,k);
    if(pos>mid)update(rt*2+1,mid+1,r,pos,k);
    a[rt]=max(a[rt*2],a[rt*2+1]);
}
void dfs1(int x,int f,int deep){
    dep[x]=deep;siz[x]=1;fa[x]=f;
    int  maxson=-1;
    for(int i=h[x];i;i=ne[i]){
        int y=ver[i];
        if(y==f)continue;
        dfs1(y,x,deep+1);
        siz[x]+=siz[y];
        if(siz[y]>maxson){
            maxson=siz[y],son[x]=y;
        }
    }
}
void dfs2(int x,int topf){
    id[x]=++cnt;
    top[x]=topf;
    if(!son[x])return;
    dfs2(son[x],topf);
    for(int i=h[x];i;i=ne[i]){
        int y=ver[i];
        if(y==fa[x]||y==son[x])continue;
        dfs2(y,y);
    }
}
void updRange(int x,int k){
    //while(top[x]!=top[y]){
    //    if(dep[top[x]]<dep[top[y]])swap(x,y);
    //    update(1,1,n,id[top[x]],id[x],k);
    //    x=fa[top[x]];
    //}
    //if(dep[x]>dep[y])swap(x,y);
    update(1,1,n,id[x],k);
} 
int qRange(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans=max(ans,query(1,1,n,id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans=max(ans,query(1,1,n,id[x],id[y]));
    return ans;
}
int qSon(int x){
    return query(1,1,n,id[x],id[x]+siz[x]-1);
}
//void updSon(int x,int k){
  //  update(1,1,n,id[x],id[x]+siz[x]-1,k);
//}
void dfs(int x,int fa,int u){
    int t=qRange(1,x);
    int t1=qSon(x);
   // cout<<qSon(4)<<" "<<x<<" "<<t<<"SS"<<t1<<endl;
    u=max(u,max(t1,t));
    ans=max(ans,dist[x]-u);
    updRange(x,dist[x]);
    for(int i=h1[x];i;i=ne1[i]){
        int y=ver1[i];
        if(y==fa)continue;
    //    cout<<"UU"<<y<<endl;
        dist[y]=dist[x]+1;
        dfs(y,x,u);
    }
    updRange(x,0);
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        ans=0;
        scanf("%d",&n);
        ans=0;
        e=tot=0;
        cnt=0;
        for(int i=0;i<=n+1;i++)dist[i]=h[i]=h1[i]=son[i]=id[i]=fa[i]=dep[i]=siz[i]=top[i]=0;
        r=1;
        for(int i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            add1(a,b);add1(b,a);
        }
        for(int i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            add(a,b);add(b,a);
        }
        dfs1(r,0,1);
        dfs2(r,r);
        build(1,1,n);
        dist[1]=1;
        dfs(1,-1,0);
        printf("%d\n",ans);
    }
    return 0;
}
全部评论

相关推荐

上周组里招人,我面了六个候选人,回来跟同事吃饭的时候聊起一个让我挺感慨的现象。前三个候选人,算法题写得都不错。第一道二分查找,五分钟之内给出解法,边界条件也处理得干净。第二道动态规划,状态转移方程写对了,空间复杂度也优化了一版。我翻他们的简历,力扣刷题量都在300以上。后三个呢,就有点参差不齐了。有的边界条件没处理好,有的直接说这道题没刷过能不能换个思路讲讲。其中有一个女生,我印象特别深——她拿到题之后没有马上写,而是先问我:“面试官,我能先跟你确认一下我对题目的理解吗?”然后她把自己的思路讲了一遍,虽然最后代码写得不是最优解,但整个沟通过程非常顺畅。这个女生的代码不是最优的,但当我问她“如果这里是线上环境,你会怎么设计’的时候,她给我讲了一套完整的方案——异常怎么处理、日志怎么打、怎么平滑发布。她对这是之前在实习的时候踩过的坑。”我在想LeetCode到底在筛选什么?我自己的经历可能有点代表性。我当年校招的时候,也是刷了三百多道题才敢去面试。那时候大家都刷,你不刷就过不了笔试关。后来工作了,前三年基本没再打开过力扣。真正干活的时候,没人让你写反转链表,也没人让你手撕红黑树。更多的是:这个接口为什么慢了、那个服务为什么OOM了、线上数据对不上了得排查一下。所以后来我当面试官,慢慢调整了自己的评判标准。算法题我还会出,但目的变了。我出算法题,不是想看你能不能背出最优解。而是想看你拿到一个陌生问题的时候,是怎么思考的。你会先理清题意吗?你会主动问边界条件吗?你想不出来的时候会怎么办?你写出来的代码,变量命名乱不乱、结构清不清楚?这些才是工作中真正用得到的能力。LeetCode是一个工具,不是目的。它帮你熟悉数据结构和常见算法思路,这没问题。但如果你刷了三百道题,却说不清楚自己的项目解决了什么问题、遇到了什么困难、你是怎么解决的,那这三百道题可能真的白刷了。所以还要不要刷LeetCode?要刷,但别只刷题。刷题的时候,多问自己几个为什么:为什么用这个数据结构?为什么这个解法比那个好?如果换个条件,解法还成立吗?把刷题当成锻炼思维的方式,而不是背答案的任务。毕竟面试官想看到的,从来不是一台背题机器,而是一个能解决问题的人。
牛客51274894...:意思是光刷力扣还不够卷
AI时代还有必要刷lee...
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务