bzoj3589 动态树
题意:给你一棵树,要求你维护两个操作.
0:某子树的所有节点权值加上一个数.
1:求某些链并集的权值和.
对于第一个操作,大力树链剖分加线段树.
对于第二个操作,考虑到链的数量很少,可以容斥.
代码:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<iomanip>
using namespace std;
#define LL long long
#define up(i,j,n) for(int i=j;i<=n;i++)
#define pii pair<int,int>
#define db double
#define eps 1e-4
#define FILE "dealing"
int read(){
int x=0,f=1,ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0',ch=getchar();}
return x*f;
}
const LL maxn=1000000,inf=1000000000000000LL,limit=20000;
bool cmin(int& a,int b){return a>b?a=b,true:false;}
bool cmax(int& a,int b){return a<b?a=b,true:false;}
int n,m;
struct node{
int y,next;
}e[maxn];
int len=0,linkk[maxn],top[maxn],fa[maxn],dep[maxn],pre[maxn],low[maxn];
void insert(int x,int y){
e[++len].y=y;
e[len].next=linkk[x];
linkk[x]=len;
}
void init(){
n=read();
up(i,1,n-1){
int x=read(),y=read();
insert(x,y);
insert(y,x);
}
}
namespace treepou{
int dfs_clock=0,siz[maxn],son[maxn];
void dfs1(int x){
siz[x]=1;
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==fa[x])continue;
fa[e[i].y]=x;
dep[e[i].y]=dep[x]+1;
dfs1(e[i].y);
siz[x]+=siz[e[i].y];
if(siz[e[i].y]>siz[son[x]])son[x]=e[i].y;
}
}
void dfs2(int x){
pre[x]=++dfs_clock;
if(son[x]){
top[son[x]]=top[x];
dfs2(son[x]);
}
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==son[x]||e[i].y==fa[x])continue;
top[e[i].y]=e[i].y;
dfs2(e[i].y);
}
low[x]=dfs_clock;
}
void solve(){
dfs1(1);
top[1]=1;
dfs2(1);
}
};
namespace SegTree{
int del[maxn],siz[maxn],sum[maxn],L,R,key,id[maxn];
int Add(int x,int d){
del[x]+=d;
sum[x]+=siz[x]*d;
}
void updata(int x){
sum[x]=sum[x<<1]+sum[x<<1|1];
}
void pushdown(int x){
if(del[x]){
Add(x<<1,del[x]);
Add(x<<1|1,del[x]);
del[x]=0;
}
}
void change(int o,int l,int r){
if(l>R||r<L)return;
if(l>=L&&r<=R){Add(o,key);return;}
int mid=(l+r)>>1;
pushdown(o);
change(o<<1,l,mid);
change(o<<1|1,mid+1,r);
updata(o);
}
void Change(int x,int y,int Del){
L=x,R=y,key=Del;
change(1,1,n);
}
int query(int l,int r,int o){
if(l>R||r<L)return 0;
if(l>=L&&r<=R)return sum[o];
int mid=(l+r)>>1;
pushdown(o);
return query(l,mid,o<<1)+query(mid+1,r,o<<1|1);
}
int Query(int x,int y){
if(x>y)swap(x,y);
L=x,R=y;
return query(1,n,1);
}
void build(int l,int r,int o){
if(l==r){
siz[o]=1;
sum[o]=0;
return;
}
int mid=(l+r)>>1;
build(l,mid,o<<1);
build(mid+1,r,o<<1|1);
siz[o]=siz[o<<1]+siz[o<<1|1];
}
void Build(){
up(i,1,n)id[pre[i]]=i;
build(1,n,1);
}
}
namespace solve{
pii getlian(int x,int y){
int f1,f2;
int sum=0;
while(true){
if(dep[x]>dep[y])swap(x,y);
f1=top[x],f2=top[y];
if(f1==f2){
sum+=SegTree::Query(pre[x],pre[y]);
return make_pair(x,sum);
}
if(dep[f1]>dep[f2])swap(x,y),swap(f1,f2);
sum+=SegTree::Query(pre[f2],pre[y]);
y=fa[f2];
}
}
int getlca(int x,int y){
int f1,f2;
while(true){
if(dep[x]>dep[y])swap(x,y);
f1=top[x],f2=top[y];
if(f1==f2)return x;
if(dep[f1]>dep[f2])swap(x,y),swap(f1,f2);
y=fa[f2];
}
}
pii t[maxn];
int query(){
int m=read();
up(i,1,m){
t[i].first=read(),t[i].second=read();
if(dep[t[i].first]>dep[t[i].second])swap(t[i].first,t[i].second);
}
pii y=make_pair(0,0);
int ans=0;
for(int i=1;i<(1<<m);i++){
int g=-1,flag=0;
up(j,1,m){
if(i&(1<<j-1))g*=-1;
else continue;
if(!flag)y=t[j],flag=1;
else {
int w=getlca(y.second,t[j].second);
if(dep[w]<dep[y.first]||dep[w]<dep[t[j].first]){
y=make_pair(0,0);
break;
}
else {
if(dep[y.first]<dep[t[j].first])y.first=t[j].first;
y.second=w;
}
}
}
//printf("%d %d %d\n",g,y.first,y.second);
if(y.first)ans+=g*getlian(y.first,y.second).second;
}
return ans;
}
};
int main(){
freopen(FILE".in","r",stdin);
freopen(FILE".out","w",stdout);
init();
treepou::solve();//树剖 返回top,dep,fa
SegTree::Build();//prepare SegTree
int m=read();
while(m--){
int ch=read();
if(ch==0){
int x=read(),del=read();
SegTree::Change(pre[x],low[x],del);
}
else printf("%d\n",solve::query()&((1<<31)-1));
}
return 0;
}
