题目描述
小A
有一棵
N
N
N 个点的树,每个点都有一个小于
2
20
2^{20}
220 的非负整数权值。现在小A
从树中随机选择一个点
x
x
x ,再随机选择一个点
y
y
y (
x
,
y
x,y
x,y 可以是同一个点),并对从
x
x
x 到
y
y
y 的路径上所有的点的权值分别做
and
\text{and}
and 、
or
\text{or}
or 、
xor
\text{xor}
xor 运算,最终会求得三个整数。小A
想知道,他求出的三个数的期望值分别是多少。
题解
考虑拆位,即我们考虑一棵点权为 0 / 1 0/1 0/1 的树,操作后为 1 1 1 的链有多少条。
于是做个 dp \text{dp} dp : f [ i ] [ 0 / 1 ] f[i][0/1] f[i][0/1] 表示 i i i 子树内,到 i i i 经操作后为 0 / 1 0/1 0/1 的链有多少条,每次 s o n son son 合并上来的时候计算答案即可。
效率: O ( 20 n ) O(20n) O(20n) 。
代码
#include <bits/stdc++.h>
#define db double
using namespace std;
const int N=1e5+5;db k,s[3];
int T,n,hd[N],V[N<<1],nx[N<<1],t,f[3][2][N],a[N],b[N];
void add(int u,int v){
nx[++t]=hd[u];V[hd[u]=t]=v;
}
void dfs(int u,int fr){
for (int j=0;j<3;j++)
f[j][b[u]][u]=1,f[j][!b[u]][u]=0;
for (int v,i=hd[u];i;i=nx[i]){
if ((v=V[i])==fr) continue;dfs(v,u);
s[0]+=k*f[0][1][u]*f[0][1][v];
s[1]+=k*f[1][1][u]*f[1][1][v]+k*f[1][1][u]*f[1][0][v]+k*f[1][0][u]*f[1][1][v];
s[2]+=k*f[2][0][u]*f[2][1][v]+k*f[2][1][u]*f[2][0][v];
for (int j=0;j<2;j++)
f[0][j&b[u]][u]+=f[0][j][v],
f[1][j|b[u]][u]+=f[1][j][v],
f[2][j^b[u]][u]+=f[2][j][v];
}
}
void work(){
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&a[i]);
for (int i=1,u,v;i<n;i++)
scanf("%d%d",&u,&v),
add(u,v),add(v,u);
k=1;s[0]=s[1]=s[2]=0;
for (int i=0;i<20;i++){
for (int j=1;j<=n;j++){
b[j]=(a[j]>>i)&1;
if (b[j]) s[0]+=k,s[1]+=k,s[2]+=k;
}
k*=2;dfs(1,0);
}
for (int i=0;i<3;i++)
printf("%.3lf%c",s[i]/(1.*n*n),i<2?' ':'\n');
for (int i=1;i<=n;i++) hd[i]=0;t=0;
}
int main(){for (cin>>T;T--;work());return 0;}