本文主要是介绍AtCoder Beginner Contest 351 G. Hash on Tree(树剖维护动态dp 口胡题解),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目
n(n<=2e5)个点,给定一个长为a的初始权值数组,
以1为根有根树,
树哈希值f计算如下:
(1)如果一个点u是叶子节点,则f[u]=a[u]
(2)否则,
q(q<=2e5)次点权修改,每次问修改之后根节点(点1)的树哈希值f
思路来源
qiuzx_代码
题解
新开一个口胡题解专栏,记录当前题目水平不在我能力范围之内
但是看别人的代码能看懂,自己写可能要调一万年的这种题,描述一下他是怎么写的
心得
动态 DP - OI Wiki
动态dp可以参考这篇,
基本是对于有根树树形dp,想动态获取根节点的dp值,dp值和树上点权相关,点权带修,
序列的动态dp,一般是线段树维护矩阵乘法(可广义换成其他运算)或线段树节点合并
树上版本,是树剖维护矩阵转移或者线段树节点
本题中,先树链剖分,求出重儿子,重链维护线段树
线段树上每个节点维护一个y=bx+a的点对(a,b),
其中,b是f值非零的轻儿子的f积,x是重儿子的f值(未知,待代入)
a是当前节点u的a[u]值 ,然后就能b1*(b2x+a2)+a1这么线段树往上合并了
递归到叶子的时候,由于叶子不存在重儿子,x值为0,
所以链头的f值,就是当线段树区间[l,r],
l为重链头的dfs序,r为叶子重链尾的dfs序时,这个区间节点对应的(a,b)中a的值
由于为0的值会对运算值有影响,并且儿子f值从一个0变成没有0的时候难以恢复之前的乘积,
所以单独记录0的个数,所以变更的时候,动态记录链头点的f值为0的儿子的数量
代码
qiuzx_代码
//ANMHLIJKTJIY!
#pragma GCC optimize(2)
#pragma GCC optimize("Ofast")
#pragma GCC optimize("inline","fast-math","unroll-loops","no-stack-protector")
#pragma GCC diagnostic error "-fwhole-program"
#pragma GCC diagnostic error "-fcse-skip-blocks"
#pragma GCC diagnostic error "-funsafe-loop-optimizations"
#include <bits/stdc++.h>
#define INF 1000000000
#define LINF 1000000000000000000
#define MOD 1000000007
#define mod 998244353
#define F first
#define S second
#define ll long long
#define N 200010
//#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
using namespace std;
char buf[1<<23],*p1=buf,*p2=buf;
ll rint(){ll ret=0;char c=getchar();while(!isdigit(c)) c=getchar();while(isdigit(c)) ret=ret*10+(c-'0'),c=getchar();return ret;
}
ll ksm(ll x,ll y)
{ll ret=1;while(y>0){if(y&1){ret=(ret*x)%mod;}x=(x*x)%mod;y>>=1;}return ret;
}
struct Node{ll a,b;Node(){a=b=0;}Node(ll _a,ll _b){a=_a,b=_b;}Node operator + (const Node &x)const{return Node((a*x.b+x.a)%mod,(b*x.b)%mod);}
};
struct SegT{ll lo[N<<2],hi[N<<2],pa[N],pb[N];Node val[N<<2];void build(ll x,ll l,ll r){lo[x]=l,hi[x]=r;if(l==r){val[x]=Node(pa[l],pb[l]);return;}ll mid=(l+r)>>1,a=x<<1;build(a,l,mid);build(a|1,mid+1,r);val[x]=val[a|1]+val[a];return;}void update(ll x,ll l,Node v){ll tl=lo[x],tr=hi[x];if(tl==tr){val[x]=v;return;}ll mid=(tl+tr)>>1,a=x<<1;if(mid>=l){update(a,l,v);}else{update(a|1,l,v);}val[x]=val[a|1]+val[a];return;}Node query(ll x,ll l,ll r){ll tl=lo[x],tr=hi[x];if(l<=tl&&tr<=r){return val[x];}ll mid=(tl+tr)>>1,a=x<<1;if(mid>=r){return query(a,l,r);}if(mid<l){return query(a|1,l,r);}return query(a|1,l,r)+query(a,l,r);}
}segt;
ll n,q,fa[N],a[N],b[N],f[N],num0[N],dfn[N],dcnt=0,tp[N],lw[N],sz[N],hson[N];
vector<ll> vt[N];
void predfs(ll x)
{ll i;sz[x]=1,hson[x]=-1;for(i=0;i<vt[x].size();i++){predfs(vt[x][i]);sz[x]+=sz[vt[x][i]];if(hson[x]==-1||sz[hson[x]]<sz[vt[x][i]]){hson[x]=vt[x][i];}}return;
}
void dfs(ll x,bool ist=true)
{tp[x]=ist?x:tp[fa[x]];dfn[x]=++dcnt;b[x]=1;if(hson[x]!=-1){dfs(hson[x],false);lw[x]=lw[hson[x]];f[x]=f[hson[x]];}else{lw[x]=x;f[x]=a[x];return;}ll i;for(i=0;i<vt[x].size();i++){if(vt[x][i]!=hson[x]){dfs(vt[x][i]);f[x]=(f[x]*f[vt[x][i]])%mod;if(f[vt[x][i]]){b[x]=(b[x]*f[vt[x][i]])%mod;}else{num0[x]++;}}}f[x]=(f[x]+a[x])%mod;return;
}
ll calc(ll x)
{Node cur=segt.query(1,dfn[x],dfn[lw[x]]);return cur.a;
}
int main(){ll i,x,y;n=rint(),q=rint();for(i=1;i<n;i++){fa[i]=rint()-1;vt[fa[i]].push_back(i);}fa[0]=-1;for(i=0;i<n;i++){a[i]=rint();}predfs(0);dfs(0);for(i=0;i<n;i++){segt.pa[dfn[i]]=a[i];segt.pb[dfn[i]]=num0[i]?0:b[i];}segt.build(1,1,n);while(q--){x=rint()-1,y=rint();a[x]=y;segt.update(1,dfn[x],Node(a[x],num0[x]?0:b[x]));while(true){ll pre=f[tp[x]];f[tp[x]]=calc(tp[x]);x=tp[x];if(fa[x]<0){break;}if(pre){b[fa[x]]=(b[fa[x]]*ksm(pre,mod-2))%mod;}else{num0[fa[x]]--;}if(f[x]){b[fa[x]]=(b[fa[x]]*f[x])%mod;}else{num0[fa[x]]++;}x=fa[x];segt.update(1,dfn[x],Node(a[x],num0[x]?0:b[x]));}printf("%lld\n",f[0]);}return 0;
}
这篇关于AtCoder Beginner Contest 351 G. Hash on Tree(树剖维护动态dp 口胡题解)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!