题目链接
http://codeforces.com/contest/482/problem/E
题解
T2智商题T3大LCT题,我一个也不会= =
CF的标算好像是分块?反正现在LCT都普及了就用LCT好了。
首先算期望推个式子,易得答案为\(\sum_u a[u](sz[u]^2-\sum_{v\in son[u]} sz[v]^2)\) (\(sz\)为子树大小),令求和的那个东西等于\(f[u]\)
并且如果往一个\(u\)里新添一个儿子\(v\),增添后的子树大小是\(sz[v]\), 那么新增的答案是\(2a[u]sz[v](sz[u]-sz[v])\)
然后我们要支持换父亲,动态维护这个东西
后面的就是莽上一个LCT。
这个只能详见代码,解释一下代码里变量的含义
对于一个Splay结构体\(u\):
\(ans\): 这个点Splay的子树中所有点的\(f\)之和。Splay的根的\(ans\)就是要求的答案。
\(sz01\): 这个点所有虚子树的大小之和加上本身的\(1\)。(\(sz\)是原树中子树大小)
\(sz02\): 这个点本身所有虚子树的大小平方和,不包括本身。
\(ans0\): 这个点本身所有虚儿子的\(ans\)之和。
\(sz11\): 这个点所有实儿子和虚儿子的大小之和。
\(sum\): 这个点及其所有实儿子\(v\)的\(sz01[v]\times a[v]\)之和。
这些变量看起来有些匪夷所思,那么我们看下怎么维护。
首先,所有对虚儿子求和的变量(也就是名字里带0的三个)都在access时处理,pushup时无需处理。
现在考虑pushup那个函数,前两行处理的是\(sz11\)和\(sum\), 这个根据定义求就行,也没有什么好说的。
后四行是重点——\(ans\)的更新。
我们把\(ans[u]\)分了四部分统计。(选两个点\(x\)和\(y\)求\(a[lca(x,y)]\)之和)
第一部分: 两个点都选在\(u\)的祖先(splay左子树内),或都选在splay右子树内,或都选在\(u\)的同一棵虚子树内。
spl[u].ans = spl[ls].ans+spl[u].ans0+spl[rs].ans;
第二部分: 两个点选在\(u\)的两棵不同虚子树内。
spl[u].ans += (spl[u].sz01*spl[u].sz01-spl[u].sz02)*a[u];
第三部分: 两个点之一选在\(u\)的虚子树内,另一个点选在\(u\)的splay右子树内。这样LCA一定是\(u\).
spl[u].ans += 2ll*spl[u].sz01*spl[rs].sz11*a[u];
第四部分: 两个点之一选在\(u\)的祖先(splay左子树)或其虚子树内,另一个选在\(u\)的整个splay子树及子树内所有点的虚子树去掉\(u\)上方(splay左子树内及其虚子树)的部分,也就是\(u\)及其下方(splay右子树内)所有点及其虚子树内。
spl[u].ans += 2ll*spl[ls].sum*(spl[u].sz11-spl[ls].sz11);
于是就做完了。(第四部分确实有点复杂)
最后膜拜一发考场切此题的新初三巨佬
zjr nb!
代码
#include<cstdio>
#include<cstdlib>
#include<iostream>
#define llong long long
using namespace std;
void read(int &x)
{
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
x*=f;
}
const int N = 1e5;
struct SplayNode
{
int son[2],fa; llong sz01,sz11,sz02,sum,ans0,ans;
} spl[N+3];
int fa[N+3];
llong a[N+3];
int n,q;
bool isroot(int u) {return spl[spl[u].fa].son[0]!=u && spl[spl[u].fa].son[1]!=u;}
void pushup(int u)
{
int ls = spl[u].son[0],rs = spl[u].son[1];
spl[u].sz11 = spl[ls].sz11+spl[u].sz01+spl[rs].sz11;
spl[u].sum = spl[ls].sum+spl[rs].sum+spl[u].sz01*a[u];
spl[u].ans = spl[ls].ans+spl[u].ans0+spl[rs].ans;
spl[u].ans += (spl[u].sz01*spl[u].sz01-spl[u].sz02)*a[u];
spl[u].ans += 2ll*spl[u].sz01*spl[rs].sz11*a[u];
spl[u].ans += 2ll*spl[ls].sum*(spl[u].sz11-spl[ls].sz11);
}
void rotate(int u)
{
int x = spl[u].fa,y = spl[x].fa;
spl[u].fa = y;
if(!isroot(x)) {spl[y].son[x==spl[y].son[1]] = u;}
int dir = u==spl[x].son[0];
spl[x].son[dir^1] = spl[u].son[dir];
if(spl[u].son[dir]) {spl[spl[u].son[dir]].fa = x;}
spl[u].son[dir] = x; spl[x].fa = u;
pushup(x); pushup(u);
}
void splaynode(int u)
{
while(!isroot(u))
{
int x = spl[u].fa,y = spl[x].fa;
if(!isroot(x)) {x==spl[y].son[1] ^ u==spl[x].son[1] ? rotate(u) : rotate(x);}
rotate(u);
}
pushup(u);
}
void access(int u)
{
for(int i=0; u; i=u,u=spl[u].fa)
{
splaynode(u);
int ls = spl[u].son[0],rs = spl[u].son[1];
spl[u].sz01 += spl[rs].sz11;
spl[u].sz02 += spl[rs].sz11*spl[rs].sz11;
spl[u].ans0 += spl[rs].ans; //not ans0
rs = spl[u].son[1] = i;
spl[u].sz01 -= spl[rs].sz11;
spl[u].sz02 -= spl[rs].sz11*spl[rs].sz11;
spl[u].ans0 -= spl[rs].ans;
pushup(u);
}
}
void link(int u,int v)
{
access(v); splaynode(v);
access(u); splaynode(u); //in order to pushup
spl[u].sz01 += spl[v].sz11;
spl[u].sz02 += spl[v].sz11*spl[v].sz11;
spl[u].ans0 += spl[v].ans;
spl[v].fa = u;
pushup(u);
}
void cut(int u,int v)
{
access(u); splaynode(u);
splaynode(v); //v is in the virtual subtree of u
spl[u].sz01 -= spl[v].sz11;
spl[u].sz02 -= spl[v].sz11*spl[v].sz11;
spl[u].ans0 -= spl[v].ans;
spl[v].fa = 0;
pushup(u);
}
bool isanc(int u,int v) //if u is ancestor of v
{
access(v); splaynode(v);
splaynode(u);
if(!isroot(v)) return true;
return false;
}
void printans(llong x)
{
double ans = (double)x/(double)n/(double)n;
printf("%.12lf\n",ans);
}
int main()
{
scanf("%d",&n);
for(int i=2; i<=n; i++) scanf("%d",&fa[i]);
for(int i=1; i<=n; i++) scanf("%I64d",&a[i]);
for(int i=1; i<=n; i++) spl[i].sz11 = spl[i].sz01 = 1ll,spl[i].ans = spl[i].sum = a[i];
for(int i=2; i<=n; i++)
{
link(fa[i],i);
}
access(1); splaynode(1);
printans(spl[1].ans);
scanf("%d",&q);
for(int i=1; i<=q; i++)
{
char opt[5]; scanf("%s",opt+1);
if(opt[1]=='P')
{
int x,y; scanf("%d%d",&x,&y);
if(isanc(x,y)) {swap(x,y);}
cut(fa[x],x);
fa[x] = y;
link(fa[x],x);
access(1); splaynode(1);
printans(spl[1].ans);
}
else if(opt[1]=='V')
{
int x; llong y; scanf("%d%I64d",&x,&y);
access(x); splaynode(x);
a[x] = y;
pushup(x);
printans(spl[x].ans);
}
}
return 0;
}