2分木のノードの指定方法を変えたい
下のプログラムはコマンドライン引数できまった形で入力した数を2分木にしてそれを表示させるc言語のプログラムです。
きまった形というのは
短縮形は [ 8 [ 7 2 5 ] [ 3 1 _ ] ]
短縮形でないのは [ 8 [ 7 [ 2 _ _ ] [ 5 _ _ ] ] [ 3 [ 1 _ _ ] _ ] ] ]
のような形で、実行結果はそれぞれ
a.exe [ 8 [ 7 2 5 ] [ 3 1 _ ] ]
入力データ [ 8 [ 7 2 5 ] [ 3 1 _ ] ]
a.exe [ 8 [ 7 [ 2 _ _ ] [ 5 _ _ ] ] [ 3 [ 1 _ _ ] _ ] ] ]
入力データ [ 8 [ 7 2 5 ] [ 3 1 _ ] ]
のようになります。
これをこのような形ではなくコマンドラインで 8 7 3 2 5 1 と入力するだけで2分木になるように
下のプログラムを変えたいのです。
よろしくお願いします。
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef int BITREE_TYPE; /* 格納データの型 */
struct node {
BITREE_TYPE value; /* ノードの値 */
struct node *left; /* 左ノードのアドレス */
struct node *right; /* 右ノードのアドレス */
};
typedef struct node BITREE_NODE;
void error(char *msg);
BITREE_NODE *createNode(BITREE_TYPE x);
void destroyBITree(BITREE_NODE *p);
int isLeafNode(BITREE_NODE *p);
void printBITree(BITREE_NODE *p, int tabs, int brief);
BITREE_NODE *inputBITree(char *str[], int len, int *end);
int gShortFormat = 1; /*１ならば短縮形で出力する*/
void error(char *msg){
fflush(stdout);
fprintf(stderr,"%s\n", msg);
exit(1);
}
BITREE_NODE *createNode(BITREE_TYPE x){
BITREE_NODE *new;
new = malloc(sizeof(struct node));
if(new == NULL)
error("createNode: メモリがありません");
new->value = x;
new->left = NULL;
new->right = NULL;
return new;
}
void destroyBITree(BITREE_NODE *p){
if(p == NULL)
return;
destroyBITree(p->left);
destroyBITree(p->right);
memset(p, 0, sizeof(struct node));
free(p);
}
int isLeafNode(BITREE_NODE *p){
return(p->left == NULL) && (p->right == NULL);
}
void printSubtree(BITREE_NODE *p){
if(p == NULL){
printf("_");
return;
}
if(gShortFormat != 0 && isLeafNode(p)){
printf("%d", p->value);
} else{
printf("[ ");
printf("%d ", p->value);
printSubtree(p->left);
printf(" ");
printSubtree(p->right);
printf(" ]");
}
}
void printBITree(BITREE_NODE *p, int tabs, int brief){
int i;
gShortFormat = brief;
for (i = 0; i<tabs; i++)
printf("\t");
printSubtree(p);
printf("\n");fflush(stdout);
}
BITREE_NODE *inputBITree(char *str[], int len, int *end){
BITREE_NODE *p;
int i =0;
if(len < 1)
error("inputBITree:データがありません");
/*短縮形の処理*/
if(strcmp(str[0], "[") != 0){
if(strcmp(str[0], "_") == 0)
error("inputBTITree:値に＿は指定できません");
*end =1;
return createNode(atoi(str[0]));
}
p = createNode(atoi(str[1]));
i = 2;
if(strcmp(str[i], "_") != 0){
p->left = inputBITree(&str[i], len -i, end);
i+= *end;
} else{
i++;
}
if(strcmp(str[i], "_") != 0){
p->right = inputBITree(&str[i], len -i, end);
i+= *end;
} else{
i++;
}
if(strcmp(str[i], "]") != 0)
error("inputBITree: 入力データが]で終わっていません");
*end = i + 1;
return p;
}
int main(int argc, char *argv[]){
BITREE_NODE *p;
int end = 0;
p = inputBITree(argv+1, argc-1, &end);
printf("入力データ ");
printBITree(p,0,1);
destroyBITree(p);
return 0;
}
お礼
はい、ちゃんと動作していてます! 呼ぶときではなく、返るときに+1するのですね。 それとよくよく考えると、呼び出されたときにnullならそのまま返ってくるので p.leftとp.rightがnullかどうかなんて判断しなくてもよかったんですね。 とても勉強になりました。 本当に、本当に助かりました。 ありがとうございました!