文明

http://47.96.116.66/problem.php?cid=1177&pid=8

http://47.96.116.66/problem.php?cid=1180&pid=8

题解:LCA+线段树

来自:https://blog.csdn.net/kzn2683331518/article/details/88807519

#include <bits/stdc++.h>
#define ll long long
using namespace std;
namespace _{
    char buf[100000], *p1 = buf, *p2 = buf; bool rEOF = 1;//为0表示文件结尾
    inline char nc(){ return p1 == p2 && rEOF && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) ? (rEOF = 0, EOF) : *p1++; }
    template<class _T>
    inline void read(_T &num){
        char c = nc(), f = 1; num = 0;
        while (c<'0' || c>'9')c == '-' && (f = -1), c = nc();
        while (c >= '0'&&c <= '9')num = num * 10 + c - '0', c = nc();
        num *= f;
    }
    inline bool need(char &c){ return c >= 'a'&&c <= 'z' || c >= '0'&&c <= '9' || c >= 'A'&&c <= 'Z'; }//读入的字符范围
    inline void read_str(char *a){
        while ((*a = nc()) && !need(*a) && rEOF);   ++a;
        while ((*a = nc()) && need(*a) && rEOF)++a; --p1, *a = '\0';
    }
    template<class _T>
    inline void println(_T x){
        static int buf[30], len; len = 0; do buf[len++] = x % 10, x /= 10; while (x);
        while (len) putchar(buf[--len] + 48);
        putchar('\n');
    }
}using namespace _;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9+7;
const int MAXN = 500005;
 
int n, q, k;
int a[MAXN];
 
struct edge{
    int v,next;
}e[MAXN<<1];
int head[MAXN], cnt;
inline void add(int u,int v){
    e[cnt] = (edge){v,head[u]}, head[u] = cnt++;
}
 
int dep[MAXN], low[MAXN], dfn[MAXN], tot, fa[MAXN][20];
void dfs(int u,int f){
    dep[u] = dep[f] + 1, dfn[u] = ++tot, fa[u][0] = f;
    for(int i = 1;i<20;i++)fa[u][i] = fa[fa[u][i-1]][i-1];
    for(int i = head[u];~i;i = e[i].next){
        if(e[i].v == f)continue;
        dfs(e[i].v, u);
    }
    low[u] = tot;
}
inline int lca(int u,int v){
    if(dep[u]<dep[v])swap(u,v);
    for(int i = 0, del = dep[u]-dep[v];del;i++, del>>=1)
        if(del&1)u = fa[u][i];
    if(u == v)return u;
    for(int i = 19;i>=0;i--)
        if(fa[u][i]!=fa[v][i])u = fa[u][i], v = fa[v][i];
    return fa[u][0];
}
#define mid ((l+r)>>1)
struct Tree{
    int lazy, sum;
}T[MAXN<<2];
inline void pushdown(int u,int l,int r){
    if(~T[u].lazy){
        T[u<<1].lazy = T[u<<1|1].lazy = T[u].lazy;
        T[u<<1].sum = T[u].lazy*(mid-l+1);
        T[u<<1|1].sum = T[u].lazy*(r-mid);
        T[u].lazy = -1;
    }
}
void update(int u,int l,int r,int L,int R,int x){
    if(L<=l&&r<=R){
        T[u].lazy = x;
        T[u].sum = x*(r-l+1);
        return;
    }
    pushdown(u,l,r);
    if(L<=mid)update(u<<1,l,mid,L,R,x);
    if(R>mid)update(u<<1|1,mid+1,r,L,R,x);
    T[u].sum = T[u<<1].sum + T[u<<1|1].sum;
}
int main() {
    memset(head,-1,sizeof head);
    read(n), read(q);
    for(int i = 1,u,v;i<n;i++){
        read(u), read(v), add(u,v), add(v,u);
    }
    dfs(1,1);
    while(q--){
        read(k); for(int i = 1;i<=k;i++) read(a[i]);
        if(k == 1){
            println(n); continue;
        }
        T[1].lazy = 0;
        int rt = a[1];
        for(int i = 2;i<=k;i++){
            if(dfn[rt] <= dfn[a[i]] && dfn[a[i]] <= low[rt]){   //情况1,在rt的子树中
                int d = ((dep[a[i]] + dep[rt])>>1) + 1, u = a[i];
                for(int j = 19;j>=0;j--)
                    if(dep[fa[u][j]] >= d) u = fa[u][j];
                update(1,1,n,dfn[u],low[u],1);        //对u整个子树打标记
            }else{
                int lc = lca(rt, a[i]);
                int dis = ((dep[a[i]] + dep[rt] - (dep[lc]<<1))>>1)+1;
                if(dep[a[i]] >= dep[rt]){                        //情况2
                    int d = dis - dep[rt] + (dep[lc]<<1), u = a[i];
                    for(int j = 19;j>=0;j--)
                        if(dep[fa[u][j]] >= d) u = fa[u][j];
                    update(1,1,n,dfn[u],low[u],1);    //对u整个子树打标记
                }else{                                           //情况3
                    int d = dep[rt] - dis + 1, u = rt;    //d表示u的深度
                    for(int j = 19;j>=0;j--)
                        if(dep[fa[u][j]] >= d) u = fa[u][j];
                    if(1 <= dfn[u]-1)update(1,1,n,1,dfn[u]-1,1); //对u子树外的点打标记
                    if(low[u]+1 <= n)update(1,1,n,low[u]+1,n,1); //对u子树外的点打标记
                }
            }
        }
        println(n-T[1].sum);    //答案为总点数-被标记点数
    }
    return 0;
}

 

全部评论

相关推荐

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

创作者周榜

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