HDU-1710 Binary Tree Traversals

地址:http://acm.hdu.edu.cn/showproblem.php?pid=1710

思路:通過對先序遍歷和中序遍歷的遍歷順序分析,對於先序遍歷,第一個節點一定是根節點,再定位到中序遍歷中當前根節點的位置,可以將先序遍歷拆分成兩個子樹的先序遍歷,這樣就可以利用遞歸將遍歷集合一步步縮小,從而得到解。

Code:

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

const int MAX_N=1005;
int n,tt;
int pre[MAX_N],in[MAX_N];
unordered_map<int,int> imap;

void DFS(int l,int r,int l1);
int main()
{
	ios::sync_with_stdio(false);
	while(cin>>n){
		imap.clear();
		for(int i=0;i<n;++i)
			cin>>pre[i];
		for(int i=0;i<n;++i)
		{
			cin>>in[i];
			imap[in[i]]=i;
		}
		tt=0;
		DFS(0,n-1,0);
		cout<<endl;
	}
	
	return 0;
}

void DFS(int l,int r,int l1)
{
	if(l>r)	return;
	int t=imap[pre[l]];
	DFS(l+1,l+t-l1,l1);
	DFS(l+t-l1+1,r,t+1);
	if(tt)	cout<<" ";
	cout<<pre[l];
	tt=1;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章