傳送門

發(fā)現(xiàn)其實(shí)就是左右2棵子樹，左兒子選到某個(gè)值的概率就是
$選最大值的概率?右兒子的值比它小的概率$
$+選最小值的概率?右兒子的值比他大的概率$

右兒子類似

由于保證所有值不都相等
然后上線段樹合并$merge$的時(shí)候$dp$一下就完了

復(fù)雜度$O(nlogn)$

#include<bits/stdc++.h> using namespace std; const int RLEN=1<<20|1; inline char gc(){static char ibuf[RLEN],*ib,*ob;(ib==ob)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));return (ib==ob)?EOF:*ib++; } inline int read(){char ch=gc();int res=0,f=1;while(!isdigit(ch))f^=ch=='-',ch=gc();while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=gc();return f?res:-res; } const int mod=998244353; const int N=500005; inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b; } inline void selfadd(int &a,int b){a=add(a,b); } inline int dec(int a,int b){return a>=b?a-b:a-b+mod; } inline int mul(int a,int b){return 1ll*a*b>=mod?1ll*a*b%mod:a*b; } inline void selfmul(int &a,int b){a=mul(a,b); } inline int ksm(int a,int b,int res=1){for(;b;b>>=1,a=mul(a,a))if(b&1)selfmul(res,a);return res; } int cnt[N],son[N][2],rt[N],val[N],pmx[N]; int fa[N],ori[N],ans,n; struct node{int k,p;friend inline bool operator <(const node &a,const node &b){return a.k<b.k;} }q[N];int tot; namespace Seg{#define mid ((l+r)>>1)int tot,cnt,mxa,mxb;int lc[N*22],rc[N*22],tag[N*22],tr[N*22];void update(int &u,int l,int r,int k){u=++tot,tag[u]=1,tr[u]=1;if(l==r)return;if(k<=mid)update(lc[u],l,mid,k);else update(rc[u],mid+1,r,k);}inline void pushup(int u){tr[u]=add(tr[lc[u]],tr[rc[u]]);}inline void pushnow(int u,int t){selfmul(tag[u],t);selfmul(tr[u],t);}inline void pushdown(int u){if(tag[u]==1)return;int &t=tag[u];if(lc[u])pushnow(lc[u],t);if(rc[u])pushnow(rc[u],t);t=1;}int merge(int r1,int r2,int g){if(!r1&&!r2)return 0;pushdown(r1),pushdown(r2);if(!r1){selfadd(mxb,tr[r2]);pushnow(r2,dec(add(mxa,g),mul(2,mul(mxa,g))));return r2;}if(!r2){selfadd(mxa,tr[r1]);pushnow(r1,dec(add(mxb,g),mul(2,mul(mxb,g))));;return r1;}rc[r1]=merge(rc[r1],rc[r2],g);lc[r1]=merge(lc[r1],lc[r2],g);pushup(r1);return r1;}void dfs(int u){if(!::cnt[u])return;else if(::cnt[u]==1)dfs(son[u][0]),rt[u]=rt[son[u][0]];else if(::cnt[u]==2){dfs(son[u][0]),dfs(son[u][1]);mxa=mxb=0;rt[u]=merge(rt[son[u][0]],rt[son[u][1]],pmx[u]);}}void calc(int u,int l,int r){if(!tr[u])return;pushdown(u);if(l==r){cnt++;selfadd(ans,mul(mul(mul(cnt,ori[l]),tr[u]),tr[u]));return;}calc(lc[u],l,mid),calc(rc[u],mid+1,r);} } signed main(){n=read();for(int i=1;i<=n;i++){fa[i]=read();son[fa[i]][cnt[fa[i]]++]=i;}int inv=ksm(10000,mod-2);for(int i=1;i<=n;i++){int k=read();if(cnt[i])pmx[i]=mul(k,inv);else{q[++tot]=(node){k,i};}}sort(q+1,q+tot+1);for(int i=1;i<=tot;i++){ori[i]=q[i].k,Seg::update(rt[q[i].p],1,tot,i);}Seg::dfs(1);Seg::calc(rt[1],1,tot);cout<<ans; }

轉(zhuǎn)載于:https://www.cnblogs.com/stargazer-cyk/p/11145564.html

《新程序員》：云原生和全面數(shù)字化實(shí)踐50位技術(shù)專家共同創(chuàng)作，文字、視頻、音頻交互閱讀

總結(jié)

以上是生活随笔為你收集整理的【BZOJ5461】 【PKUWC2018】—Minimax（线段树合并优化dp）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。