传送门
Solution
发现叶子节点的值都不样,所以可以线段树合并。
然后因为我们要维护一个后缀,所以我们先合并右儿子,在合并左儿子
Code?
//2019.1.14 8:59~10:15 PaperCloud
#include<bits/stdc++.h>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
return x*f;
}
#define MN 300005
#define mod 998244353
int const INV_1e4=796898467;
int N,c[MN][3],fa[MN],p[MN];
struct Node
{
int val,id;
bool operator<(const Node&o)const {return val<o.val;}
}a[MN];int tot;
struct node{int ls,rs,lazy,p;}t[MN*15];int pin,root[MN];
#define mid ((l+r)>>1)
inline void Modify(int &x,int l,int r,int val)
{
if(!x) x=++pin;t[x].lazy=t[x].p=1ll;if(l==r) return;
val<=mid?Modify(t[x].ls,l,mid,val):Modify(t[x].rs,mid+1,r,val);
}
ll L,R,P;
inline void pushdown(int x)
{
if(t[x].lazy==1) return;
t[x].p=1ll*t[x].p*t[x].lazy%mod;
t[t[x].ls].lazy=1ll*t[t[x].ls].lazy*t[x].lazy%mod;
t[t[x].rs].lazy=1ll*t[t[x].rs].lazy*t[x].lazy%mod;
t[x].lazy=1;
}
inline void pushup(int x){t[x].p=(t[t[x].ls].p+t[t[x].rs].p)%mod;}
inline int Merge(int x,int y)
{
// printf("%d %d\n",x,y);
if(!x&&!y)return 0;
pushdown(x),pushdown(y);
if(!y)
{
(L+=t[x].p)%=mod,t[x].lazy=1ll*t[x].lazy*(R+P-2ll*R*P%mod+mod)%mod;
pushdown(x);return x;
}
if(!x)
{
(R+=t[y].p)%=mod,t[y].lazy=1ll*t[y].lazy*(L+P-2ll*L*P%mod+mod)%mod;
pushdown(y);return y;
}
t[x].rs=Merge(t[x].rs,t[y].rs);t[x].ls=Merge(t[x].ls,t[y].ls);pushup(x);
return x;
}
inline void dfs(int x=1)
{
if(!c[x][0]) return;
else if(c[x][0]==1) dfs(c[x][1]),root[x]=root[c[x][1]];
else
{
dfs(c[x][1]),dfs(c[x][2]);L=R=0;P=p[x];
root[x]=Merge(root[c[x][1]],root[c[x][2]]);
}
}
inline void calc(int x=root[1],int l=1,int r=tot)
{
static ll ans=0;pushdown(x);
if(l==r)
{
(ans+=(1ll*l*a[l].val%mod*t[x].p%mod*t[x].p%mod)%mod)%=mod;
if(l==tot) printf("%lld\n",ans);
return;
}
calc(t[x].ls,l,mid);calc(t[x].rs,mid+1,r);
}
int main()
{
// freopen("1.in","r",stdin);
N=read();register int i;
for(i=1;i<=N;++i) fa[i]=read(),c[fa[i]][++c[fa[i]][0]]=i;
for(i=1;i<=N;++i) if(c[i][0]>0) p[i]=1ll*read()*INV_1e4%mod; else a[++tot]=(Node){read(),i};
std::sort(a+1,a+tot+1);
for(i=1;i<=tot;++i) p[a[i].id]=i,Modify(root[a[i].id],1,tot,i);
dfs();calc();return 0;
}
Blog来自PaperCloud,未经允许,请勿转载,TKS!
原文地址:https://www.cnblogs.com/PaperCloud/p/10275295.html
时间: 2024-10-30 01:09:03