二叉树

题目描述

九条可怜现在有一棵 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 有关：

1. 如果 u 是 v 的祖先，那么 u 一定在 v 前面
2. 如果最近公共祖先 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