【51Nod1681】公共祖先
来源:互联网 发布:php数组赋值给js数组 编辑:程序博客网 时间:2024/06/06 01:49
有一个庞大的家族,共n人。已知这n个人的祖辈关系正好形成树形结构(即父亲向儿子连边)。
在另一个未知的平行宇宙,这n人的祖辈关系仍然是树形结构,但他们相互之间的关系却完全不同了,原来的祖先可能变成了后代,后代变成的同辈……
两个人的亲密度定义为在这两个平行宇宙有多少人一直是他们的公共祖先。
整个家族的亲密度定义为任意两个人亲密度的总和。
Input
第一行一个数n(1<=n<=100000)
接下来n-1行每行两个数x,y表示在第一个平行宇宙x是y的父亲。
接下来n-1行每行两个数x,y表示在第二个平行宇宙x是y的父亲。
Output
一个数,表示整个家族的亲密度。
Input示例
5
1 3
3 5
5 4
4 2
1 2
1 3
3 4
1 5
Output示例
6
题解
把问题考虑为每个点为祖先对答案的贡献,问题就转化成求某个点为根的子树在两颗树中相同的个数。
有关子树的问题,可以利用dfs序处理,先扫一棵树记录dfs序,在扫第二棵树,利用树状数组记录信息。
代码
#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<set>#include<ctime>#include<vector>#include<cmath>#include<algorithm>#include<map>#define mod 1000000007 #define ll long long #define inf 1e9using namespace std;inline int read(){ int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f;}int rt,tot,ret[100005],Next[100005],Head[100005];int tim,n,l[100005],r[100005],t[100005];bool vis[100005];ll ans;inline void ins(int u,int v){ ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;}void dfs(int u,int pre){ l[u]=++tim; for (int i=Head[u];i;i=Next[i]) { if (ret[i]==pre) continue; dfs(ret[i],u); } r[u]=tim;}inline int lowbit(int x){return x&-x;}void update(int x){ for (int i=x;i<=n;i+=lowbit(i)) t[i]++;}int query(int x){ int ans=0; for (int i=x;i;i-=lowbit(i)) ans+=t[i]; return ans;}void find(int u,int pre){ int now=query(r[u])-query(l[u]); for (int i=Head[u];i;i=Next[i]) { if (ret[i]==pre) continue; update(l[ret[i]]); find(ret[i],u); } now=query(r[u])-query(l[u])-now; ans+=(ll)(now-1)*now/2;}int main(){ n=read(); for (int i=1;i<n;i++) { int x=read(),y=read(); ins(x,y); vis[y]=1; } for (int i=1;i<=n;i++) { if (!vis[i]) rt=i; vis[i]=0; } dfs(rt,0); for (int i=1;i<=n;i++) Head[i]=0;tot=0; for (int i=1;i<n;i++) { int x=read(),y=read(); ins(x,y); vis[y]=1; } for (int i=1;i<=n;i++) if (!vis[i]){rt=i;break;} find(rt,0); printf("%lld",ans); return 0;}
阅读全文
0 0
- 【51Nod1681】公共祖先
- [杂题 树状数组] 51Nod1681 公共祖先
- DFS序+主席树 51Nod1681 公共祖先
- 51NOD1681 公共祖先 【dfs序+树状数组】
- 51nod1681公共祖先(dfs序+树状数组)
- 51Nod-1681-公共祖先
- 51nod 1681 公共祖先
- 51nod 1681公共祖先
- lca最小公共祖先祖先
- 最低公共祖先
- 最近公共祖先LCA
- 最近公共祖先(LCA)
- 最近公共祖先
- Lca 最近公共祖先
- 【最近公共祖先】Tree
- 最低公共祖先问题
- LCA----最近公共祖先
- 树最低公共祖先
- Wireshark和Fiddler分析Android中的TLS协议包数据(附带案例样本)
- 如何应用BIM技术加快项目进度?
- Https 与 iOS 信息安全
- 第十四章 多播与广播
- C语言之#define(宏定义命令)
- 【51Nod1681】公共祖先
- Frequent values RMQ
- 融云即时通讯云5大高级功能解决App运营诸多痛点
- 花一块钱----快速部署一个自己的静态网站(京东云服务器)
- 关于canvas、paint的相关方法汇总
- 扒完社交网络关系才明白,《权力的游戏》凭什么是神作
- 北斗卫星导航定位系统应用及发展前景
- 高仿饿了吗自定义数字加减
- 自学nginx(五): 使用fluentd搬运log到mysql