LOJ #6509. 「雅礼集训 2018 Day7」C

神仙题

LOJ #6509

---

题意

给定一棵树，点权为0/1，每次随机一个点（可能和之前所在点相同）走到该点并将其点权异或上1

求期望的移动距离使得所有点点权相同

---

题解

根本不会解方程

容易发现如果一个点不是最后一次被走到，就会随机下一个点并走过去

即如果我们能求出每个点非最后一次走到的期望次数，就可以算出答案

由于完全随机，初始相同颜色的点非最后一次走到的次数相同

设$ f_{i,0/1}$表示在有$ i$个1的时候，0/1非最后一次走到的期望次数

很艰难的列出方程如下

$$ f_{i,0} = \frac{i}{n} f_{i-1, 0} + \frac{n-i-1}{n} f_{i+1, 0} + \frac{1}{n} f_{i+1,1} + \frac{1}{n} $$
$$ f_{i,1} = \frac{n-i}{n} f_{i+1, 1} + \frac{i-1}{n} f_{i-1, 1} + \frac{1}{n} f_{i-1,0} + \frac{1}{n}$$

注意由于记录的是非最后一次走到，需要特判边界

即当前已经有$ n-1$个黑色时将最后一个白点翻成黑点不算入期望次数，黑点翻成白点同理

然后可以暴力高消了

但这个方程有很多优美的性质

不容易发现$ f_{i+1,1}$只和$ f_{i,1}$和$ f_{i-1,0/1}$有关

因此如果知道前面的信息就可以推出$ f_{i+1,1}$

同理再借助$ f_{i+1,1}$就能推出$ f_{i+1,0}$

因此我们设$ f_{1,0}=x,f_{1,1}=y$并把所有$ f_{x,0/1}$用$ Ax+By+c$表示

然后在最后联立两个方程解出$ x,y$就做完了

边界细节真**多

---

代码

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define p 1000000007
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x=;char zf=;char ch=getchar();
    while(ch!='-'&&!isdigit(ch))ch=getchar();
    if(ch=='-')zf=-,ch=getchar();
    while(isdigit(ch))x=x*+ch-'',ch=getchar();return x*zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int k,m,n,x,y,z,cnt;
struct fc{
    ll a,b,c;//ax+by+c
    fc operator +(const fc s)const{return {a+s.a,b+s.b,c+s.c};}
    fc operator -(const fc s)const{return {a-s.a,b-s.b,c-s.c};}
    fc operator *(const int s)const{return {a*s%p,b*s%p,c*s%p};}
    fc operator +(const int s)const{return {a,b,c+s};}
    fc operator -(const int s)const{return {a,b,c-s};}
}f[][];
int inv[],fa[],size[];ll ans[];
char c[];
vector<int>E[];
int ksm(int x,int y=p-){
    x%=p;
    while(x<)x+=p;
    int ans=;
    for(rt i=y;i;i>>=,x=1ll*x*x%p)if(i&)ans=1ll*ans*x%p;
    return ans;
}
void dfs(int x){
    size[x]=;
    for(auto i:E[x]){
        dfs(i);
        size[x]+=size[i];
        ans[x]+=ans[i]+size[i];
    }
}
void dfs2(int x){
    if(x!=)ans[x]=ans[fa[x]]+n-size[x]*;
    for(auto i:E[x])dfs2(i);
}
int main(){
    n=read();scanf("%s",c+);int sum=;
    for(rt i=;i<=n;i++)sum+=(c[i]=='');
    inv[]=inv[]=;
    for(rt i=;i<=n;i++)inv[i]=1ll*inv[p%i]*(p-p/i)%p;
    f[][].a=;f[][].b=;
    for(rt i=;i<=n-;i++){
        if(i==)f[i+][]=(f[i][]*n-f[i-][]*(i-)-f[i-][])*inv[n-i];
        else f[i+][]=(f[i][]*n-f[i-][]*(i-)-f[i-][]-)*inv[n-i];
        f[i+][]=(f[i][]*n-f[i-][]*i-f[i+][]-)*inv[n-i-];
    }
    fc x=f[n][];
    fc a0=f[n-][]*(n-)-f[n-][]*n,a1=f[n-][]*(n-)-f[n-][]*n+f[n-][]+;
    ll a=a0.a,b=a0.b,C=a0.c,d=a1.a,e=a1.b,F=a1.c;
    int invd=ksm(d);
    int Y=(F%p*invd%p*a%p-C)%p*ksm(b%p-e%p*invd%p*a%p)%p;
    int X=(b%p*Y%p+C%p)*ksm(a)%p;X=-X;
    int B=(f[sum][].a*X+f[sum][].b*Y+f[sum][].c)%p+inv[n];
    int W=(f[sum][].a*X+f[sum][].b*Y+f[sum][].c)%p+inv[n];
    B%=p;W%=p;B=1ll*B*inv[n]%p,W=1ll*W*inv[n]%p;
    for(rt i=;i<=n;i++){
        fa[i]=read();
        E[fa[i]].push_back(i);
    }
    dfs();dfs2();ll ret=;
    for(rt i=;i<=n;i++)if(c[i]=='')(ret+=1ll*B*(ans[i]%p)%p)%=p;
    else (ret+=1ll*W*(ans[i]%p)%p)%=p;
    cout<<(ret+p)%p;
    return ;
}