Matlab编程练习7(二叉树)

问题

(问题都是来源某牛的题，用matlab编程实现)

  有一棵二叉树，树上的叶子节点定义为“樱桃”。现在需要找出树上有多少个满足如下子结构的“樱桃”串，即一串上刚好有两颗“樱桃”。
  樱桃:
  比如如下的一棵树，红框标示的有两个符合要求的结构，答案就是2

  又比如下面的这颗树，没有任何符合要求的子结构，则答案是0。

输入例子

1 left 2
1 right 3
2 left 4
2 right 5
3 right 6
6 left 7
6 right 8
8 left 9
8 right 10

输出例子

2

例子说明
   如题目说明的第一个样例图，可以看到，2-4-5子串，8-9-10子串，两个子串符合条件，所以答案为2

---

解题思路

   题目给出来的输入是二叉树的广度遍历顺序，观察可以发现，非叶子节点的节点都在输入的右边即T{i}位置，因此可以先找出所有非叶子节点的结点，然后再判断其两个孩子是否为叶子节点…
   因为输入的数据结构比较复杂，既有字符串也有数字，因此可以考虑用元组接受输入。

---

Matlab代码实现

T={
    1 'left' 2,
    1 'right' 3,
    2 'left' 4,
    2 'right' 5,
    3 'right' 6,
    6 'left' 7,
    6 'right' 8,
    8 'left' 9,
    8 'right' 10};
%% 找出二叉树中的所有父节点
t=[];                  %t只保留两个孩子的父节点
tt=[];                 %%tt保留所有父节点
for i=1:size(T,1)-1    %遍历树
    tt=[tt,T{i}];
    if T{i}==T{i+1}    
        t=[t,T{i}];
    end
end
t=unique(t);
tt=unique(tt);
A=[];                  %A保存答案
%% 对于父节点，判断其子节点是否满足樱桃
% 樱桃：两个子节点都不是父节点，两个子节点都有
for i=1:size(t,2)          %遍历所有两个孩子的父节点
    for j=1:size(T,1)-1    %遍历树
        if T{j}==t(i)      %是两个孩子的父节点
            c=0;
            for k=1:size(tt,2)  %判断其孩子是否都为叶子节点
                if T{j,3}==tt(k) || T{j+1,3}==tt(k) %是父节点
                    c=c+1;
                end
            end
            if c==0             %是叶子节点
                A=[A,T{j}];
            end
        end  
    end
end
if size(A,1)==0
    disp(0);
else
    disp(size(A,2));
end

总结

   虽然本题的解题思路好像没用到太多二叉树的知识，但是对于一些常用的二叉树操作比如深度遍历二叉树、广度遍历二叉树、统计二叉树层数、统计节点个数、存储二叉树结构等等还是需要有一定了解。