标签:
链剖即可。其实就是利用了链剖后子树都在一段连续的区间内所以可以做到O(logn)查询和修改。
线段树细节打错了。。要专心!肉眼差错都能找出一堆出来显然是不行的!。
#include<cstdio> #include<cstring> #include<cctype> #include<algorithm> using namespace std; #define rep(i,s,t) for(int i=s;i<=t;i++) #define dwn(i,s,t) for(int i=s;i>=t;i--) #define clr(x,c) memset(x,c,sizeof(x)) #define qwq(x) for(edge *o=head[x];o;o=o->next) #define lson l,mid,x<<1 #define rson mid+1,r,x<<1|1 #define ll long long int read(){ int x=0;char c=getchar(); while(!isdigit(c)) c=getchar(); while(isdigit(c)) x=x*10+c-‘0‘,c=getchar(); return x; } const int nmax=5e4+5; const int inf=0x7f7f7f7f; struct edge{ int to;edge *next; }; edge es[nmax<<1],*pt=es,*head[nmax]; void add(int u,int v){ pt->to=v;pt->next=head[u];head[u]=pt++; pt->to=u;pt->next=head[v];head[v]=pt++; } int size[nmax],son[nmax],idx[nmax],id[nmax],w[nmax],n,m; ll sm[nmax<<2],col[nmax<<2]; void dfs(int x,int fa){ size[x]=1; qwq(x) if(o->to!=fa){ dfs(o->to,x);size[x]+=size[o->to]; if(!son[x]||size[o->to]>size[son[x]]) son[x]=o->to; } } void DFS(int x){ id[++id[0]]=x;idx[x]=id[0]; if(son[x]) DFS(son[x]); qwq(x) if(!idx[o->to]) DFS(o->to); } void build(int l,int r,int x){ col[x]=-1; if(l==r) { sm[x]=w[id[l]];return ; } int mid=(l+r)>>1;build(lson);build(rson);sm[x]=sm[x<<1]+sm[x<<1|1]; } void pushdown(int x,int cnt){ if(col[x]!=-1){ if(col[x<<1]!=-1) col[x<<1]+=col[x];else col[x<<1]=col[x]; if(col[x<<1|1]!=-1) col[x<<1|1]+=col[x];else col[x<<1|1]=col[x]; sm[x<<1]+=col[x]*(cnt-(cnt>>1));sm[x<<1|1]+=col[x]*(cnt>>1); col[x]=-1; } } void update(int p,int add,int l,int r,int x){ if(l==r) { sm[x]+=add;return ; } pushdown(x,r-l+1); int mid=(l+r)>>1; p<=mid?update(p,add,lson):update(p,add,rson); sm[x]=sm[x<<1]+sm[x<<1|1]; } void Update(int tl,int tr,int add,int l,int r,int x){ if(tl<=l&&tr>=r) { if(col[x]!=-1) col[x]+=add;else col[x]=add; sm[x]+=(ll)add*(r-l+1);return ; } pushdown(x,r-l+1); int mid=(l+r)>>1; if(tl<=mid) Update(tl,tr,add,lson); if(tr>mid) Update(tl,tr,add,rson); sm[x]=sm[x<<1]+sm[x<<1|1]; } ll query(int tl,int tr,int l,int r,int x){ if(tl<=l&&tr>=r) return sm[x]; pushdown(x,r-l+1); int mid=(l+r)>>1;ll ans=0; if(tl<=mid) ans+=query(tl,tr,lson); if(tr>mid) ans+=query(tl,tr,rson); return ans; } void UPDATE(int u,int v,int d){ ll sm=query(idx[u],idx[u]+size[u]-1,1,n,1); if(sm>=(ll)v*size[u]) return ; Update(idx[u],idx[u]+size[u]-1,d,1,n,1); } void print(int l,int r,int x){ printf("%d %d %d\n",l,r,sm[x]); if(l==r) return ; int mid=(l+r)>>1; print(lson);print(rson); } ll get(int p,int l,int r,int x){ if(l==r) return sm[x]; pushdown(x,r-l+1); int mid=(l+r)>>1; return p<=mid?get(p,lson):get(p,rson); } char s[5]; int main(){ n=read(),m=read();int u,v,d; rep(i,2,n) u=read()+1,w[i]=read(),add(u,i); dfs(1,0);DFS(1);build(1,n,1); //printf("->_->\n");print(1,n,1); rep(i,1,m){ scanf("%s",s);u=read()+1,v=read(),d=read(); if(s[0]==‘S‘) { if(get(idx[u],1,n,1)<v) update(idx[u],d,1,n,1); } else UPDATE(u,v,d); //printf("->_->\n");print(1,n,1); } rep(i,1,n) printf("%lld\n",get(idx[i],1,n,1)); return 0; }
第1行:2个数N, M,N为节点的数量,M为操作的数量(1 <= N, M <= 50000)。 第2 - N行:每行描述一个节点N[i]的信息,第2行对应编号为1的节点,第N行对应编号为N - 1的节点。具体内容为:每行2个数P[i], W[i]。P[i]为当前节点的父节点的编号,W[i]为当前节点的权值。(0 <= W[i] <= 10^5, P[i] < i) 第N + 1 - N + M行:每行表示一个操作,N x y z或A x y z,(0 <= y, z <= 10^5)。
输出共N行,每行1个数W[i],表示经过M次后,编号为0 - N - 1的节点的权值。
4 3 0 10 0 10 1 2 S 0 1 1 A 0 20 1 S 3 2 1
2 11 11 3
51nod1199 Money out of Thin Air
标签:
原文地址:http://www.cnblogs.com/fighting-to-the-end/p/5874741.html