迭代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;
    }
}