迭代N叉树结构的指定Parent节点的所有Son节点ID

jerry thinkphp 2015年11月19日收藏

本人在给客户开发一款直销软件的时候，需要计算差额业绩，也就是计算N叉树结构的任意父节点以下的所有子节点的数据。
常规的递归是循环的方式去循环的方式去嵌套，但是在实际过程中，这样的方法可能效率没有那么高，那么，本人花了两天时间采用了另外一种快速递归的方式，可以降低递归的次数。

思路是：采用parent数组的也许能降低迭代的次数，同时更能针对性的读取需要的数组数据，避免毫无意义的迭代，大大增加的方法的使用效率。
废话不多说，直接上代码。

        public function monthDiff2(){

        // 导入一个店主记录

        $info = M('member')->where(array('uid'=>UID))->find();

        // 迭代法计算该店主的直接招商的所有IDs

        $res = M('member')->field('uid,parent,nickname,sons')->select();

        foreach($res as $key=>$vo){

            // 方法一

            $data[$vo['uid']]['id'] = $vo['uid'];

            $data[$vo['uid']]['pid'] = $vo['parent'];

            // 方法二

            $parent[$vo['parent']][$vo['uid']] = $vo['uid'];

        }

        $func = 2;

        $pid = 1;

        import("@.Library.VMSExtends");

        $obj = new \VMSExtends($host,$dbuser,$dbpwd,$dbname);

        if($func==1){

            $resTree = $obj->tree($data,$pid); 

            foreach($resTree as $key){

                $list[$key['id']] = $key['id'];    

                asort($list);

            }

        }else{

            $resTree = $obj->treeParent($parent,$pid);

            asort($resTree);

            $list = $resTree;

        }

        dump($list);

    }

class VMSExtends{

        VMSExtends::$treeList = array(); 清空

        static public $treeList = array(); 

    

    public function __construct(){

    

    }

    static public $flg=0;

    static public function tree(&$data,$pid = 0,$count=0) {

        foreach ($data as $key => $value){

            if($value['pid']==$pid){

                $value['count'] = $count;

                self::$treeList []=$value;

                unset($data[$key]);

                self::tree($data,$value['id'],$count+1);

            }

            //echo '['.self::$flg++.']';

        }

        return self::$treeList;

    }

    static public function treeParent(&$parent,$pid = 0,$count=0) {

        if($parent[$pid]){

            foreach ($parent[$pid] as $key => $value){

                self::$treeList[$key] = $key;

                if(empty($parent[$pid])){

                    unset($parent[$value]);

                }

                self::treeParent($parent,$key,$count+1);

            }

            unset($parent[$pid]);

        }else{

            self::$treeList[$pid] = $pid;

            unset($parent[$pid]);

        }

        echo '['.self::$flg++.']';

        return self::$treeList;

    }

}