[낙곡 루거] P5018-대칭 이차수(귀속)

원본 링크:https://www.luogu.org/problemnew/show/P5018

제목의 대의.


두 갈래 나무를 정해 가장 큰 대칭자 나무를 구하다.

해법


문제풀이에서 전해지는 해법과 유사하여 주로 복잡도가 예측하기 어렵다.
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define forn(i, n) for (int i = 0; i < (int)(n); ++i)
#define for1(i, n) for (int i = 1; i <= (int)(n); ++i)
#define pb push_back

using namespace std;
typedef long long LL;
const int N=1000000+5;

struct Node{
	int w;
	int l,r;
}Tree[N];

bool issame(int x,int y){
	if(x==-1&&y==-1) return 1;
	if(x==-1||y==-1) return 0;
	if(Tree[x].w!=Tree[y].w) return 0;
	if(!issame(Tree[x].l,Tree[y].r)) return 0;
	if(!issame(Tree[x].r,Tree[y].l)) return 0;
	return 1;
}
int d[N],ans=0,n;
void dfs(int u){
	int l=Tree[u].l,r=Tree[u].r;
	if(l==-1&&r==-1){
		d[u]=1;
		return;
	}
	if(l!=-1) dfs(l);
	if(r!=-1) dfs(r);
	d[u]=(l!=-1?d[l]:0)+(r!=-1?d[r]:0)+1;
//	cout<
}

void dfs_(int u){
	int l=Tree[u].l,r=Tree[u].r;
	if(l==-1&&r==-1){
		ans=max(ans,1);
		return;
	}
	if(issame(l,r)){
		ans=max(ans,1+d[l]+d[r]);
		return ;
	}
	if(l!=-1) dfs_(l);
	if(r!=-1) dfs_(r);
}

int main() {

#ifdef LOCAL_DEFINE
	freopen("P5018.in", "rt", stdin);
#endif
	scanf("%d",&n);
	for1(i,n) scanf("%d",&Tree[i].w);
	for1(i,n) scanf("%d%d",&Tree[i].l,&Tree[i].r);
	dfs(1);
	dfs_(1);
	printf("%d
"
,ans); #ifdef LOCAL_DEFINE cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.
"
; #endif return 0; }

좋은 웹페이지 즐겨찾기