解题思路:
lca+dfs。
注意事项:
参考代码:
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 | #include <bits/stdc++.h>using namespace std;const int N = 1e5 + 10;int n,q;vector<int> t1[N];int type[N]; int a[N][23];int fa[N][21];int dep[N];void dfs(int v,int f){ fa[v][0] = f; dep[v] = dep[f] + 1; for(int i=1;i<=23;i++){ a[v][i] = a[f][i]; a[v][type[v]]++; } for(int k=1;(1 << k) <= dep[v] - 1;k++){ fa[v][k] = fa[fa[v][k-1]][k-1]; } for(int i = 0;i<t1[v].size();i++){ int u = t1[v][i]; if( u != f) dfs(u,v); }} int lca(int u,int v){ if(dep[u] < dep[v]){ swap(u,v); } for(int k=20;k>=0;k--){ if(dep[v]+(1 << k) <= dep[u]){ u = fa[u][k]; } } if(u == v) return u; for(int k=20;k>=0;k--){ if(fa[u][k] != fa[v][k]){ u = fa[u][k],v = fa[v][k]; } } return fa[v][0]; }int main(){ //记住这两个优化输入输出的方法。 ios::sync_with_stdio(false); cin.tie(0); cin >> n >> q; int tmp = 0; for(int i=1;i<=n;i++){ cin >> tmp; type[i] = tmp; } for(int k=0;k<n-1;k++){ int i,j; cin >> i >>j; t1[i].push_back(j); t1[j].push_back(i); } dfs(1,0); int begin,end; while(q--){ cin >> begin >> end; int fa = lca(begin,end); int ans = 0; for(int i=1;i<=22;i++){ int val = a[begin][i] + a[end][i] - 2*a[fa][i] + (i == type[fa]); if(val > 0) ans++; } cout << ans << endl; } return 0;} |
0 分
0 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程，并配有习题和答疑，点击了解：
一点编程也不会写的：零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码？手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程：算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复