题目链接:
http://www.lydsy.com/JudgeOnline/problem.php?id=2243
题解:
树剖的题,差不多算是裸题吧,线段树中维护三个值,为
tr [ id ],表示 id 所代表的区间中总共有多少段颜色。
lx [ id ],表示 id 所代表的区间中最左端的点的颜色。
rx [ id ],表示 id 所代表的区间中最右端的点的颜色。
然后按树剖的做法往下做即可,注意在树剖中查询后合并两段查询结果时的细节,需要判断一下合并的两个接点颜色是否相同,具体看代码。
代码:
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<vector>
#define maxn (100005)
#define lson (id*2)
#define rson (id*2+1)
using namespace std;
int tr[4*maxn],lx[4*maxn],rx[4*maxn],lazy[4*maxn];
//tr,lx,rx,lazy,线段树中维护的操作
int tot,n,m,fu[maxn],c[maxn],vis[maxn],in[maxn],size[maxn],son[maxn],fa[maxn],dp[maxn],top[maxn];
//这些都是树剖时用的信息
vector<int>lin[maxn];
char p[3];
//dfs1找重链
void dfs1(int x,int f,int d)
{
size[x]=1;
fa[x]=f;
dp[x]=d;
for (int i=0;i<lin[x].size();i++)
if (lin[x][i]!=f)
{
dfs1(lin[x][i],x,d+1);
size[x]+=size[lin[x][i]];
if (son[x]==0||size[son[x]]<size[lin[x][i]])
son[x]=lin[x][i];
}
}
//dfs2将节点对应在线段树中
void dfs2(