zoukankan      html  css  js  c++  java
  • 【BZOJ3730】震波 动态树分治+线段树

    【BZOJ3730】震波

    Description

    在一片土地上有N个城市,通过N-1条无向边互相连接,形成一棵树的结构,相邻两个城市的距离为1,其中第i个城市的价值为value[i]。
    不幸的是,这片土地常常发生地震,并且随着时代的发展,城市的价值也往往会发生变动。
    接下来你需要在线处理M次操作:
    0 x k 表示发生了一次地震,震中城市为x,影响范围为k,所有与x距离不超过k的城市都将受到影响,该次地震造成的经济损失为所有受影响城市的价值和。
    1 x y 表示第x个城市的价值变成了y。
    为了体现程序的在线性,操作中的x、y、k都需要异或你程序上一次的输出来解密,如果之前没有输出,则默认上一次的输出为0。

    Input

    第一行包含两个正整数N和M。
    第二行包含N个正整数,第i个数表示value[i]。
    接下来N-1行,每行包含两个正整数u、v,表示u和v之间有一条无向边。
    接下来M行,每行包含三个数,表示M次操作。

    Output

    包含若干行,对于每个询问输出一行一个正整数表示答案。

    Sample Input

    8 1
    1 10 100 1000 10000 100000 1000000 10000000
    1 2
    1 3
    2 4
    2 5
    3 6
    3 7
    3 8
    0 3 1

    Sample Output

    11100101

    HINT

    1<=N,M<=100000
    1<=u,v,x<=N
    1<=value[i],y<=10000
    0<=k<=N-1

    题解:直接动态树分治,对于每一个分治中心,我们用维护一棵线段树维护它分治子树中的所有节点,下标为节点到分治中心的距离,权值为城市的权值。同时为了防重,还要维护一个从父亲那里减掉的版本。查询时,从当前点一直向父亲移动,每次在线段树中查一下就行了。

    #include <cstdio>
    #include <cstring>
    #include <iostream>
    using namespace std;
    const int maxn=100010;
    struct node
    {
    	int ls,rs,sum;
    }s[maxn*200];
    int n,m,cnt,tot,root,mn,ans;
    int to[maxn<<1],next[maxn<<1],head[maxn],v[maxn],fa[maxn],siz[maxn],r1[maxn],r2[maxn],vis[maxn],Log[maxn<<1];
    int md[20][maxn<<1],pos[maxn],dep[maxn];
    void getrt(int x,int fa)
    {
    	siz[x]=1;
    	int i,tmp=0;
    	for(i=head[x];i!=-1;i=next[i])	if(!vis[to[i]]&&to[i]!=fa)
    		getrt(to[i],x),siz[x]+=siz[to[i]],tmp=max(tmp,siz[to[i]]);
    	tmp=max(tmp,tot-siz[x]);
    	if(tmp<mn)	root=x,mn=tmp;
    }
    void solve(int x)
    {
    	vis[x]=1;
    	for(int i=head[x];i!=-1;i=next[i])	if(!vis[to[i]])	tot=siz[to[i]],mn=1<<30,getrt(to[i],x),fa[root]=x,solve(root);
    }
    void dfs(int x)
    {
    	md[0][++pos[0]]=dep[x],pos[x]=pos[0];
    	for(int i=head[x];i!=-1;i=next[i])	if(!dep[to[i]])	dep[to[i]]=dep[x]+1,dfs(to[i]),md[0][++pos[0]]=dep[x];
    }
    inline void add(int a,int b)
    {
    	to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
    }
    inline int getmin(int a,int b)
    {
    	if(a>b)	swap(a,b);
    	int k=Log[b-a+1];
    	return min(md[k][a],md[k][b-(1<<k)+1]);
    }
    inline int dis(int a,int b)
    {
    	return dep[a]+dep[b]-2*getmin(pos[a],pos[b]);
    }
    void updata(int l,int r,int &x,int a,int b)
    {
    	if(!x)	x=++tot;
    	s[x].sum+=b;
    	if(l==r)	return ;
    	int mid=(l+r)>>1;
    	if(a<=mid)	updata(l,mid,s[x].ls,a,b);
    	else	updata(mid+1,r,s[x].rs,a,b);
    }
    int query(int l,int r,int x,int a,int b)
    {
    	if(!x||a>b)	return 0;
    	if(a<=l&&r<=b)	return s[x].sum;
    	int mid=(l+r)>>1;
    	if(b<=mid)	return query(l,mid,s[x].ls,a,b);
    	if(a>mid)	return query(mid+1,r,s[x].rs,a,b);
    	return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);
    }
    inline char nc(){
        static char buf[100000],*p1=buf,*p2=buf;
        return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
    }
    inline int rd(){
        char ch=nc();int sum=0;
        while(!(ch>='0'&&ch<='9'))ch=nc();
        while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
        return sum;
    }
    int main()
    {
    	//freopen("bz3730.in","r",stdin);
    	//freopen("bz3730.out","w",stdout);
    	n=rd(),m=rd();
    	int a,b,c;
    	register int i,j,x,y;
    	memset(head,-1,sizeof(head));
    	for(i=1;i<=n;i++)	v[i]=rd();
    	for(i=1;i<n;i++)	a=rd(),b=rd(),add(a,b),add(b,a);
    	dep[1]=1,dfs(1),tot=n,mn=1<<30,getrt(1,0),solve(root);
    	for(i=2;i<=pos[0];i++)	Log[i]=Log[i>>1]+1;
    	for(j=1;(1<<j)<=pos[0];j++)	for(i=1;i+(1<<j)-1<=pos[0];i++)	md[j][i]=min(md[j-1][i],md[j-1][i+(1<<(j-1))]);
    	tot=0;
    	for(i=1;i<=n;i++)
    	{
    		x=i;
    		while(x)
    		{
    			y=fa[x],updata(0,n,r1[x],dis(x,i),v[i]);
    			if(y)	updata(0,n,r2[x],dis(y,i),v[i]);
    			x=y;
    		}
    	}
    	for(i=1;i<=m;i++)
    	{
    		c=rd(),a=rd()^ans,b=rd()^ans;
    		if(!c)
    		{
    			ans=0,x=a;
    			while(x)
    			{
    				y=fa[x],ans+=query(0,n,r1[x],0,b-dis(x,a));
    				if(y)	ans-=query(0,n,r2[x],0,b-dis(y,a));
    				x=y;
    			}
    			printf("%d
    ",ans);
    		}
    		else
    		{
    			x=a;
    			while(x)
    			{
    				y=fa[x],updata(0,n,r1[x],dis(x,a),b-v[a]);
    				if(y)	updata(0,n,r2[x],dis(y,a),b-v[a]);
    				x=y;
    			}
    			v[a]=b;
    		}
    	}
    	return 0;
    }
  • 相关阅读:
    Jmeter接口测试与数据驱动
    Jmeter内存溢出
    Mac下多版本pip共存
    Linux命令: ps
    intellij ultimate2019.3 注册
    纪念已经逝去的十年
    H3C路由器设置NAT回环、端口回流
    理清PHP在Linxu下执行时的文件权限
    windows下vagrant的安装使用
    无限极分类(adjacency list)的三种方式(迭代、递归、引用)
  • 原文地址:https://www.cnblogs.com/CQzhangyu/p/7536695.html
Copyright © 2011-2022 走看看