无限极分类查找所有子孙节点的改进算法

在以前，遇到无限极分类返回一个节点的所有子孙节点时，我都是用递归计算的，后来发现时间复杂度和空间复杂度都太高了，后来自己研究了一下改进了算法.

节点数据如下：键值对分别是自己对应父亲节点

<?php
$tree=array(
1=>0,
2=>1,
3=>2,
4=>3,
5=>4,
6=>5,
7=>6,
8=>7,
9=>8,
10=>9,
11=>10,
12=>11,
13=>12,
14=>13,
15=>14,
16=>15,
17=>16,
18=>17,
19=>18,
20=>19,
21=>20,
22=>21,
23=>22,
24=>23,
25=>24,
26=>25,
27=>26,
28=>27,
29=>28,
30=>29,
31=>30,
32=>31,
33=>32,
34=>33,
35=>34,
36=>35,
37=>36,
38=>37,
39=>38,
40=>39,
41=>40,
42=>41,
43=>42,
44=>43,
45=>44,
46=>45,
47=>46,
48=>47,
49=>48,
50=>49,
51=>50,
52=>51,
53=>52,
54=>53,
55=>54,
56=>55,
57=>56,
58=>57,
59=>58,
60=>59,
61=>60,
62=>61,
63=>62,
64=>63,
65=>64,
66=>65,
67=>66,
68=>67,
69=>68,
70=>69,
71=>70,
72=>71,
73=>72,
74=>73,
75=>74,
76=>75,
77=>76,
78=>77,
79=>78,
80=>79,
81=>80,
82=>81,
83=>82,
84=>83,
85=>84,
86=>85,
87=>86,
88=>87,
89=>88,
90=>89,
91=>90,
92=>91,
93=>92,
94=>93,
95=>94,
96=>95,
97=>96,
98=>97,
99=>98,
100=>99,//如果是递归的方法 到这一层的时候 就报错了 可以注释该层
);
?>

以往算法如下:

<?php
function get_list($tree,$id=0)
{
static $list=array();//定义静态数组存放子孙
foreach ($tree as $key => $v)
{
if($v==$id)
{
$list[]=$key;//把子孙放入数组中
unset($tree[$key]);//去掉这个数组的键 避免无效循环
get_list($tree,$key);//再次调用自身，把子节点变成父节点去寻找子孙
}
}
return $list;//返回数组
}
?>

结果如图



去掉最后一层的那个元素 就可以获取了。

这说明这样的算法在遇到节点层级非常深的时候会崩溃，例如超过100层，php会报错误的。后来改进了一种算法如下：

<?php
$id=2;

$new_array=array();//存放子孙的数组
$id_list[]=$id;//存放父亲的数组
//遍历所有子孙数据 每次去取已经有的数据的子孙 并把它加入父亲数组
for ($i=1; $i <= count($tree); $i++)
{
foreach ($tree as $key => $v)
{
if( in_array($v,$id_list) )
{
$new_array[$key]=$v;//把我放进子孙数组
$id_list[$key]=$key;//把子孙也放入父亲数组
unset($tree[$key]);//这个已经用过了可以回收了避免无效循环
}
}
}
$son_list=array_keys($new_array);//拿到存放子孙的数组 取键即可
print_r($son_list);
exit();
?>