hdu 5044 Tree

        题意:一棵树。有两种操作,对两个点uv路径上的所有点加权和对路径上的所有边加权。最后输出所有点的权和所有边的权。

        思路:树链剖分+树状数组。感觉这个题出出来就是为了报复社会的,我一共跪了25次。。又是扩栈又是输入挂又是内联,最后只有用c++交能过,g++爆栈。不过话说回来这个题也挺模版的,都是树链剖分的基本操作。因为没学过对边的处理,我把边绑在点上了,更改查询边时,去操作边向远离根方向连接的点。另外树状数组不能区间更新,所以每次更新在起点加上k,终点+1减去k。


#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<string.h>
#include<cstdio>

using namespace std;

#define maxn 100010
#define ll long long 


int head[maxn];
int next[maxn*2];
int to[maxn*2];
int e2v[maxn]; //第i条边向下接着那个哪个点
int edgeu[maxn];
int edgev[maxn];

int siz[maxn];  //以i为根的子树的节点数
int dep[maxn];  //节点深度
int top[maxn];  //所在链的顶端节点
int fa[maxn];   //父节点
int son[maxn];  //重儿子
int w[maxn];    //节点在线段树中的位置
int w_1[maxn];	//上面的反函数 
bool vis[maxn];
int pos;
int end;

int n,m,p;

ll cv[maxn];
ll ce[maxn];
ll ans1[maxn];
ll ans2[maxn];

void scanf_(int &num){
  char in;
  bool neg=false;
  while(((in=getchar()) > '9' || in<'0') && in!='-') ;
  if(in=='-')
  {
    neg=true;
    while((in=getchar()) >'9' || in<'0');
  }
  num=in-'0';
  while(in=getchar(),in>='0'&&in<='9')
    num*=10,num+=in-'0';
  if(neg)
    num=0-num;
}

void init(){
    memset(siz,0,sizeof(siz));
    memset(top,0,sizeof(top));
    memset(son,0,sizeof(son));
    memset(head,-1,sizeof(head));
    memset(cv,0,sizeof(cv));
    memset(ce,0,sizeof(ce));
    pos=0;
    end=0;
}

inline void addedge(int u,int v){
    to[end]=v,next[end]=head[u],head[u]=end++;
    to[end]=u,next[end]=head[v],head[v]=end++;
}

void dfs(int x,int pre){
    vis[x]=1;
    dep[x]=dep[pre]+1;
    fa[x]=pre;
    for(int i=head[x];i!=-1;i=next[i]){
        if(to[i]==pre)continue;
        if(vis[to[i]])continue;
        dfs(to[i],x);
        siz[x]+=siz[to[i]];
        if(son[x]==0)son[x]=to[i];
        else if(siz[son[x]]<siz[to[i]])son[x]=to[i];
    }
    siz[x]++;
}

void dfs2(int x,int pre){
    vis[x]=1;
    pos++;
    w[x]=pos;
    w_1[pos]=x;
    if(son[fa[x]]==x)top[x]=top[fa[x]];
    else top[x]=x;
    //先递归重儿子
    if(son[x]==0)return;
    dfs2(son[x],x);
    for(int i=head[x];i!=-1;i=next[i]){
        if(to[i]==pre)continue;
        if(to[i]==son[x])continue;
        if(vis[to[i]])continue;
        dfs2(to[i],x);
    }
}


inline int lowbit(int x){
	return x&(-x);
}

ll sum(int end,int type){
	if(type==1){
		ll sum=0;
		while(end){
			sum+=cv[end];
			end-=lowbit(end);
		}
		return sum;
	}else{
		ll sum=0;
		while(end){
			sum+=ce[end];
			end-=lowbit(end);
		}
		return sum;
	}
}

void update(int pos,ll num,int type){
	if(type==1){
		while(pos<=n+1){
			cv[pos]+=num;
			pos+=lowbit(pos);
		}
	}else{
		while(pos<=n+1){
			ce[pos]+=num;
			pos+=lowbit(pos);
		}
	}
}

int main(){
    int t;
    cin>>t;
    for(int cas=1;cas<=t;cas++){
        cin>>n>>m;
        init();
        for(int i=1;i<n;i++){
            scanf_(edgeu[i]); scanf_(edgev[i]);
            addedge(edgeu[i],edgev[i]);
        }

        memset(vis,0,sizeof(vis));
        dfs(1,0);
        memset(vis,0,sizeof(vis));
        dfs2(1,0);

        for(int i=1;i<n;i++){
            if(fa[edgeu[i]]==edgev[i]){
                e2v[i]=edgeu[i];
            }else{
                e2v[i]=edgev[i];
            }
        }

        char op[8];
        for(int i=1;i<=m;i++){
            scanf("%s",op);
            int u,v,k;
            scanf_(u); scanf_(v); scanf_(k);
            if(op[3]=='1'){
                while(top[u]!=top[v]){
                    if(dep[top[u]]<dep[top[v]])swap(u,v);
                    update(w[top[u]],k,1);
                    update(w[u]+1,-k,1);
                    u=fa[top[u]];
                }
                if(w[u]>w[v])swap(u,v);
                update(w[u],k,1);
                update(w[v]+1,-k,1);
            }else{
                while(top[u]!=top[v]){
                    if(dep[top[u]]<dep[top[v]])swap(u,v);
                    update(w[top[u]],k,2);
                    update(w[u]+1,-k,2);
                    u=fa[top[u]];
                }
                if(w[u]>w[v])swap(u,v);
                update(w[u]+1,k,2);	//注意少更新一个点
                update(w[v]+1,-k,2); 
            }
        }

        printf("Case #%d:\n",cas);
        ll t=0;
        for(int i=1;i<=n;i++){
        	t+=sum(i,1);
			t-=sum(i-1,1);
			ans1[i]=t;
        }
        t=0;
        for(int i=1;i<=n;i++){
        	t+=sum(i,2);
			t-=sum(i-1,2);
			ans2[i]=t;
        }
		for(int i=1;i<=n;i++){
            printf("%I64d",ans1[w[i]]);
            if(i!=n)printf(" ");
        }
        printf("\n");
        for(int i=1;i<n;i++){
        	printf("%I64d",ans2[w[e2v[i]]]);
            if(i!=n-1)printf(" ");
        }
        printf("\n");
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值