Description
Bob有一棵n个点的有根树,其中1号点是根节点。Bob在每个点上涂了颜色,并且每个点上的颜色不同。定义一条路
径的权值是:这条路径上的点(包括起点和终点)共有多少种不同的颜色。Bob可能会进行这几种操作:
1 x:
把点x到根节点的路径上所有的点染上一种没有用过的新颜色。
2 x y:
求x到y的路径的权值。
3 x y:
在以x为根的子树中选择一个点,使得这个点到根节点的路径权值最大,求最大权值。
Bob一共会进行m次操作
Input
第一行两个数n,m。
接下来n-1行,每行两个数a,b,表示a与b之间有一条边。
接下来m行,表示操作,格式见题目描述
1<=n,m<=100000
Output
每当出现2,3操作,输出一行。
如果是2操作,输出一个数表示路径的权值
如果是3操作,输出一个数表示权值的最大值
Sample Input
5 6
1 2
2 3
3 4
3 5
2 4 5
3 3
1 4
2 4 5
1 5
2 4 5
1 2
2 3
3 4
3 5
2 4 5
3 3
1 4
2 4 5
1 5
2 4 5
Sample Output
3
4
2
2
4
2
2
HINT
Source
LCT
我们把相同颜色的点之间的边认为是preferred path
发现第一个操作就是access
然后用链剖线段树维护区间最值即可
#include <bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
#define fill( x, y ) memset( x, y, sizeof x )
#define copy( x, y ) memcpy( x, y, sizeof x )
using namespace std;
typedef long long LL;
typedef pair < int, int > pa;
inline int read()
{
int sc = 0; char ch = getchar();
while( ch < '0' || ch > '9' ) ch = getchar();
while( ch >= '0' && ch <= '9' ) sc = sc * 10 + ch - '0', ch = getchar();
return sc;
}
const int MAXN = 100010;
struct edge { int to, nxt; } e[MAXN << 1];
int n, m, head[MAXN], e_cnt, f[MAXN], dep[MAXN], fa[MAXN], ch[MAXN][2];
int sz[MAXN], bl[MAXN], dfn[MAXN], tim, rl[MAXN], mx[MAXN << 2], tag[MAXN << 2];
inline void add(int x, int y) { e[ ++e_cnt ] = { y, head[ x ] }; head[ x ] = e_cnt; }
inline void addedge(int x, int y) { add( x, y ); add( y, x ); }
inline void dfs(int x)
{
sz[ x ] = 1;
for( int i = head[ x ] ; i ; i = e[ i ].nxt ) if( e[ i ].to ^ f[ x ] )
f[ e[ i ].to ] = x, dep[ e[ i ].to ] = dep[ x ] + 1, dfs( e[ i ].to ), sz[ x ] += sz[ e[ i ].to ];
}
inline void dfs2(int x, int chain)
{
bl[ rl[ dfn[ x ] = ++tim ] = x ] = chain; int k = 0;
for( int i = head[ x ] ; i ; i = e[ i ].nxt ) if( e[ i ].to != f[ x ] && sz[ e[ i ].to ] > sz[ k ] ) k = e[ i ].to;
if( !k ) return ;
dfs2( k, chain );
for( int i = head[ x ] ; i ; i = e[ i ].nxt ) if( e[ i ].to != f[ x ] && e[ i ].to != k ) dfs2( e[ i ].to, e[ i ].to );
}
inline int lca(int x, int y)
{
for( ; bl[ x ] ^ bl[ y ] ; x = f[ bl[ x ] ] )
if( dep[ bl[ x ] ] < dep[ bl[ y ] ] ) swap( x, y );
return ( dep[ x ] < dep[ y ] ) ? x : y;
}
inline void pushup(int x) { mx[ x ] = max( mx[ x << 1 ], mx[ x << 1 | 1 ] ) + tag[ x ]; }
inline void build(int x, int l, int r)
{
if( l == r ) { mx[ x ] = dep[ rl[ l ] ]; return ; }
int mid = l + r >> 1; build( x << 1, l, mid ); build( x << 1 | 1, mid + 1, r );
pushup( x );
}
inline void modify(int x, int l, int r, int ql, int qr, int v)
{
if( l == ql && r == qr ) { tag[ x ] += v; mx[ x ] += v; return ; }
int mid = l + r >> 1;
if( qr <= mid ) modify( x << 1, l, mid, ql, qr, v );
else if( ql > mid ) modify( x << 1 | 1, mid + 1, r, ql, qr, v );
else modify( x << 1, l, mid, ql, mid, v ), modify( x << 1 | 1, mid + 1, r, mid + 1, qr, v );
pushup( x );
}
inline int query(int x, int l, int r, int ql, int qr)
{
if( l == ql && r == qr ) return mx[ x ];
int mid = l + r >> 1;
if( qr <= mid ) return query( x << 1, l, mid, ql, qr ) + tag[ x ];
if( ql > mid ) return query( x << 1 | 1, mid + 1, r, ql, qr ) + tag[ x ];
return max( query( x << 1, l, mid, ql, mid ), query( x << 1 | 1, mid + 1, r, mid + 1, qr ) ) + tag[ x ];
}
inline bool isroot(int x) { return ch[ fa[ x ] ][ 0 ] != x && ch[ fa[ x ] ][ 1 ] != x; }
inline void rotate(int x)
{
int y = fa[ x ], z = fa[ y ], k = ch[ y ][ 1 ] == x;
if( !isroot( y ) ) ch[ z ][ ch[ z ][ 1 ] == y ] = x;
fa[ ch[ y ][ k ] = ch[ x ][ k ^ 1 ] ] = y;
fa[ fa[ ch[ x ][ k ^ 1 ] = y ] = x ] = z;
}
inline void splay(int x)
{
for( int y = fa[ x ] ; !isroot( x ) ; rotate( x ), y = fa[ x ] ) if( !isroot( y ) )
rotate( ( ( ch[ y ][ 1 ] == x ) == ( ch[ fa[ y ] ][ 1 ] == y ) ) ? y : x );
}
inline void access(int x)
{
for( int t = 0 ; x ; t = x, x = fa[ x ] )
{
splay( x );
if( ch[ x ][ 1 ] )
{
int cur = ch[ x ][ 1 ];
while( ch[ cur ][ 0 ] ) cur = ch[ cur ][ 0 ];
modify( 1, 1, n, dfn[ cur ], dfn[ cur ] + sz[ cur ] - 1, 1 );
}
ch[ x ][ 1 ] = t;
if( ch[ x ][ 1 ] )
{
int cur = ch[ x ][ 1 ];
while( ch[ cur ][ 0 ] ) cur = ch[ cur ][ 0 ];
modify( 1, 1, n, dfn[ cur ], dfn[ cur ] + sz[ cur ] - 1, -1 );
}
}
}
int main()
{
#ifdef wxh010910
freopen( "data.in", "r", stdin );
#endif
n = read(), m = read();
for( int i = 1 ; i < n ; i++ ) addedge( read(), read() );
dep[ 1 ] = 1; dfs( 1 ); dfs2( 1, 1 ); build( 1, 1, n );
for( int i = 1 ; i <= n ; i++ ) fa[ i ] = f[ i ];
while( m-- )
{
int opt = read();
if( opt == 1 ) access( read() );
else if( opt == 2 )
{
int x = read(), y = read(), k = lca( x, y );
printf( "%d\n", query( 1, 1, n, dfn[ x ], dfn[ x ] ) + query( 1, 1, n, dfn[ y ], dfn[ y ] ) - 2 * query( 1, 1, n, dfn[ k ], dfn[ k ] ) + 1 );
}
else
{
int x = read();
printf( "%d\n", query( 1, 1, n, dfn[ x ], dfn[ x ] + sz[ x ] - 1 ) );
}
}
}