CodeChef
来源:互联网 发布:java飞机大战扇形子弹 编辑:程序博客网 时间:2024/05/16 19:17
CodeChef - ANCESTOR Ancestors in Two Trees
数据结构,dfs序
题意
给你两颗树,都是n个节点,1是根。问你对于2~n的所有点,在两颗树中的共同祖先有几个。
思路
dfs序的应用吧。其实求出两个树的dfs序,就是公共区间覆盖的问题。
具体做法是,求出第一棵树的dfs序,然后对第二颗树dfs,每到一个点u,就在线段树或树状数组中记录这个编号在第一棵树的节点所掌握的区间。若记为L、R,就是在线段树里对L加1,对R减一。然后查询1到u的和,就是公共祖先个数。dfs完在复原。
因为如果第二课树的u节点在第一棵树里面的区间与第二颗树的区间无交集,那么前缀和的1、-1会抵消。
%潘学姐
代码
#include<cstdio>#include<cstring>#include<vector>#include<algorithm>#define M(a,b) memset(a,b,sizeof(a))#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1using namespace std;const int MAXN=500007;const int oo=0x3f3f3f3f;int sum[MAXN<<2];void pushup(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; }void build(int l, int r, int rt){ sum[rt]=0; if(l==r) return; int m=(l+r)>>1; build(lson); build(rson); pushup(rt);}void update(int pos, int v, int l, int r, int rt){ if(l==r) { sum[rt]+=v; return; } int m=(l+r)>>1; if(pos<=m) update(pos, v, lson); else update(pos, v, rson); pushup(rt);}int query(int L, int R, int l, int r, int rt){ if(L<=l&&r<=R) return sum[rt]; int m=(l+r)>>1; int res=0; if(L<=m) res+=query(L, R, lson); if(m<R) res+=query(L, R, rson); return res;}struct Edge{ int to, ne; Edge(int _u, int _v) { to=_u, ne=_v; }};vector<Edge> G1, G2;int head1[MAXN], head2[MAXN];int in[MAXN], ou[MAXN],res[MAXN];int dfsnum;void dfs1(int u, int fa){ in[u]=++dfsnum; for(int i=head1[u];~i;i=G1[i].ne) if(G1[i].to!=fa) dfs1(G1[i].to, u); ou[u]=dfsnum;}void dfs2(int u, int fa, int n){ update(in[u], 1, 1, n, 1); update(ou[u]+1, -1, 1, n, 1); res[u]=query(1, in[u], 1, n, 1)-1; for(int i=head2[u];~i;i=G2[i].ne) if(G2[i].to!=fa) dfs2(G2[i].to, u, n); update(in[u], -1, 1, n, 1); update(ou[u]+1, 1, 1, n, 1);}int main(){ int T;scanf("%d", &T); while(T--) { M(head1, -1), M(head2, -1); G1.clear(), G2.clear();dfsnum=0; int n; scanf("%d", &n); for(int i=1;i<n;i++) { int u, v;scanf("%d%d", &u, &v); G1.push_back(Edge(v, head1[u])); head1[u]=G1.size()-1; G1.push_back(Edge(u, head1[v])); head1[v]=G1.size()-1; } for(int i=1;i<n;i++) { int u, v;scanf("%d%d", &u, &v); G2.push_back(Edge(v, head2[u])); head2[u]=G2.size()-1; G2.push_back(Edge(u, head2[v])); head2[v]=G2.size()-1; } build(1, ++n, 1); dfs1(1, -1);dfs2(1, -1, n); for(int i=1;i<n;i++) printf("%d%c", res[i], i==n-1 ? '\n' : ' '); } return 0;}
阅读全文
0 0
- CodeChef
- CodeChef
- CodeChef
- CodeChef
- (CodeChef
- CodeChef
- CodeChef
- CodeChef
- CodeChef
- CodeChef
- CodeChef
- CodeChef CIELQUIZ
- 【codechef除草】
- CODECHEF::FCTRL
- CodeChef A
- codechef Flooring
- codechef Wall
- codechef COINS
- 交换一维数组中最大、最小元素
- JS函数6
- Promise 是什么?
- jquery准确获取元素
- 16CF1--B-1002
- CodeChef
- php扩展开发入门
- Java中list集合为空或为null的区别
- Android 基于SurfaceView 开发示波器
- MemoryAnalyzer 调试内存泄漏
- 递归算法问题: 有一对兔子,从出生后第3个月起每个月都生一对兔子,小兔子长到第三个月后每个月又生一对兔子,假如兔子都不会死。问: 第20个月的兔子总对数为多少?(提示:先分析兔子的增长规律)
- 爬虫遇到的问题
- 深入理解IIS的多线程工作机制
- OpenCV—形态学运算定义与实现