当前位置: 代码迷 >> 综合 >> CF276E Little Girl and Problem on Trees
  详细解决方案

CF276E Little Girl and Problem on Trees

热度:31   发布时间:2024-02-10 13:25:02.0

一、题目

点此看题

二、解法

这道题翻译有点问题,有一个性质没有提到:除 1 1 以外的点,每个点的度数最多是 2 2

对于一个修改,先考虑把当前点所在的这条链更新,对每条链的点重新编号即可。再考虑更新 1 1 连出去的其他链,这些链影响的深度都是一样的,所以再维护每一个深度的公共权值(注意要在原来的链删去这个权值)

思路不难,实则难写的一批,心态被写炸了。。。。

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 100005;
int read()
{int x=0,f=1;char c;while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}return x*f;
}
int n,m,cnt,tot,ans,f[M],dep[M],num[M],l[M],r[M];
int val[4*M][2];
struct edge
{int v,next;
}e[2*M];
void dfs(int u,int fa)
{num[u]=++cnt;dep[u]=dep[fa]+1;for(int i=f[u];i;i=e[i].next)if(e[i].v^fa)dfs(e[i].v,u);
}
void modify(int i,int l,int r,int L,int R,int v,int d)
{if(l>R || L>r) return ;if(L<=l && r<=R){val[i][d]+=v;return ;}int mid=(l+r)>>1;modify(i<<1,l,mid,L,R,v,d);modify(i<<1|1,mid+1,r,L,R,v,d);
}
void ask(int i,int l,int r,int id,int d)
{ans+=val[i][d];if(l==r) return ;int mid=(l+r)>>1;if(id<=mid) ask(i<<1,l,mid,id,d);else ask(i<<1|1,mid+1,r,id,d);
}
int main()
{n=read();m=read();for(int i=1;i<n;i++){int u=read(),v=read();e[++tot]=edge{v,f[u]},f[u]=tot;e[++tot]=edge{u,f[v]},f[v]=tot;}num[1]=++cnt;dep[1]=1;for(int i=f[1];i;i=e[i].next){int st=cnt+1;dfs(e[i].v,1);for(int j=st;j<=cnt;j++)l[j]=st,r[j]=cnt;}for(int i=1;i<=m;i++){int op=read(),v=read();if(op==0){int x=read(),d=read();if(v!=1) modify(1,1,n,max(l[num[v]],num[v]-d),min(r[num[v]],num[v]+d),x,0);if(dep[v]<=d+1){modify(1,1,n,1,d-dep[v]+2,x,1);if(dep[v]<=d) modify(1,1,n,l[num[v]],min(r[num[v]],l[num[v]]+d-dep[v]),-x,0);}}else{ans=0;ask(1,1,n,num[v],0);ask(1,1,n,dep[v],1);printf("%d\n",ans);}}
}
  相关解决方案