剣指OFFERの上から下へ二叉木を印刷(九度OJ 1523)
7054 ワード
タイトルの説明:
ツリーの各ノードを上から下に印刷し、同レイヤノードを左から右に印刷します.
入力:
入力には複数のテストサンプルが含まれ、入力はEOFで終了します.各テストケースについて、入力する最初の行の整数n(1<=n<=1000、:nは入力するツリー要素の個数(ノード1から番号付け)を表します.次の行には、i番目のツリーノードの要素の値を表すn個の数字があります.次はn行で、各行に1文字Ciがあります.Ci=’d’は,i番目のノードに2人の子があり,次いで左の子番号と右の子番号であることを示す.Ci=’l’は、i番目のノードに左の子があり、左の子の番号が続くことを示す.Ci=’r’は、i番目のノードに右の子があり、右の子の番号が続くことを示す.Ci=’z’はi番目のノードに子供がいないことを示す.
出力:
各テストケースに対応して、ツリーノードの値を上下から左から右に印刷します.
サンプル入力:
7
8 6 5 7 10 9 11
d 2 5
d 3 4
z
z
d 6 7
z
z
サンプル出力:
8 6 10 5 7 9 11
問題解決の考え方:
古典的な広さ優先アルゴリズムで、何も言うことはありません.
広さ優先アルゴリズムはここを見て
アルゴリズムの考え方:
1最上位レベルのツリーノードをスキャンし、その子供ノードをキューに入れます.
2キューヘッダノードをスキャンするたびに、その子供をチームに追加し、左の子供、右の子供の順にして、1階の左右の順序を保証します.
3キューが空になるまで.
//main()
findTree(a,n,1);
while(begin_q != end_q){
findTree(a,n,Quene[begin_q++]);
}
//
void findTree(treeArr *a,int n,int j){
if(j<=n){
result[top_result++]=a->arr[j].num;
}
if(a->arr[j].lchild != 0){
Quene[end_q++] = a->arr[j].lchild;
}
if(a->arr[j].rchild != 0){
Quene[end_q++] = a->arr[j].rchild;
}
}
すべてのコード:
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#define MAXSIZE 1005
typedef struct treenode{
int num;
int lchild;
int rchild;
}Tree;
typedef struct treearr{
struct treenode arr[MAXSIZE];
}treeArr;
int Quene[MAXSIZE]={0};
int begin_q,end_q;
int result[MAXSIZE]={0};
int top_result;
void findTree(treeArr *a,int n,int j);
int main(){
int n,i;
char c;
int n1,n2;
while(scanf("%d",&n)!=EOF && n>=1 && n<=1000){
treeArr *a = (treeArr *)malloc(sizeof(treeArr));
memset(&Quene,0,sizeof(int)*MAXSIZE);
memset(&result,0,sizeof(int)*MAXSIZE);
begin_q=0;
end_q = 0;
top_result = 0;
for(i=0;i<MAXSIZE;i++){
a->arr[i].num = 0;
a->arr[i].lchild = 0;
a->arr[i].rchild = 0;
}
for(i=1;i<=n;i++){
scanf("%d",&a->arr[i].num);
}
for(i=1;i<=n;i++){
scanf("
%c",&c);
if(c == 'd'){
scanf("%d %d",&n1,&n2);
a->arr[i].lchild = n1;
a->arr[i].rchild = n2;
}else if(c == 'l'){
scanf("%d",&n1);
a->arr[i].lchild = n1;
}else if(c == 'r'){
scanf("%d",&n1);
a->arr[i].rchild = n1;
}else{
}
}
findTree(a,n,1);
while(begin_q != end_q){
//printf("findtree --- begin_q %d end_q %d
",begin_q,end_q );
findTree(a,n,Quene[begin_q++]);
}
for(i=0;i<n-1;i++){
printf("%d ", result[i]);
}
printf("%d
", result[n-1]);
}
return 0;
}
void findTree(treeArr *a,int n,int j){
if(j<=n){
result[top_result++]=a->arr[j].num;
}
if(a->arr[j].lchild != 0){
Quene[end_q++] = a->arr[j].lchild;
}
if(a->arr[j].rchild != 0){
Quene[end_q++] = a->arr[j].rchild;
}
}
/**************************************************************
Problem: 1523
User: xhalo
Language: C
Result: Accepted
Time:0 ms
Memory:920 kb
****************************************************************/