【HDU】5044 Tree 树链剖分

最新推荐文章于 2017-04-13 20:47:00 发布
最新推荐文章于 2017-04-13 20:47:00 发布
阅读量1.3k 收藏
点赞数
分类专栏： 树链剖分 文章标签： HDU
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u013368721/article/details/39617167
版权

传送门：【HDU】5044 Tree


题目分析：如果不看卡常数这回事的话。。这就是一个很裸的树链剖分。。然后就可以用线段树维护。

但是！这样对于本题是一定会超时的！因为出题人特意想卡。。。

于是我换成树状数组+输入优化卡过。。。


但这题还有更好的方法！我们可以在树链剖分上用标记法，每次对连续区间的位置L标记+v，位置R+1标记-v，最后扫一遍结果就出来了。。


O(nlog^2n)代码如下：


#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std ;

typedef long long LL ;

#pragma comment(linker, "/STACK:16777216")
#define rep( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define For( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define rev( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define travel( e , H , u ) for ( Edge* e = H[u] ; e ; e = e -> next )
#define clr( a , x ) memset ( a , x , sizeof a )
#define cpy( a , x ) memcpy ( a , x , sizeof a )
#define ls ( o << 1 )
#define rs ( o << 1 | 1 )
#define lson ls , l , m
#define rson rs , m + 1 , r
#define mid ( ( l + r ) >> 1 )
#define root 1 , 1 , n

const int MAXN = 100005 ;
const int MAXE = 200005 ;

struct Edge {
	int v ;
	Edge* next ;
} E[MAXE] , *H[MAXN] , *edge ;

struct Seg {
	int u , v ;
	Seg () {}
	Seg ( int u , int v ) : u ( u ) , v ( v ) {}
} seg[MAXN] ;

int siz[MAXN] ;
int top[MAXN] ;
int pre[MAXN] ;
int pos[MAXN] ;
int dep[MAXN] ;
int son[MAXN] ;
int tree_idx ;
int ans1[MAXN] , ans2[MAXN] ;
int idx[MAXN] ;
int seg_idx[MAXN] ;
LL T[2][MAXN] ;
LL addv[2][MAXN << 2] ;
int n , q ;

void clear () {
	edge = E ;
	tree_idx = 0 ;
	pre[1] = 0 ;
	siz[0] = 0 ;
	clr ( H , 0 ) ;
	clr ( T , 0 ) ;
}

void addedge ( int u , int v ) {
	edge -> v = v ;
	edge -> next = H[u] ;
	H[u] = edge ++ ;
}

void dfs ( int u ) {
	siz[u] = 1 ;
	son[u] = 0 ;
	travel ( e , H , u ) {
		int v = e -> v ;
		if ( v != pre[u] ) {
			pre[v] = u ;
			dep[v] = dep[u] + 1 ;
			dfs ( v ) ;
			siz[u] += siz[v] ;
			if ( siz[v] > siz[son[u]] ) son[u] = v ;
		}
	}
}

void rewrite ( int u , int top_element ) {
	top[u] = top_element ;
	pos[u] = ++ tree_idx ;
	idx[tree_idx] = u ;
	if ( son[u] ) rewrite ( son[u] , top_element ) ;
	travel ( e , H , u ) {
		int v = e -> v ;
		if ( v != son[u] && v != pre[u]) rewrite ( v , v ) ;
	}
}

void add ( int x , int o , int v ) {
	while ( x ) {
		T[o][x] += v ;
		x -= x & -x ;
	}
}

LL sum ( int x , int o , LL ans = 0 ) {
	while ( x <= n ) {
		ans += T[o][x] ;
		x += x & -x ;
	}
	return ans ;
}

void pushdown ( int x , int o ) {
	if ( addv[x][o] ) {
		addv[x][ls] += addv[x][o] ;
		addv[x][rs] += addv[x][o] ;
		addv[x][o] = 0 ;
	}
}

void sub_update ( int L , int R , int x , int v , int o , int l , int r ) {
	if ( L <= l && r <= R ) {
		addv[x][o] += v ;
		return ;
	}
	int m = mid ;
	pushdown ( x , o ) ;
	if ( L <= m ) sub_update ( L , R , x , v , lson ) ;
	if ( m <  R ) sub_update ( L , R , x , v , rson ) ;
}

void update ( int x , int y , int o , int v ) {
	while ( top[x] != top[y] ) {
		if ( dep[top[x]] > dep[top[y]] ) {
			add ( pos[x] , o , v ) ;
			add ( pos[top[x]] - 1 , o , -v ) ;
			//sub_update ( pos[top[x]] , pos[x] , o , v , root ) ;
			x = pre[top[x]] ;
		} else {
			add ( pos[y] , o , v ) ;
			add ( pos[top[y]] - 1 , o , -v ) ;
			//sub_update ( pos[top[y]] , pos[y] , o , v , root ) ;
			y = pre[top[y]] ;
		}
	}
	if ( o && x == y ) return ;
	if ( dep[x] > dep[y] ) {
		add ( pos[x] , o , v ) ;
		add ( pos[y] + o - 1 , o , -v ) ;
		//sub_update ( pos[y] + o , pos[x] , o , v , root ) ;
	} else {
		add ( pos[y] , o , v ) ;
		add ( pos[x] + o - 1 , o , -v ) ;
		//sub_update ( pos[x] + o , pos[y] , o , v , root ) ;
	}
}

void down ( int o , int l , int r ) {
	if ( l == r ) {
		ans1[idx[l]] = addv[0][o] ;
		ans2[seg_idx[l]] = addv[1][o] ;
		return ;
	}
	int m = mid ;
	pushdown ( 0 , o ) ;
	pushdown ( 1 , o ) ;
	down ( lson ) ;
	down ( rson ) ;
}

void scanf ( int& x , char c = 0 , int flag = 0 ) {
	while ( ( c = getchar () ) != '-' && ( c < '0' || c > '9' ) ) ;
	if ( c == '-' ) flag = 1 , x = 0 ;
	else x = c - '0' ;
	while ( ( c = getchar () ) >= '0' && c <= '9' ) x = x * 10 + c - '0' ;
	if ( flag ) x = -x ;
}

void solve () {
	char buf[10] ;
	int u , v , w ;
	clear () ;
	scanf ( "%d%d" , &n , &q ) ;
	rep ( i , 1 , n ) {
		scanf ( u ) , scanf ( v ) ;
		seg[i] = Seg ( u , v ) ;
		addedge ( u , v ) ;
		addedge ( v , u ) ;
	}
	dfs ( 1 ) ;
	rewrite ( 1 , 1 ) ;
	rep ( i , 1 , n ) {
		u = seg[i].u ;
		v = seg[i].v ;
		if ( dep[u] < dep[v] ) seg_idx[pos[v]] = i ;
		else seg_idx[pos[u]] = i ;
	}
	while ( q -- ) {
		scanf ( "%s" , buf ) ;
		scanf ( u ) , scanf ( v ) , scanf ( w ) ;
		if ( buf[3] == '1' ) update ( u , v , 0 , w ) ;
		else update ( u , v , 1 , w ) ;
	}
	For ( i , 1 , n ) ans1[idx[i]] = sum ( i , 0 ) ;
	For ( i , 2 , n ) ans2[seg_idx[i]] = sum ( i , 1 ) ;
	//down ( root ) ;
	For ( i , 1 , n ) printf ( "%d%s" , ans1[i] , i < n ? " " : "" ) ;
	printf ( "\n" ) ;
	rep ( i , 1 , n ) printf ( "%d%s" , ans2[i] , i < n - 1 ? " " : "" ) ;
	printf ( "\n" ) ;
}

int main () {
	int T , cas = 0 ;
	scanf ( "%d", &T ) ;
	while ( T -- ) {
		printf ( "Case #%d:\n" , ++ cas ) ;
		solve () ;
	}
	return 0 ;
}

O(nlogn + n)代码如下： 


#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std ;

typedef long long LL ;

#pragma comment(linker, "/STACK:16777216")
#define rep( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define For( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define rev( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define travel( e , H , u ) for ( Edge* e = H[u] ; e ; e = e -> next )
#define clr( a , x ) memset ( a , x , sizeof a )
#define cpy( a , x ) memcpy ( a , x , sizeof a )
#define ls ( o << 1 )
#define rs ( o << 1 | 1 )
#define lson ls , l , m
#define rson rs , m + 1 , r
#define mid ( ( l + r ) >> 1 )
#define root 1 , 1 , n

const int MAXN = 100005 ;
const int MAXE = 200005 ;

struct Edge {
	int v ;
	Edge* next ;
} E[MAXE] , *H[MAXN] , *edge ;

struct Seg {
	int u , v ;
	Seg () {}
	Seg ( int u , int v ) : u ( u ) , v ( v ) {}
} seg[MAXN] ;

int siz[MAXN] ;
int top[MAXN] ;
int pre[MAXN] ;
int pos[MAXN] ;
int dep[MAXN] ;
int son[MAXN] ;
int tree_idx ;
LL ans1[MAXN] , ans2[MAXN] ;
int idx[MAXN] ;
int seg_idx[MAXN] ;
LL a[2][MAXN] ;
int n , q ;

void clear () {
	edge = E ;
	tree_idx = 0 ;
	pre[1] = 0 ;
	siz[0] = 0 ;
	clr ( H , 0 ) ;
	clr ( a , 0 ) ;
}

void addedge ( int u , int v ) {
	edge -> v = v ;
	edge -> next = H[u] ;
	H[u] = edge ++ ;
}

void dfs ( int u ) {
	siz[u] = 1 ;
	son[u] = 0 ;
	travel ( e , H , u ) {
		int v = e -> v ;
		if ( v != pre[u] ) {
			pre[v] = u ;
			dep[v] = dep[u] + 1 ;
			dfs ( v ) ;
			siz[u] += siz[v] ;
			if ( siz[v] > siz[son[u]] ) son[u] = v ;
		}
	}
}

void rewrite ( int u , int top_element ) {
	top[u] = top_element ;
	pos[u] = ++ tree_idx ;
	idx[tree_idx] = u ;
	if ( son[u] ) rewrite ( son[u] , top_element ) ;
	travel ( e , H , u ) {
		int v = e -> v ;
		if ( v != son[u] && v != pre[u]) rewrite ( v , v ) ;
	}
}

void update ( int x , int y , int o , int v ) {
	while ( top[x] != top[y] ) {
		if ( dep[top[x]] > dep[top[y]] ) {
			a[o][pos[top[x]]] += v ;
			a[o][pos[x] + 1] -= v ;
			x = pre[top[x]] ;
		} else {
			a[o][pos[top[y]]] += v ;
			a[o][pos[y] + 1] -= v ;
			y = pre[top[y]] ;
		}
	}
	if ( o && x == y ) return ;
	if ( dep[x] > dep[y] ) {
		a[o][pos[y] + o] += v ;
		a[o][pos[x] + 1] -= v ;
	} else {
		a[o][pos[x] + o] += v ;
		a[o][pos[y] + 1] -= v ;
	}
}

void scanf ( int& x , char c = 0 , int flag = 0 ) {
	while ( ( c = getchar () ) != '-' && ( c < '0' || c > '9' ) ) ;
	if ( c == '-' ) flag = 1 , x = 0 ;
	else x = c - '0' ;
	while ( ( c = getchar () ) >= '0' && c <= '9' ) x = x * 10 + c - '0' ;
	if ( flag ) x = -x ;
}

void solve () {
	char buf[10] ;
	int u , v , w ;
	clear () ;
	scanf ( "%d%d" , &n , &q ) ;
	rep ( i , 1 , n ) {
		scanf ( u ) , scanf ( v ) ;
		seg[i] = Seg ( u , v ) ;
		addedge ( u , v ) ;
		addedge ( v , u ) ;
	}
	dfs ( 1 ) ;
	rewrite ( 1 , 1 ) ;
	rep ( i , 1 , n ) {
		u = seg[i].u ;
		v = seg[i].v ;
		if ( dep[u] < dep[v] ) seg_idx[pos[v]] = i ;
		else                   seg_idx[pos[u]] = i ;
	}
	while ( q -- ) {
		scanf ( "%s" , buf ) ;
		scanf ( u ) , scanf ( v ) , scanf ( w ) ;
		if ( buf[3] == '1' ) update ( u , v , 0 , w ) ;
		else update ( u , v , 1 , w ) ;
	}
	LL x = 0 , y = 0 ;
	For ( i , 1 , n ) {
		x += a[0][i] ;
		ans1[idx[i]] = x ;
	}
	For ( i , 2 , n ) {
		y += a[1][i] ;
		ans2[seg_idx[i]] = y ;
	}
	For ( i , 1 , n ) printf ( "%I64d%s" , ans1[i] , i < n ? " " : "" ) ;
	printf ( "\n" ) ;
	rep ( i , 1 , n ) printf ( "%I64d%s" , ans2[i] , i < n - 1 ? " " : "" ) ;
	printf ( "\n" ) ;
}

int main () {
	int T , cas = 0 ;
	scanf ( "%d", &T ) ;
	while ( T -- ) {
		printf ( "Case #%d:\n" , ++ cas ) ;
		solve () ;
	}
	return 0 ;
}


poursoul
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
专栏目录
HDU 5044 Tree 树链剖分+差分
noone0的博客
09-16 244
题意:n点m条边的带权树,点边都带权，Q次操作. op1:将u,v路径上所有点的权值加k. op2:将u,v路径上所有边的权值加k. n,m 操作为区间+,查询只要最后一次,可以不用线段树,破树成链以后,用两个数组差分一下就好了. #pragma comment(linker, "/STACK:1024000000,1024000000") #include usin
hdu 3333 turing tree 解题报告
03-10
题目“HDU 3333 Turing Tree”要求解决的问题是：给定一个整数序列和一系列区间，计算每个区间内不重复数字的和。由于数据规模较大（N ,000, K ,000），直接的暴力方法效率过低，因此我们需要采用一种更高效的数据...
HDU 5044 Tree 树链剖分
弱智就要多努力
05-21 372
题目：http://acm.hdu.edu.cn/showproblem.php?pid=5044 题意：给定一个树，点和边都有权值，初始都为0。有t组数据，每组数据输入n和m，代表节点数和操作数，接下来n - 1行，每行两个数，代表两点有边相连。然后m行，形式如 ADD1 a b c 或 ADD2 a b c，前一种是把点a到点b路径上的点权都加上c，后一种是把点a到点b路径上的边权
hdu 5044 Tree 树链剖分
xiaoyu1_1的专栏
09-29 438
树链剖分：http://blog.sina.com.cn/s/blog_7a1746820100wp67.html
HDU 5044 Tree 树链剖分
weixin_30491641的博客
10-10 52
一棵树，初始边权和点权都为0 现在有m个操作，每一个操作： ADD1 u v k: for nodes on the path from u to v, the value of these nodes increase by k.ADD2 u v k: for edges on the path from u to v, the value of these edges increase ...
Hdu5044Tree 树链剖分
weixin_30323961的博客
11-15 54
　　输入输出挂，扩栈挂，模板树链剖分。 #include<iostream> #include<cstdio> #include<cstring> #include<map> #include<vector> #include<stdlib.h> #include<algorithm> usin...
HDU 5044 Tree 树链剖分+区间标记
weixin_30855761的博客
04-13 127
Tree Problem Description You are given a tree (an acyclic undirected connected graph) with N nodes. The tree nodes are numbered from 1 to NThere are N - 1 edges numbered from 1 to N - 1.Each node...
hdu5044 Tree 树链剖分，点剖分，边剖分，非递归版
冰
10-06 1215
hdu5044 Tree 树链剖分，点剖分，边剖分，非递归版 //#pragma warning (disable: 4786) //#pragma comment (linker, "/STACK:16777216") //#pragma comment(linker, "/STACK:60400000,60400000") //HEAD #include #include #i
HDU-5044 Tree 树链剖分
Darkrai
06-27 272
树链剖分
hdu 5044 Tree(树链剖分）
铅笔头的博客
09-02 266
题意：给一个简单图，在图上进行n此操作，最后按顺序输出点权和边权的值。T组数据，每组数据第一行两个数n，m分别代表树的点数，操作的次数，接下来n-1行每行两个数表示一条边。接下来m行表示m次操作，每次操作有一一个字符串add1，或add2，接下来是三个数u，v，c。add1表示在修改点权u->v的路径上的点加c，add2表示修改边权u->v的路径上的边权加c。解题思路：知道题虽然是比较经典的树链剖分
hdu acm1166线段树
05-03
hdu 1166线段树代码
hdu.rar_hdu
最新发布
09-23
3. **数据结构**：数组、链表、栈、队列、树（二叉树、平衡树如AVL和红黑树）、图（邻接矩阵、邻接表、最小生成树、拓扑排序等）。 4. **字符串处理**：KMP算法、Manacher's Algorithm、Rabin-Karp算法等。 5. **...
二叉搜索树练习 HDU3791
04-14
**二叉搜索树（Binary Search Tree，BST）**是一种特殊的二叉树结构，它具有以下特性： 1. **性质**：对于任意节点，其左子树中的所有节点的值都小于该节点的值，右子树中的所有节点的值都大于该节点的值。 2. **...
hdu.zip_ACM_hdu
09-20
在ACM(国际大学生程序设计竞赛)中，数据结构与算法是至关重要的组成部分，而"杭电"(杭州电子科技大学)的在线判题系统HDU提供了许多这类问题供参赛者练习。"hdu.zip_ACM_hdu"这个压缩包很可能是包含了一道或若干道...
【HDU】5111 Alexandra and Two Trees 树链剖分+主席树【树上路径的交集】【在线算法】
poursoul
11-25 2607
传送门：【HDU】5111 Alexandra and Two Trees
【HDU】5029 Relief grain 树链剖分+离线标记法
poursoul
09-21 1811
传送门：【HDU】5029 Relief grain
【codeforces】487E. Tourists 点双连通+树链剖分
poursoul
11-25 1400
传送门：【codeforces】487E. Tourists 题目分析：这么多天了终于艹掉了这题，太感动了T ^ T 看到这题，询问路径最小值问题，第一反应就是缩成一棵树然后树链剖分求解。兴冲冲的敲了一个边双连通，测了样例没过，发现不可以用边双连通。。。然后换成点双连通，将一个点双连通块中的边缩成一个点，然后所有的割顶缩成一个点，然后一个连通块我们用一个set保存最小值。在没有更新的情
2014 Asia AnShan Regional Contest 题解
poursoul
10-30 1353
B：5071 Chat
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值