【HDU】5029 Relief grain 樹鍊剖分+離線标記法

傳送門：【HDU】5029 Relief grain

題目分析：這題的方法太美妙了！

首先看到這是一顆樹，那麼很容易想到樹鍊剖分。然後想到可以将查詢排個序，然後每一種查詢執行完以後更新每個點的最優值。但是這樣進行的複雜度太高！尤其是當z給的沒有一樣的時候尤其如此。

那麼我們是否可以找到更加高效的算法？

答案是肯定的！

先簡化一下問題，如果這些操作是線上段上進行的，我們怎麼求解？

我們很容易可以想到标記法：區間【L，R】染上顔色X，則位置L标記為X，表示從L開始染色X，位置R+1标記為-X，表示從R+1開始結束染色。用鄰接表儲存目前位置上的标記，然後從左往右，每到一個點上就把所有的标記執行了，為此我們建立一棵權值線段樹，+X，我們就在位置X上+1，-X，我們就在位置X上-1，然後用maxv标記記錄區間最大值。執行完這些操作以後就是查詢了，順着maxv == 最大值走，盡量往左走就行了。

然後我們回到本題，完全就是一個套路啊！

隻不過把連續的區間分成logN個不是連續的區間而已。還是按照上面的操作就可以了，不過要注意掃描時的點是線段樹上的點，而不是原來的點，還要映射回去。

而且由于本題的特殊性，我們直接将線段樹改成非遞歸效率更佳。

11693886

2014-09-21 10:19:32

Accepted

5029

734MS

12044K

3608 B

C++

poursoul

代碼如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std ;

typedef long long LL ;

#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 , 100000
#define rt o , l , r

const int MAXN = 100005 ;
const int MAXE = 2000005 ;

struct Edge {
	int v ;
	Edge* next ;
} ;

Edge E[MAXE] , *H[MAXN] , *h[MAXN] , *edge ;
int maxv[MAXN << 2] ;
int real[MAXN] ;
int siz[MAXN] ;
int top[MAXN] ;
int pos[MAXN] ;
int dep[MAXN] ;
int val[MAXN] ;
int pre[MAXN] ;
int son[MAXN] ;
int idx[MAXN] ;
int ans[MAXN] ;
int tree_idx ;
int n , m ;

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

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

void add ( 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 != pre[u] && v != son[u] ) rewrite ( v , v ) ;
	}
}

void mark ( int x , int y , int v ) {
	while ( top[x] != top[y] ) {
		if ( dep[top[x]] < dep[top[y]] ) swap ( x , y ) ;
		add ( pos[top[x]] , v ) ;
		add ( pos[x] + 1 , -v ) ;
		x = pre[top[x]] ;
	}
	if ( dep[x] > dep[y] ) swap ( x , y ) ;
	add ( pos[x] , v ) ;
	add ( pos[y] + 1 , -v ) ;
}

void build ( int o , int l , int r ) {
	if ( l == r ) {
		real[l] = o ;
		return ;
	}
	int m = mid ;
	build ( lson ) , build ( rson ) ;
}

void update ( int v , int o ) {
	maxv[o] += v ;
	while ( o > 1 ) {
		o >>= 1 ;
		maxv[o] = max ( maxv[ls] , maxv[rs] ) ;
	}
}

int query ( int v , int o , int l , int r ) {
	while ( l < r ) {
		int m = mid ;
		if ( maxv[ls] == v ) {
			r = m ;
			o = ls ;
		} else {
			l = m + 1 ;
			o = rs ;
		}
	}
	return l ;
}

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

void solve () {
	int x , y , c ;
	clear () ;
	rep ( i , 1 , n ) {
		scanf ( x ) , scanf ( y ) ;
		addedge ( x , y ) ;
		addedge ( y , x ) ;
	}
	dfs ( 1 ) ;
	rewrite ( 1 , 1 ) ;
	while ( m -- ) {
		scanf ( x ) ; scanf ( y ) ; scanf ( c ) ;
		mark ( x , y , c ) ;
	}
	build ( root ) ;
	FOR ( i , 1 , n ) {
		travel ( e , h , i ) {
			int v = e -> v ;
			if ( v > 0 ) update (  1 , real[ v] ) ;
			else         update ( -1 , real[-v] ) ;
		}
		if ( maxv[1] ) ans[idx[i]] = query ( maxv[1] , root ) ;
		else ans[idx[i]] = 0 ;
	}
	FOR ( i , 1 , n ) printf ( "%d\n" , ans[i] ) ;
}

int main () {
	while ( ~scanf ( "%d%d" , &n , &m ) && ( n || m ) ) solve () ;
	return 0 ;
}