L2-011 두 갈래 나무 돌리기(25분)
1796 단어 PTA 계단 2 단계 문제집
입력 형식: 첫 번째 줄을 입력하면 정수 N(≤30)을 주고 두 갈래 나무의 결점의 개수입니다.두 번째 줄은 그 중의 서열을 반복한다.세 번째 줄은 앞의 순서를 반복해서 보여 준다.숫자 사이는 공백으로 구분된다.
출력 형식: 한 줄에서 이 트리가 반전된 후 겹쳐진 시퀀스를 출력합니다.숫자 사이는 1개의 공백으로 구분되며, 줄의 앞뒤에 여분의 공백이 있어서는 안 된다.
샘플 입력:
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7
내보내기 예제:
4 6 1 7 5 3 2
이 문제의 사고방식은 매우 간단하다. 거울 두 갈래 트리를 복원한다(원래 두 갈래 트리를 복원한 후에 거울 두 갈래 트리를 만들 필요가 없다. 직접 복원할 수 있다. 복원할 때 여전히 귀환 복원 알고리즘을 사용한다. 단지 순서를 바꾸었을 뿐이다. 먼저 귀환 두 갈래 트리를 복원하고 다시 귀환 왼쪽 트리를 복원하면 복원할 수 있다. 그러면 코드를 구체적으로 보면 된다)
#include
#include
#include
using namespace std;
int pre[35];
int in[35];
typedef struct node* bintree;
queue q;
struct node{
bintree left;
bintree right;
int data;
};
bintree huanyuan(int prel,int prer,int inl,int inr)
{
if(prel>prer)return NULL;
bintree bt=new node;
bt->data=pre[prel];
int k;
for(int i=inl;i<=inr;i++)
{
if(in[i]==pre[prel])
{
k=i;
break;
}
}
bt->right=huanyuan(prel+1,prel+k-inl,inl,k-1);
bt->left=huanyuan(prel+k-inl+1,prer,k+1,inr);
return bt;
}
void layeror(bintree bt,vector &vec)
{
if(bt==NULL)return;
q.push(bt);
while(!q.empty())
{
vec.push_back(q.front()->data);
bintree t=q.front();
q.pop();
if(t->left)q.push(t->left);
if(t->right)q.push(t->right);
}
}
int main()
{
int n;
cin>>n;
for(int i=0;i>in[i];
}
for(int i=0;i>pre[i];
}
bintree bt=huanyuan(0,n-1,0,n-1);
vector vec;
layeror(bt,vec);
for(int i=0;i