标签：return 剖分 int 题解 void dep Loj139 inline sum
\(\text{Solution:}\)
先不考虑换根，那其他的就很板子。
考虑换根的影响，首先对在路径上的操作没有影响，唯一有影响的就是子树操作。
考虑直接先在 \(root=1\) 的情况下树剖，然后直接在上面分类讨论。
如果当前根就是要修改的点，那就直接全局修改；
如果当前根在我要修改的子树之外，那就直接修改这个子树；
否则，我们画图可以发现，把根转上去之后，\(u\) 的子树就是全树去掉 \((u,root)\) 路径上深度最浅的点的子树。
由于加法有逆元，所以直接减回去即可。
总复杂度 \(O(n\log^2 n),\) 找点需要再来一个倍增。
#include<bits/stdc++.h>
using namespace std;
typedef double db;
#define int long long
const int mod=1e9+7;
const db eps=1e-14;
inline int Max(int x,int y){return x>y?x:y;}
inline int Min(int x,int y){return x<y?x:y;}
inline db Max(db x,db y){return x-y>eps?x:y;}
inline db Min(db x,db y){return x-y<eps?x:y;}
inline int Add(int x,int y,int M=mod){return (x+y)%M;}
inline int Mul(int x,int y,int M=mod){return 1ll*x*y%M;}
inline int Dec(int x,int y,int M=mod){return (x-y+M)%M;}
inline int Abs(int x){return x<0?-x:x;}
inline int read(){
int s=0,w=1;
char ch=getchar();
while(!isdigit(ch)){if(ch=='-')w=-1;ch=getchar();}
while(isdigit(ch)){s=s*10+ch-'0';ch=getchar();}
return s*w;
}
inline void write(int x){
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
}
inline int qpow(int x,int y){
int res=1;
while(y){
if(y&1)res=Mul(res,x);
x=Mul(x,x);y>>=1;
}
return res;
}
typedef pair<int,int> pr;
#define fi first
#define se second
#define mk make_pair
#define pb emplace_back
#define poly vector<int>
const int N=2e5+10;
namespace SGT{
int ls[N],rs[N],node,tag[N],sum[N];
inline void pushup(int x){sum[x]=sum[ls[x]]+sum[rs[x]];}
inline void pushdown(int x,int l,int r){
if(tag[x]!=0){
int mid=(l+r)>>1;
int p=tag[x];
tag[ls[x]]+=p;
tag[rs[x]]+=p;
sum[ls[x]]+=p*(mid-l+1);
sum[rs[x]]+=p*(r-mid);
tag[x]=0;
}
}
void change(int x,int L,int R,int l,int r,int v){
if(L>=l&&R<=r){
tag[x]+=v;
sum[x]+=(R-L+1)*v;
return;
}
pushdown(x,L,R);
int mid=(L+R)>>1;
if(l<=mid)change(ls[x],L,mid,l,r,v);
if(mid<r)change(rs[x],mid+1,R,l,r,v);
pushup(x);
}
int query(int x,int L,int R,int l,int r){
if(L>=l&&R<=r)return sum[x];
int mid=(L+R)>>1,ans=0;
pushdown(x,L,R);
if(l<=mid)ans=query(ls[x],L,mid,l,r);
if(mid<r)ans+=query(rs[x],mid+1,R,l,r);
return ans;
}
}
namespace Refined_heart{
using namespace SGT;
const int SN=21;
int a[N],pa[N],n,m,dfstime,rt;
int dep[N],top[N],rk[N],id[N];
int head[N],tot,son[N],siz[N];
struct E{int nxt,to;}e[N];
int root,f[N][SN];
void build(int &x,int l,int r){
x=++node;tag[x]=0;
if(l==r){
sum[x]=a[rk[l]];
return;
}
int mid=(l+r)>>1;
build(ls[x],l,mid);
build(rs[x],mid+1,r);
pushup(x);
}
inline void link(int x,int y){
e[++tot]=(E){head[x],y};
head[x]=tot;
}
void dfs1(int x,int fa){
dep[x]=dep[fa]+1;
siz[x]=1;pa[x]=fa;
f[x][0]=fa;
for(int i=1;i<SN;++i)f[x][i]=f[f[x][i-1]][i-1];
for(int i=head[x];i;i=e[i].nxt){
int j=e[i].to;
if(j==fa)continue;
dfs1(j,x);
siz[x]+=siz[j];
if(siz[j]>siz[son[x]])son[x]=j;
}
}
void dfs2(int u,int t){
top[u]=t;
rk[id[u]=++dfstime]=u;
if(!son[u])return;
dfs2(son[u],t);
for(int i=head[u];i;i=e[i].nxt){
int j=e[i].to;
if(j==pa[u]||j==son[u])continue;
dfs2(j,j);
}
}
int LCA(int x,int y){
while(top[x]!=top[y]){
if(dep[x]<dep[y])swap(x,y);
x=pa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
return x;
}
int findpoint(int ac,int now){
for(int i=SN-1;~i;--i)if(dep[f[now][i]]>dep[ac])now=f[now][i];
return now;
}
void chain(int x,int y,int opt,int v=0){
int ans=0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
if(opt){
ans+=query(rt,1,n,id[top[x]],id[x]);
x=pa[top[x]];
}
else{
change(rt,1,n,id[top[x]],id[x],v);
x=pa[top[x]];
}
}
if(dep[x]<dep[y])swap(x,y);
if(opt)ans+=query(rt,1,n,id[y],id[x]),write(ans),putchar('\n');
else change(rt,1,n,id[y],id[x],v);
}
void solve(){
n=read();
for(int i=1;i<=n;++i)a[i]=read();
for(int i=2;i<=n;++i)pa[i]=read();
for(int i=1;i<=n;++i)link(pa[i],i),link(i,pa[i]);
dfs1(1,0);dfs2(1,1);build(rt,1,n);
m=read();root=1;
for(int i=1;i<=m;++i){
int opt=read();
if(opt==1){
int u=read();
root=u;
}
if(opt==2){
int u=read(),v=read(),val=read();
chain(u,v,0,val);
}
if(opt==3){
int u=read(),k=read();
if(u!=root&&LCA(u,root)==u){
change(rt,1,n,1,n,k);
int pos=findpoint(u,root);
change(rt,1,n,id[pos],id[pos]+siz[pos]-1,-k);
}
else if(u==root)change(rt,1,n,1,n,k);
else change(rt,1,n,id[u],id[u]+siz[u]-1,k);
}
if(opt==4){
int u=read(),v=read();
chain(u,v,1);
}
if(opt==5){
int u=read();int ans=0;
if(u!=root&&LCA(u,root)==u){
ans+=query(rt,1,n,1,n);
int pos=findpoint(u,root);
ans-=query(rt,1,n,id[pos],id[pos]+siz[pos]-1);
}
else if(u==root)ans=query(rt,1,n,1,n);
else ans=query(rt,1,n,id[u],id[u]+siz[u]-1);
write(ans);putchar('\n');
}
}
}
}
signed main(){
// freopen("in.txt","r",stdin);
Refined_heart::solve();
return 0;
}
标签：return,剖分,int,题解,void,dep,Loj139,inline,sum 来源： https://www.cnblogs.com/h-lka/p/15481985.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。