题解 P2056 【[ZJOI2007]捉迷藏 】
写个括号序列的做法。
首先dfs整棵树一遍,进入一个节点的时候加上一个左括号,然后是节点编号,当这个节点的所有子树遍历完后再添上一个右括号,这就是括号序列。(其实就是dfs序加上了括号而已)
举个栗子,这棵树的括号序列是(1(2(3))(4(5)(6)(7(8))))
我们要求3到8的距离,截取两点间的括号序列为
3))(4(5)(6)(7(8
把编号和匹配的括号删掉
))(((
剩下了5个左右括号,而这就是3到8的距离。
这就是括号序列的性质。
怎么证明?脑补一下,到达i点时
1)添完了左右括号
那个点是i的祖先们的孩子。(不在i到根的路径上)
2)添了左括号没填右括号
那个点是i的祖先。(在i到根的路径上)
3)没添左括号
那个点是i的祖先们的孩子或者i的孩子(不在i到根的路径上)
因此,从s到t,删去的匹配括号们对于s来说是情况3,对于t来说是情况1,很显然这个点不在s到t的路径上。剩下的右括号,对于s来说是情况2,对于t来说是情况1,因此表示从s到达【t到根的这条链】经过的节点数。剩下的左括号,对于s来说是情况3,对于t来说是情况2,因此表示从t到达【s和t的lca(不包括lca)】的经过点数。
综上我们证明了删掉两点间所有匹配括号剩下的左右括号数为距离。
现在我们已经把整棵树压成了一个括号序列了,而我们要求的是两个黑点间的最大距离,我们考虑用线段树求解。
毫无疑问要记录每段区间删掉匹配括号后剩下的左右括号数,我们记右括号数为a,左括号数为b。
【注意我接下来说的所有左右区间都不限于线段树中的左右区间,任意连续的左右区间均可】
左右区间的a、b的合并:
显然左区间的左括号和右区间的右括号合并消去,谁多剩谁
//lc左区间,rc右区间
if(tr[lc].b>tr[rc].a)
tr[id].a=tr[lc].a,tr[id].b=tr[lc].b-tr[rc].a+tr[rc].b;else
tr[id].a=tr[lc].a+tr[rc].a-tr[lc].b,tr[id].b=tr[rc].b;
跨区间距离计算:
左区间右左括号数记为a1,b1,右区间a2,b2
dis=a1+abs(b1-a2)+b2=max((a1+b1)+(b2-a2),(a1-b1)+(a2+b2))
显然max(a1+b1),max(b2-a2)这种是可以单独维护的。
因此记录区间前缀的max(a+b),max(b-a)(l1,l2),后缀的max(a+b),max(a-b)(r1,r2)即可维护dis值
tr[id].dis=max(max(tr[lc].r1+tr[rc].l2,tr[lc].r2+tr[rc].l1),max(tr[lc].dis,tr[rc].dis));
l1,r1,l2,r2的维护参考合并操作,不解释了自己体会
tr[id].l1=max(tr[lc].l1,max(tr[rc].l1+tr[lc].a-tr[lc].b,tr[rc].l2+tr[lc].a+tr[lc].b));
tr[id].l2=max(tr[lc].l2,tr[rc].l2-tr[lc].a+tr[lc].b);
tr[id].r1=max(tr[rc].r1,max(tr[lc].r1-tr[rc].a+tr[rc].b,tr[lc].r2+tr[rc].a+tr[rc].b));
tr[id].r2=max(tr[rc].r2,tr[lc].r2+tr[rc].a-tr[rc].b);
因为是要求黑点间的最大距离,显然初始化时白点的l1,r1,l2,r2是没有意义的,置为-inf
至此本题解决。
//括号序列
#include<cstdio>
//(a1 b1)(a2 b2)->(a,b)
//a+b=a1+abs(b1-a2)+b2=max((a1-b1)+(a2+b2),(a1+b1)+(b2-a2))
//需要左区间后缀的max(a-b),max(a+b),右区间前缀的max(a+b),max(b-a)
int num,s[300005],pos[1000005],head[100005],n,m,cnt,tot;
bool c[100005];
struct edge{int to,next;}e[200005];
void add(int u,int v){e[++num]=(edge){v,head[u]},head[u]=num;}
struct node
{
int a,b,l1,l2,r1,r2,dis;
//a,b右左括号数,l1,l2前缀的max(a+b),max(b-a),r1,r2后缀的max(a+b),max(a-b)
}tr[1200005];
void dfs(int u,int fa)
{
s[++tot]=-1;//左括号
s[++tot]=u;pos[u]=tot;
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].to;if(v==fa)continue;
dfs(v,u);
}
s[++tot]=-2;//右括号
}
void push(int id,int x)
{
tr[id].a=tr[id].b=0;tr[id].l1=tr[id].l2=tr[id].r1=tr[id].r2=tr[id].dis=-1e9;
if(s[x]==-1)tr[id].b=1;else
if(s[x]==-2)tr[id].a=1;else
if(!c[s[x]])tr[id].l1=tr[id].r1=tr[id].r2=tr[id].l2=0;//黑点
}
int max(int a,int b){return a>b?a:b;}
void merge(int id)
{
int lc=id<<1,rc=id<<1|1;
if(tr[lc].b>tr[rc].a)
tr[id].a=tr[lc].a,tr[id].b=tr[lc].b-tr[rc].a+tr[rc].b;else
tr[id].a=tr[lc].a+tr[rc].a-tr[lc].b,tr[id].b=tr[rc].b;
tr[id].l1=max(tr[lc].l1,max(tr[rc].l1+tr[lc].a-tr[lc].b,tr[rc].l2+tr[lc].a+tr[lc].b));
tr[id].l2=max(tr[lc].l2,tr[rc].l2-tr[lc].a+tr[lc].b);
tr[id].r1=max(tr[rc].r1,max(tr[lc].r1-tr[rc].a+tr[rc].b,tr[lc].r2+tr[rc].a+tr[rc].b));
tr[id].r2=max(tr[rc].r2,tr[lc].r2+tr[rc].a-tr[rc].b);
tr[id].dis=max(max(tr[lc].r1+tr[rc].l2,tr[lc].r2+tr[rc].l1),max(tr[lc].dis,tr[rc].dis));
}
void build(int id,int l,int r)
{
if(l==r){push(id,l);return;}
int mid=l+r>>1;
build(id<<1,l,mid);build(id<<1|1,mid+1,r);
merge(id);
}
void modify(int id,int l,int r,int x)
{
if(l==r){push(id,l);return;}
int mid=l+r>>1;
if(x<=mid)modify(id<<1,l,mid,x);else modify(id<<1|1,mid+1,r,x);
merge(id);
}
int main()
{
scanf("%d",&n);
for(int i=1,u,v;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v),add(v,u);
}
dfs(1,0);cnt=n;
build(1,1,tot);
scanf("%d",&m);
for(int i=1,x;i<=m;i++)
{
char s[2];
scanf("%s",s);
if(s[0]=='C')scanf("%d",&x),cnt+=c[x]?1:-1,c[x]^=1,modify(1,1,tot,pos[x]);
else if(cnt==0)printf("-1\n");else if(cnt==1)printf("0\n");else
printf("%d\n",tr[1].dis);
}
}