题目描述
九条可怜现在有一棵 n 个节点二叉树,根节点编号为 1 ,保证这棵二叉树的所有非叶子节点都同时有左儿子和右儿子。 这天可怜闲来无聊,于是她对这棵树做了一些修改。假设这棵树有 m 个非叶子节点,其中 ai 是编号第 i 小的非叶子节点的编号。可怜首先投了 m 枚均匀的硬币(每一枚硬币结果为正反面的概率均为 ),她按照 1 到 m 的顺序观察了每一枚硬币的结果,如果第 i 枚硬币的结果是正面,则可怜会交换点 ai 的左右儿子,否则可怜什么都不做。 为了让事情变得更有趣,可怜在操作开始前问了 q 个问题:你猜在操作结束后,这棵树的先序遍历上,ui 是排在 vi 的前面还是后面呢? 作为一名新时代的优秀青年,为了最大化猜对的概率,你打算对每一个问题,先计算在操作结束后,ui 排在 vi 前面的概率,再告诉可怜你的选择。
输入描述:
第一行输入两个整数 n,q 表示节点个数和问题个数。接下来 n 行,第 i 行描述第 i 个点的儿子。每一行第一个整数 ki(ki ∈ {0,2}) 表示 i 号点的儿子个数,如果 ki=2,接下来两个整数 li,ri(1 ≤ li,ri ≤ n) 分别表示 i 的左儿子和右儿子;如果 ki=0,则这一行仅包含 ki 一个数。接下来 q 行每行两个整数 ui,vi (1 ≤ ui ≠ vi ≤ n) ,描述了一个问题。对于 的数据,n,q ≤ 40。对于 的数据,n,q≤ 5000。对于 的数据,1 ≤ n,q ≤ 105。
输出描述:
对于每一个问题输出一个实数,表示操作结束后在先序遍历中 ui 排在 vi 前面的概率。四舍五入保留三位小数。
示例1
输入
5 3
2 2 4
0
0
2 3 5
0
2 1
1 5
3 5
输出
0.000
1.000
0.500
思路
两个点 u 和 v 在先序遍历中的顺序只和 lca 有关:
- 如果 u 是 v 的祖先,那么 u 一定在 v 前面
- 如果最近公共祖先 f ≠u,v,那么哪个在左子树哪个就先
因此随机交换中,对于第一种情况,不改变 u 和 v 的顺序,第二种情况,各 有 50% 的概率改变/不改变
因此只要判祖先关系就可以了
判祖先关系可以用 DFS 序。
代码
#include<bits/stdc++.h>
using namespace std;
#define N 100005
int f[N][18],lc[N],rc[N],de[N],n,i,j,cmd,m,u,v,lg,ro;
void dfs(int x,int d)
{
if(lc[x])
dfs(lc[x],d+1),dfs(rc[x],d+1);
de[x]=d;
}
inline int dt(int x,int d)
{
for(j=lg;j>=0;j--)
if(de[f[x][j]]>=d)
x=f[x][j];
return x;
}
int main(){
for(scanf("%d%d",&n,&m),i=1;i<=n;i++)
{
scanf("%d",&cmd);
if(cmd)
scanf("%d%d",&lc[i],&rc[i]),f[lc[i]][0]=f[rc[i]][0]=i;
}
for(i=1;(1<<i)<=n;i++)
for(j=1;j<=n;j++)
{
f[j][i]=f[f[j][i-1]][i-1];
if(f[j][0]==0)
ro=j;
}
dfs(ro,0);de[0]=-1;
for(lg=i,i=0;i<m;i++)
{
scanf("%d%d",&u,&v);
if(de[u]<de[v])
printf(((dt(v,de[u])==u)?"1.000\n":"0.500\n"));
else
printf(((dt(u,de[v])==v)?"0.000\n":"0.500\n"));
}
return 0;
}
来源;nkw