UVa536_Tree Recovery

1252 단어
제목: 클릭 링크 열기
사고방식: 먼저 차례대로 나무뿌리를 찾은 다음에 중차례대로 나무뿌리를 찾아 좌우 나무 결점에서 차례대로 차례대로 차례대로 차례대로 좌우 나무를 구성한다.
#define _CRT_SECURE_NO_WARNINGS
#include<cstdio>
#include<stack>
#include<vector>
#include<queue>
#include<cstring>
#include<sstream>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;

string pre_order, in_order;        // 、 
int lch[60], rch[60];              // 、 

int build(int L1, int R1, int L2, int R2){   //L1—R1 ,L2-R2 
	if (L1 > R1) return 0;					// 
	int  root = pre_order[L1]-'A'+1;        // 
	int p = L2;
	while (in_order[p]-'A'+1 != root) p++;  // 
	int cnt = p - L2;
	lch[root] = build(L1 + 1, L1 + cnt, L2, L2 + cnt - 1);  // 
	rch[root] = build(L1 + cnt + 1, R1, L2 + cnt + 1, R2);
	return root;
}

void bfs(int root){
	if (lch[root])	bfs(lch[root]);
	if (rch[root])	bfs(rch[root]);
	printf("%c", root+'A'-1);
}

int main(){
	//freopen("random_numbers.txt","r",stdin);
	while (cin >> pre_order >> in_order){
		int len = pre_order.length();
		build(0, len - 1, 0, len - 1);
		bfs(pre_order[0]-'A'+1);		
		printf("
"); } return 0; }

좋은 웹페이지 즐겨찾기