USACO 3.4 American Heritage 美國血統

題目大意

---

給出一棵二叉樹的中序和前序遍歷，求後序遍歷。

 

樣例輸入&輸出

---

sample input

ABEDFCHG

CBADEFGH

sample output

AEFDBHGC

 

分析&反思

---

string什麼的stl也太暴力了。。。

爲了省時間手寫我也忍了。。。

 

前序遍歷 PreOrder 根節點+左子樹+右子樹

中序遍歷 InOrder 左子樹+根節點+右子樹

後序遍歷 PostOrder 左子樹+右子樹+根節點

 

中序 ABEDF C HG

前序 C BADEF GH

前序第一個即爲根，在中序中找到，得出左右子樹數量，便可以在中序中劃分左右子樹。

中序 A B EDF

前序 B A DEF

以左子樹爲新樹，重複這一過程，實現遞歸。

 

問題：1. l == r 時，得到葉子，直接打印。

           2. l > r 時，直接返回。

           一開始這兩點沒寫，錯出死循環。。。

 

代碼

---

#include<cstdio>
#include<iostream>
using namespace std;

int PreOrder[30], InOrder[30], PostOrder[30];  
int len;

void make(int root, int s, int l, int r) {
	if(l > r) return;
	if(l == r) {
		printf("%c", (char) 'A'+PreOrder[root]-1);
		return;
	}
	int lson = root+1, rson = root+(s-l)+1;
	int ls = 1, rs = 1;
	while(PreOrder[lson] != InOrder[ls]) {
		ls++;
	}
	while(PreOrder[rson] != InOrder[rs]) {
		rs++;
	}
	make(lson, ls, l, s-1);
	make(rson, rs, s+1, r);
	printf("%c", (char) 'A'+PreOrder[root]-1);
}

int main() {
	
	freopen("heritage.in", "r", stdin);
	freopen("heritage.out", "w", stdout);
	
	char c;
	while(c = getchar()) {
		if(c == '\n') break;
		InOrder[++len] = c-'A'+1;
	}
	for(int i = 1; i <= len; i++) {
		c = getchar();
		PreOrder[i] = c-'A'+1;
	}
	
	int s = 1;
	while(PreOrder[1] != InOrder[s]) {
		s++;
	}
	
	make(1, s, 1, len);
	
	printf("\n");
	
	return 0;
}