看了刘汝佳编写的《算法竞赛入门经典》中的一节(p148),用数组表示一颗二叉树,第一次见到这样的数据结构
#include<cstdio>
#include<string>
#include<iostream>
using namespace std;
//根据算法竞赛入门p148实现,节点编号问题见书
int rightc[25] = { 0 };//每个节点的左指针
int leftc[25] = { 0 };//每个节点的右指针
int value[25] = { 0 };//节点的值,0代表节点为空
void addnode(int v, char *c)//添加节点
{
int k = 1;//初始化操作指针指向数组头元素,x[0]不储存任何值,作为头来使用
int father = 0;//初始化标记,节点k若为左节点,father=1,右节点=2
for (int i = 0; i < strlen(c)-1; i++)//处理指令部分,由于指令中带有“)”,所以处理到上一位
{
if (c[i] == 'L')//若向左插入,k=k*2,二叉树节点编号规律
{
k = 2 * k;
father = 1;
}
if (c[i] == 'R')
{
father = 2;
k = 2 * k + 1;
}
}
if (strlen(c) == 1)//当输入结束时,输入(),指令处理无法进入循环,另行处理
{
k = 1;
father = 0;
}
value[k] = v;//对相应的节点赋值
//修改父节点的指向
if (father == 1)
leftc[k / 2] = k;
if (father == 2)
rightc[(k - 1) / 2] = k;
}
void input(FILE *p)
//输入指令例如(55,RRL)
//表示某个节点值为55,位置是从根节点开始的右节点的 右节点的 左节点。
{
for (;;)
{
char c[10];
char m[20];
//scanf("%s", c);
fscanf(p,"%s", c);
if (strlen(c) == 2) break;//检测到()时结束输入
int v;
sscanf(&c[1], "%d", &v);
//printf("%d\n", v);
char *beg = strchr(c, ',');//“,”开始的位置
memcpy(m, beg+1, strlen(beg));//将“,”后的字符串给m,包含最后的“)”
//printf("%s\n", m);
addnode(v, m);//插入节点
}
}
void bfs()
//二叉树的广度优先遍历顺序刚好是数组的顺序,遍历数组有值就输出
{
for (int i = 0; i < 25; i++)
{
if (value[i] == 0) continue;
else printf("%d ", value[i]);
}
}
void preorder(int k)
//先序遍历为例
{
printf("%d ", value[k]);
if (leftc[k] != 0) preorder(leftc[k]);
if (rightc[k] != 0) preorder(rightc[k]);
}
int main()
{
FILE *p;
p = fopen("123.txt", "r");
input(p);
bfs();
printf("\n");
preorder(1);
return 0;
}