P3060 [USACO12NOV]Balanced Trees G

题目

P3060 [USACO12NOV]Balanced Trees G

给出一棵树,每个节点一个括号,可能左也可能右,求树上一条合法括号路径的最大嵌套层数。

分析

很容易想到点分治,那么现在问题变成如何拼接路径。

由于这样的路径左右顺序会有影响,于是考虑分开处理,也就是拿两个东西分别记从这里出发,和在这里结束两类路径。

然后维护合法性其实就是很简单,分成两截内部和公共的,内部的直接在枚举路径的时候判断就好了,然后公共的就相当于可以对每一个值开一个桶,储存外面需要的答案即可。

那么外面需要的答案是什么呢?

考虑一个已知合法的括号序列怎么求最大嵌套层数,我们可以发现其实这就是括号栈内数量的历史最大值,于是我们在找路径的时候记录一下就好了。

拼接就是取两个历史最值的最值就好了。

剩下的都很好维护了。

代码

#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
	x=0;bool f=false;char ch=getchar();
	while(!isdigit(ch)){f|=ch=='-';ch=getchar();}
	while(isdigit(ch)){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
	x=f?-x:x;
	return ;
} 
template <typename T>
inline void write(T x){
	if(x<0) putchar('-'),x=-x;
	if(x>9) write(x/10);
	putchar(x%10^48);
	return ;
}
#define ll long long
const int N=2e5+5;
int n,head[N],nex[N],to[N],idx;
int siz[N],FMax,Root,Size;
int q[N],top,a[N],Max1[N],Max2[N],Ans;
bool vis[N];
char op[5];
inline void add(int u,int v){
	nex[++idx]=head[u];
	to[idx]=v;
	head[u]=idx;
	return ;
}
void GetRoot(int x,int fa){
	siz[x]=1;int Max=0;
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==fa||vis[y]) continue;
		GetRoot(y,x);siz[x]+=siz[y];
		Max=max(Max,siz[y]);
	}
	Max=max(Max,Size-siz[x]);
	if(Max<=FMax) FMax=Max,Root=x;
	return ;
}
struct Path{int Max1,Max2,Len1,Len2,cnt1,cnt2;}val[N];
void GetPath(int x,int fa){
	q[++top]=x;
	val[x].Len1+=a[x];
	val[x].Max1=max(val[x].Max1,-val[x].Len1);
	if(val[x].Len1>0) val[x].cnt1++,val[x].Max1++,val[x].Len1=0;
	val[x].Len2+=a[x];
	val[x].Max2=max(val[x].Max2,val[x].Len2);
	if(val[x].Len2<0) val[x].cnt2++,val[x].Max2++,val[x].Len2=0;
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(vis[y]||y==fa) continue;
		val[y]=val[x];GetPath(y,x);
	}
	return ;
}
void DFS(int x,int fa){
	FMax=Size;GetRoot(x,fa);x=Root;vis[x]=true;
	Path now=(Path){0,0,0,0,0,0};
	if(a[x]==1) now.cnt1=1;
	else now.Len1=-1;
	now.Max1=1;
	if(!now.Len1) Max1[now.cnt1]=now.Max1;
	int Cnt1,Cnt2;
	Max2[0]=top=Cnt1=Cnt2=0; 
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(vis[y]) continue;
		val[y]=now;int tmp=top+1;
		GetPath(y,x);
		for(int i=tmp;i<=top;i++){
			Path res=val[q[i]];
			Cnt1=max(Cnt1,res.cnt1),Cnt2=max(Cnt2,res.cnt2);
			if(!res.Len1 && ~Max2[res.cnt1]) Ans=max(Ans,max(Max2[res.cnt1],res.Max1));
            if(!res.Len2 && ~Max1[res.cnt2]) Ans=max(Ans,max(Max1[res.cnt2],res.Max2));
        }
        for(int i=tmp;i<=top;i++){
            Path res=val[q[i]];
            if(!res.Len1) Max1[res.cnt1]=max(Max1[res.cnt1],res.Max1);
            if(!res.Len2) Max2[res.cnt2]=max(Max2[res.cnt2],res.Max2);
        }
	}
	for(int i=0;i<=Cnt1;i++) Max1[i]=-1;
	for(int i=0;i<=Cnt2;i++) Max2[i]=-1;
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==fa||vis[y]) continue;
		Size=siz[y],DFS(y,x);
	}
	return ;
}
signed main(){
	read(n);
	for(int i=2;i<=n;i++){
		int x;read(x);
		add(x,i),add(i,x);
	}
	memset(Max1,-1,sizeof(Max1));
	memset(Max2,-1,sizeof(Max2));
	for(int i=1;i<=n;i++) scanf("%s",op),a[i]=(op[0]=='('?1:-1);
	Size=n,DFS(1,0);
	write(Ans);
	return 0;
}
原文地址:https://www.cnblogs.com/Akmaey/p/14728544.html