php实现的树形结构数据存取类实例

本文实例讲述了php实现的树形结构数据存取类。分享给大家供大家参考。

具体实现代码如下:

代码如下:

<?php 

/** 

 * Tanphp framework 

 * 

 * 

 * @category   Tanphp 

 * @package    Data_structure 

 * @version    $Id: Tree.php 25024 2012-11-26 22:22:22 tanbo $ 

 */ 

 

/** 

 * 树形结构数据存取类 

 *  

 * 用于对树形结构数据进行快速的存取 

 *  

 * @param array $arr 参数必须为标准的二维数组,包含索引字段(id)与表示树形结构的字段(path),如example中所示 

 *  

 * @example <code> 

 * $arr = array( 

 *  array( 'id' => 1, 'name' => 'php', 'path' => '1' ), 

 *  array( 'id' => 3, 'name' => 'php1', 'path' => '1-3' ), 

 *  array( 'id' => 2, 'name' => 'mysql', 'path' => '2' ), 

 *  array( 'id' => 6, 'name' => 'mysql1', 'path' => '2-6' ), 

 *  array( 'id' => 7, 'name' => 'mysql2', 'path' => '2-7' ), 

 *  array( 'id' => 5, 'name' => 'php11', 'path' => '1-3-5' ), 

 *  array( 'id' => 4, 'name' => 'php2', 'path' => '1-4' ), 

 *   ); 

 *  $cate = new Tree($arr); 

 *   

 *  $data = $cate->getChild(2); 

 *   

 *  print_r($data->toArray()); 

 * </code> 

 *  

 */ 

class Tree 

{ 

    public  $_info;                             //节点信息 

    public  $_child = array();                  //子节点 

    private $_parent;                           //父节点 

    private $_data;                             //当前操作的临时数据 

    private static $_indexs         = array();  //所有节点的索引 

    private static $_index_key      = 'id';     //索引键 

    private static $_tree_key       = 'path';   //树形结构表达键 

    private static $_tree_delimiter = '-';      //属性结构表达分割符 

     

    /** 

     * 构造函数 

     *  

     * @param array $arr 

     * @param boole $force_sort 如果为真,将会强制对$arr 进行排序 

     * @return void 

     */ 

    public function __construct(array $arr = array(),  $force_sort=true) 

    { 

        if ($force_sort === true) { 

            $arr=$this->_array_sort($arr, self::$_tree_key); 

        } 

        if (!emptyempty($arr)) { 

            $this->_init($arr); 

        } 

    } 

     

    /** 

     * 初始存储树形数据 

     *  

     * @param array $arr 

     * @return void 

     */ 

    private function _init(array $arr) 

    { 

        foreach ($arr as $item) { 

            $path        = $item[self::$_tree_key]; 

            $paths       = explode(self::$_tree_delimiter, $path); 

            $count_paths = count($paths); 

            $parent_id   = isset($paths[$count_paths-2]) ? $paths[$count_paths-2] : NULL; 

             

            if (   $count_paths>1                                   //如果有父级 

                && array_key_exists($parent_id, self::$_indexs)      //父级已经被存入索引 

                && self::$_indexs[$parent_id] instanceof Tree    //父级为Tree对象 

            ) { 

                self::$_indexs[$parent_id]->addChild($item); 

            } elseif ($count_paths == 1) { 

                $this->addChild($item); 

            } else { 

                throw new Exception("path数据错误".var_export($item, true)); 

            } 

        } 

         

        //print_r(self::$_indexs); 

    } 

     

    /** 

     * 添加子节点 

     *  

     * @param array $item 

     * @return void 

     */ 

    public function addChild(array $item, $parent = NULL) 

    { 

        $child          = new Tree(); 

        $child->_info   = $item; 

        $child->_parent = $parent == NULL ? $this : $parent; 

        $child->_parent->_child[] =  $child; 

         

        $this->_addIndex($item, $child->_getSelf());  

    } 

     

    /** 

     * 添加节点到索引 

     *  

     * @param array $item 

     * @param mix $value 

     * @return void 

     */ 

    private function _addIndex(array $item, $value) 

    { 

        if (array_key_exists(self::$_index_key, $item) && is_int($item[self::$_index_key])) { 

            self::$_indexs[$item[self::$_index_key]] = $value; 

        } else { 

            throw new Exception("id字段不存在或者不为字符串"); 

        } 

    } 

     

    /** 

     * 获取对自己的引用 

     *  

     * @return Tree object quote 

     */ 

    private function _getSelf() 

    { 

        return $this; 

    } 

     

    /** 

     * 获取指定id的节点的子节点 

     *  

     * @param int $id 

     * @return Tree object 

     */ 

    public function getChild($id) 

    { 

        $data       = self::$_indexs[$id]->_child; 

        $this->_data = $data; 

        return $this; 

    } 

     

    /** 

     * 获取指定id的节点的父节点 

     *  

     * @param int $id 

     * @return Tree object 

     */ 

    public function getParent($id) 

    { 

        $data = self::$_indexs[$id]->_parent; 

        $this->_data = $data; 

        return $this; 

    } 

     

    /** 

     * 获取指定id的节点的同级节点 

     * 

     * @param int $id 

     * @return Tree object 

     */ 

    public function getBrother($id) 

    { 

        $data = self::$_indexs[$id]->_parent->_child; 

        $this->_data = $data; 

        return $this; 

    } 

     

    /** 

     * 将Tree对象转化为数组 

     *  

     * @param  object $object 

     * @return array 

     */ 

     public function toArray($obj = NULL) 

     { 

        $obj  = ($obj === NULL) ? $this->_data : $obj; 

        $arr  = array(); 

        $_arr = is_object($obj) ? $this->_getBaseInfo($obj) : $obj; 

         

        if (is_array($_arr)) { 

            foreach ($_arr as $key => $val){ 

                 

                $val = (is_array($val) || is_object($val)) ? $this->toArray($val) : $val;

                $arr[$key] = $val; 

            } 

        } else { 

            throw new Exception("_arr不是数组"); 

        } 

      

        return $arr; 

    } 

     

    /** 

     * 过滤_parent等字段,以免造成无限循环 

     *  

     * @param object $obj 

     * @return void 

     */ 

    private function _getBaseInfo($obj) 

    { 

        $vars = get_object_vars($obj); 

        $baseInfo['_info']  =  $vars['_info']; 

        $baseInfo['_child'] =  $vars['_child']; 

        return $baseInfo; 

    } 

     

    /** 

     * 二维数组排序 

     * 

     * 根据指定的键名对二维数组进行升序或者降序排列 

     * 

     * @param array  $arr 二维数组 

     * @param string $keys 

     * @param string $type 必须为 asc或desc 

     * @throws 当参数非法时抛出异常 

     * @return 返回排序好的数组 

     */ 

    private function _array_sort(array $arr, $keys, $type = 'asc') { 

        if (!is_string($keys)) { 

            throw new Exception("非法参数keys:参数keys的类型必须为字符串"); 

        } 

     

        $keysvalue = $new_array = array(); 

        foreach ($arr as $k=>$v) { 

            if (!is_array($v) || !isset($v[$keys])) { 

                throw new Exception("参数arr不是二维数组或arr子元素中不存在键'{$keys}'"); 

            } 

            $keysvalue[$k] = $v[$keys]; 

        } 

     

        switch ($type) { 

            case 'asc': 

                asort($keysvalue); 

                break; 

            case 'desc': 

                arsort($keysvalue); 

                break; 

            default: 

                throw new Exception("非法参数type :参数type的值必须为 'asc' 或 'desc'"); 

        } 

     

        reset($keysvalue); 

        foreach ($keysvalue as $k=>$v) { 

            $new_array[$k] = $arr[$k]; 

        } 

        return $new_array; 

    } 

} 

?>

希望本文所述对大家的PHP程序设计有所帮助。

相关推荐