这篇文章主要介绍了php实现的树形结构数据存取类,实例演示了以树形数据结构存取数据的实现方法,对于学习基于PHP的数据结构有一定的参考借鉴价值,需要的朋友可以参考下

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

具体实现代码如下：

1. <?php
2. /**
3. * Tanphp framework
4. *
5. *
6. * @category Tanphp
7. * @package Data_structure
8. * @version $Id: Tree.php 25024 2012-11-26 22:22:22 tanbo $
9. */
10. /**
11. * 树形结构数据存取类
12. *
13. * 用于对树形结构数据进行快速的存取
14. *
15. * @param array $arr 参数必须为标准的二维数组，包含索引字段（id）与表示树形结构的字段(path),如example中所示
16. *
17. * @example <code>
18. * $arr = array(
19. * array( 'id' => 1, 'name' => 'php', 'path' => '1' ),
20. * array( 'id' => 3, 'name' => 'php1', 'path' => '1-3' ),
21. * array( 'id' => 2, 'name' => 'mysql', 'path' => '2' ),
22. * array( 'id' => 6, 'name' => 'mysql1', 'path' => '2-6' ),
23. * array( 'id' => 7, 'name' => 'mysql2', 'path' => '2-7' ),
24. * array( 'id' => 5, 'name' => 'php11', 'path' => '1-3-5' ),
25. * array( 'id' => 4, 'name' => 'php2', 'path' => '1-4' ),
26. * );
27. * $cate = new Tree($arr);
28. *
29. * $data = $cate->getChild(2);
30. *
31. * print_r($data->toArray());
32. * </code>
33. *
34. */
35. class Tree
36. {
37. public $_info; //节点信息
38. public $_child = array(); //子节点
39. private $_parent; //父节点
40. private $_data; //当前操作的临时数据
41. private static $_indexs = array(); //所有节点的索引
42. private static $_index_key = 'id'; //索引键
43. private static $_tree_key = 'path'; //树形结构表达键
44. private static $_tree_delimiter = '-'; //属性结构表达分割符
45. /**
46. * 构造函数
47. *
48. * @param array $arr
49. * @param boole $force_sort 如果为真，将会强制对$arr 进行排序
50. * @return void
51. */
52. public function __construct(array $arr = array(), $force_sort=true)
53. {
54. if ($force_sort === true) {
55. $arr=$this->_array_sort($arr, self::$_tree_key);
56. }
57. if (!emptyempty($arr)) {
58. $this->_init($arr);
59. }
60. }
61. /**
62. * 初始存储树形数据
63. *
64. * @param array $arr
65. * @return void
66. */
67. private function _init(array $arr)
68. {
69. foreach ($arr as $item) {
70. $path = $item[self::$_tree_key];
71. $paths = explode(self::$_tree_delimiter, $path);
72. $count_paths = count($paths);
73. $parent_id = isset($paths[$count_paths-2]) ? $paths[$count_paths-2] : NULL;
74. if ( $count_paths>1 //如果有父级
75. && array_key_exists($parent_id, self::$_indexs) //父级已经被存入索引
76. && self::$_indexs[$parent_id] instanceof Tree //父级为Tree对象
77. ) {
78. self::$_indexs[$parent_id]->addChild($item);
79. } elseif ($count_paths == 1) {
80. $this->addChild($item);
81. } else {
82. throw new Exception("path数据错误".var_export($item, true));
83. }
84. }
85. //print_r(self::$_indexs);
86. }
87. /**
88. * 添加子节点
89. *
90. * @param array $item
91. * @return void
92. */
93. public function addChild(array $item, $parent = NULL)
94. {
95. $child = new Tree();
96. $child->_info = $item;
97. $child->_parent = $parent == NULL ? $this : $parent;
98. $child->_parent->_child[] = $child;
99. $this->_addIndex($item, $child->_getSelf());
100. }
101. /**
102. * 添加节点到索引
103. *
104. * @param array $item
105. * @param mix $value
106. * @return void
107. */
108. private function _addIndex(array $item, $value)
109. {
110. if (array_key_exists(self::$_index_key, $item) && is_int($item[self::$_index_key])) {
111. self::$_indexs[$item[self::$_index_key]] = $value;
112. } else {
113. throw new Exception("id字段不存在或者不为字符串");
114. }
115. }
116. /**
117. * 获取对自己的引用
118. *
119. * @return Tree object quote
120. */
121. private function _getSelf()
122. {
123. return $this;
124. }
125. /**
126. * 获取指定id的节点的子节点
127. *
128. * @param int $id
129. * @return Tree object
130. */
131. public function getChild($id)
132. {
133. $data = self::$_indexs[$id]->_child;
134. $this->_data = $data;
135. return $this;
136. }
137. /**
138. * 获取指定id的节点的父节点
139. *
140. * @param int $id
141. * @return Tree object
142. */
143. public function getParent($id)
144. {
145. $data = self::$_indexs[$id]->_parent;
146. $this->_data = $data;
147. return $this;
148. }
149. /**
150. * 获取指定id的节点的同级节点
151. *
152. * @param int $id
153. * @return Tree object
154. */
155. public function getBrother($id)
156. {
157. $data = self::$_indexs[$id]->_parent->_child;
158. $this->_data = $data;
159. return $this;
160. }
161. /**
162. * 将Tree对象转化为数组
163. *
164. * @param object $object
165. * @return array
166. */
167. public function toArray($obj = NULL)
168. {
169. $obj = ($obj === NULL) ? $this->_data : $obj;
170. $arr = array();
171. $_arr = is_object($obj) ? $this->_getBaseInfo($obj) : $obj;
172. if (is_array($_arr)) {
173. foreach ($_arr as $key => $val){
174. $val = (is_array($val) || is_object($val)) ? $this->toArray($val) : $val;
175. $arr[$key] = $val;
176. }
177. } else {
178. throw new Exception("_arr不是数组");
179. }
180. return $arr;
181. }
182. /**
183. * 过滤_parent等字段，以免造成无限循环
184. *
185. * @param object $obj
186. * @return void
187. */
188. private function _getBaseInfo($obj)
189. {
190. $vars = get_object_vars($obj);
191. $baseInfo['_info'] = $vars['_info'];
192. $baseInfo['_child'] = $vars['_child'];
193. return $baseInfo;
194. }
195. /**
196. * 二维数组排序
197. *
198. * 根据指定的键名对二维数组进行升序或者降序排列
199. *
200. * @param array $arr 二维数组
201. * @param string $keys
202. * @param string $type 必须为 asc或desc
203. * @throws 当参数非法时抛出异常
204. * @return 返回排序好的数组
205. */
206. private function _array_sort(array $arr, $keys, $type = 'asc') {
207. if (!is_string($keys)) {
208. throw new Exception("非法参数keys:参数keys的类型必须为字符串");
209. }
210. $keysvalue = $new_array = array();
211. foreach ($arr as $k=>$v) {
212. if (!is_array($v) || !isset($v[$keys])) {
213. throw new Exception("参数arr不是二维数组或arr子元素中不存在键'{$keys}'");
214. }
215. $keysvalue[$k] = $v[$keys];
216. }
217. switch ($type) {
218. case 'asc':
219. asort($keysvalue);
220. break;
221. case 'desc':
222. arsort($keysvalue);
223. break;
224. default:
225. throw new Exception("非法参数type ：参数type的值必须为 'asc' 或 'desc'");
226. }
227. reset($keysvalue);
228. foreach ($keysvalue as $k=>$v) {
229. $new_array[$k] = $arr[$k];
230. }
231. return $new_array;
232. }
233. }
234. ?>