• 【集训DAY N】火车【LCA】【模拟】


    在这里插入图片描述

    思路:

    只要保证每个点都只被标记一次就行了
    于是我们用LCA加一些模拟来做

    c o d e code code

    #include
    #include
    
    #define ll long long
    #define re register
    
    using namespace std;
    
    const ll MAXN = 5e5 + 10;
    
    ll n, m, s, a[MAXN];
    ll head[MAXN], tot, f[MAXN][25], dep[MAXN];
    bool v[MAXN];
    
    struct node {
    	ll to, next;
    }b[MAXN * 2];
    
    inline void add(re ll x, re ll y) {
    	b[++ tot] = (node) { y, head[x] };
    	head[x] = tot;	
    }
    
    inline void dfs(re ll x, re ll fa) {
    	f[x][0] = fa;
    	dep[x] = dep[fa] + 1;
    	for(re ll i = head[x]; i; i = b[i].next) {
    		re ll y = b[i].to;
    		if(fa == y) continue;
    		dfs(y, x);
    	}
    }
    
    inline ll lca(re ll x, re ll y) {
    	if(dep[x] > dep[y]) swap(x, y);
    	re ll k = dep[y] - dep[x], j = 20;
    	while(k) {
    		re ll t = (1 << j);
    		if(k >= t) k -= t, y = f[y][j];
    		j --;
    	}
    	if(x == y) return x;
    	for(re ll i = 20; i >= 0; -- i)
    		if(f[x][i] != f[y][i]) x = f[x][i], y = f[y][i];
    	return f[x][0];
    }
    
    inline void do_(re ll x, re ll y) {
    	while(v[x] == 0 && x != y) {
    		v[x] = 1;
    		x = f[x][0];
    	}
    	v[y] = 1;
    }
    
    inline void do2_(re ll x, re ll y) {
    	while(x != y) {
    		re ll j = 0;
    		while(v[f[x][j]] == 1) {
    			++ j;
    			if(dep[f[x][j - 1]] <= dep[y]) return;
    		}
    		if(j == 0) {
    			x = f[x][0];
    			break;
    		}
    		x = f[x][j - 1];
    	}
    	while(v[x] == 0 && x != y) {
    		v[x] = 1;
    		x = f[x][0];
    	}
    	v[y] = 1;
    }
    
    int main() {
    	scanf("%lld%lld%lld", &n, &m, &s);
    	v[s] = 1;
    	for(re ll i = 1; i < n; ++ i) {
    		re ll x, y;
    		scanf("%lld%lld", &x, &y);
    		add(x, y), add(y, x);
    	}
    	dfs(1, 0);
    	for(re ll j = 1; j <= 20; ++ j)
    		for(re ll i = 1; i <= n; ++ i)
    			f[i][j] = f[f[i][j - 1]][j - 1];
    	for(re ll i = 1; i <= m; ++ i) scanf("%lld", &a[i]);
    	ll i = 1, ans = 0;
    	while(i <= m) {
    		if(v[a[i]]) {
    			i ++;
    			continue;
    		}
    		re ll k = lca(s, a[i]);
    		do_(a[i], k);
    		do2_(s, k);
    		ans += dep[a[i]] + dep[s] - dep[k] * 2;
    		s = a[i];
    	}
    	printf("%lld", ans);
    	return 0;
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
    • 42
    • 43
    • 44
    • 45
    • 46
    • 47
    • 48
    • 49
    • 50
    • 51
    • 52
    • 53
    • 54
    • 55
    • 56
    • 57
    • 58
    • 59
    • 60
    • 61
    • 62
    • 63
    • 64
    • 65
    • 66
    • 67
    • 68
    • 69
    • 70
    • 71
    • 72
    • 73
    • 74
    • 75
    • 76
    • 77
    • 78
    • 79
    • 80
    • 81
    • 82
    • 83
    • 84
    • 85
    • 86
    • 87
    • 88
    • 89
    • 90
    • 91
    • 92
    • 93
    • 94
    • 95
    • 96
    • 97
    • 98
    • 99
    • 100
    • 101
    • 102
    • 103
  • 相关阅读:
    ElasticSearch7.3学习(二十八)----聚合实战之电视案例
    Web渗透测试_目录遍历
    第15章 注销器实现原理与触发
    14015.xilinx-芯片手册阅读笔记
    pdp部分依赖图
    【RSocket】使用 RSocket (一)——建立连接
    将爱心代码设为电脑屏保,俘获少女芳心,还能假装黑客,在酷炫的界面中保护隐私
    Spring系列八:Spring AOP 和 AspectJ AOP 区别
    数据结构 —— 顺序表(超详细图解 & 接口函数实现)
    opencv-python图片转换、尺寸、传输
  • 原文地址:https://blog.csdn.net/liuziha/article/details/126372353