封装一个通用递归算法，使用TreeIterator和TreeMap来简化你的开发工作。

在实际工作中，你肯定会经常的对树进行遍历，并在树和集合之间相互转换，你会频繁的使用递归。

事实上，这些算法在逻辑上都是一样的，因此可以抽象出一个通用的算法来简化工作。

 

在这篇文章里，我向你介绍，我封装的两个类，TreeIterator和TreeMap，使用他们，你不必再写递归就可以在任意的树和任意的集合之间相互装换。

一 TreeIterator
1.1 TreeIterator功能描述：
TreeIterator封装了对树的遍历算法，他提供了如下功能：
1)遍历树
2)将任意一颗树转换为一个任意集合。
使用TreeIterator只需要一个方法调用就可以完成树的遍历，或者将树映射到一个集合，譬如，你可以使用TreeIterator方便的对一个dom进行遍历，或者将其导出到一个集合中，期间你可以将dom节点自由的映射到任意一个自定义对象。

1.2 TreeIterator优点：

不必再手写递归来遍历树，在实际工作中，这可以极大的节省你的时间，因为在写递归的时候，会经常忽略递归条件而造成死递归。

1.3 TreeIterator缺点：
TreeIterator内部使用了递归，所以会影响到性能。其次，第一次使用TreeIterator时，需要适应一下，一旦使用了几次，你会发现他确实可以减少你的工作量。

1.4 TreeIterator的设计思路：
在实际项目中，可能经常的要对一颗树进行遍历，然后将其导出到一个集合中，通常的做法是手写一个递归。然而，细心一点就会发现，这些递归操作的逻辑大部分都是相同的， 首先，都要获取根节点的所有子节点，然后递归的遍历每一个子节点，并将每一个子节点映射到一个自定义对象中。伪代码如下：

     public void Each(currentNode){
         childNodesOfCurrentNode=get child nodes of currentNode;//获取当前节点的所有子节点。
         foreach(var eachElement in childNodesOfCurrentNode){
            Map(eachElement,target);//将当前子节点映射到target对象中
            Each(eachElement);//然后对每一个子节点递归遍历。
         }
     }

事实上，完全可以将这些重复的逻辑封装起来，然而要想做到一个通用的封住算法，需要解决如下几个问题:
1.如何获取当前节点的子节点。
2.将这个子节点映射到什么类型的对象。
这两个问题都是一个变化点，需要将他们留给使用者来做，因此可以使用委托来将这两个变化点提取成一个参数，这样，就可以让使用者来返回子节点以及将子节点映射到什么类型。

 

1.5 TreeIterator的完整代码：

using System;
using System.Collections.Generic;
using System.Collections;
using System.Text;

namespace Francis {
    public class TreeIteractor {
        /// <summary>
        /// 遍历一个树
        /// </summary>
        /// <typeparam name="TSrc">树节点的类型</typeparam>
        /// <param name="root">树的根节点</param>
        /// <param name="childNodesFilter">返回当前节点的子节点</param>
        /// <param name="callbackThatMapsElementFromSrcToDes">这个节点返回树的当前节点</param>
        public static void Each<TSrc>(TSrc root, Func<TSrc, IEnumerable> childNodesFilter, Action<TSrc> callbackThatMapsElementFromSrcToDes) {
            SelfEach(root, childNodesFilter, callbackThatMapsElementFromSrcToDes);
        }


        public static void Each<TSrc>(TSrc root, Func<TSrc, IEnumerable<TSrc>> childNodesFilter, Action<TSrc> callbackThatMapsElementFromSrcToDes) {
            SelfEach(root, childNodesFilter, callbackThatMapsElementFromSrcToDes);
        }

       
        protected static void SelfEach<TSrc>(TSrc root, Func<TSrc, IEnumerable> childNodesFilter, Action<TSrc> callbackThatMapsElementFromSrcToDes) {
            TSrc specifiedRootWithinSrc = root;

            IEnumerable childElements = childNodesFilter(specifiedRootWithinSrc);
            foreach(var eachElement in childElements as IEnumerable<TSrc> ?? childElements) {
                callbackThatMapsElementFromSrcToDes((TSrc)eachElement);
                SelfEach((TSrc)eachElement, childNodesFilter, callbackThatMapsElementFromSrcToDes);
            }
        }
    }
}

1.6 一个例子
下面这个例子演示了如何遍历一个TreeView菜单：

  List<string> list = new List<String>();

            Francis.TreeIteractor.Each<TreeNode>(
                treeView1.Nodes[0],//treeView1是Winform中的TreeView树菜单
                root => {
                    return root.Nodes;//返回当前节点的子节点
                },
                root => {
                    //root是当前节点，因此，这里你可以将其映射到任意和对象，并将这个对象加入到你的集合中。
                    list.Add(root.Text);
                }
                );

            list.ForEach(element => {
                textBox1.Text +=element+ "
";
            });

2.1 TreeMap功能描述

TreeMap提供了和和TreeIterator相反的功能，他可以将一个任意的集合映射到一个树，这听起来很不可思议：

　　1)以树的方式遍历一个集合
　　2)将一个集合映射到一颗树。

举个例子，List集合中存储了Area对象，而Area对象里面存储了省市的id，name，pid，此时你可能想要遍历List<Area>并将其转换为一个dom存储到xml文件中，或者，你要将它显示到一个TreeView菜单中。完成这项工作，你需要写不少的代码，而使用TreeMap只需要一个方法的调用就OK了。

2.2 TreeMap优点：
同样，他可以减少你的工作量，这也是使用它的最大原因。

2.3 TreeMap缺点：
他比TreeIterator更加复杂，需要多使用几次从而适应他。
在使用TreeMap时需要自己制定一个条件用于退出递归。

2.4 TreeMap设计思路：
怎么将一个任意的集合转换为一个任意的树呢，我想首先应该在集合中指定一个根，然后还要指定一个筛选条件，用来筛选当前根的子节点，在筛选的过程中还需要构造树，这听起来很复杂。

2.5 TreeMap完整代码：

using System;
using System.Collections.Generic;
using System.Collections;

namespace Francis.Common{
   public class TreeMap<TSrc> {
        #region fields
        IEnumerable<TSrc> _src;
        IEnumerable _unGenericSrc;
        TSrc _specifiedSrcRoot;
        Func<TSrc,bool> _rootFilter;
       #endregion

        #region property
        protected TSrc SpecifiedRoot {
            get {
                if(_specifiedSrcRoot == null) {
                    bool finded = false;
                    foreach(var eachElement in _src) {
                        if(_rootFilter(eachElement)) {
                            _specifiedSrcRoot = eachElement;
                            finded = true;
                            break;
                        }
                    }
                    if(!finded)
                        throw new ArgumentException("没有找到指定的源根节点，请仔细检查rootFilter的内部逻辑！");
                }
                return _specifiedSrcRoot;
            }
        }

        #endregion property 
       
        #region initialization
        public TreeMap(IEnumerable<TSrc> src,Func<TSrc,bool>rootFilter)
            :this(src, default(TSrc), rootFilter){
        }

        public TreeMap(IEnumerable<TSrc> src, TSrc specifiedSrcRoot)
            :this(src, specifiedSrcRoot, null){
        }

        public TreeMap(IEnumerable src, TSrc specifiedSrcRoot)
            :this(src, specifiedSrcRoot, null){
        }
        
        protected TreeMap(IEnumerable src, TSrc specifiedSrcRoot,Func<TSrc,bool>rootFilter) {
            if(src == null)
                throw new ArgumentException("src不能为空");
            if(specifiedSrcRoot == null&&rootFilter==null)
                throw new ArgumentException("specifiedSrcRoot和rootFilter不能同时为空");

            _src = src as IEnumerable<TSrc>;
            _unGenericSrc = src;
            _specifiedSrcRoot = specifiedSrcRoot;
            _rootFilter = rootFilter;
        }

        #endregion

        #region OtherToTree

        /// <summary>
        /// 将一个源集合映射到一棵树
        /// </summary>
        /// <param name="isChildNodeOfSpecifiedRoot">判断源集合中的节点是否为指根节点的子节点</param>
        /// <param name="callbackThatMapsElementFromSrcToDes">将源集合中的元素映射到一个指定的元素</param>
        public void Each<TDes>(Func<TSrc, TSrc, bool> isChildNodeOfSpecifiedRoot, Func<TSrc, TDes,TDes> callbackThatMapsElementFromSrcToDes) {
            OtherToTree(SpecifiedRoot, default(TDes), isChildNodeOfSpecifiedRoot, callbackThatMapsElementFromSrcToDes);
        }


        protected virtual void OtherToTree<TDes>(TSrc srcRoot, TDes des, Func<TSrc, TSrc, bool> isChildNodeOfSpecifiedRoot, Func<TSrc, TDes, TDes> callbackThatMapsElementFromSrcToDes) {
            des= callbackThatMapsElementFromSrcToDes(srcRoot,des);
       
            foreach(var currentElement in _src != null ? _src : _unGenericSrc) {
                if(isChildNodeOfSpecifiedRoot((TSrc)currentElement, srcRoot)) {
                    OtherToTree((TSrc)currentElement, des, isChildNodeOfSpecifiedRoot, callbackThatMapsElementFromSrcToDes);
                }
            }
        }

        /// <summary>
        /// 对源集合以树的方式进行遍历。
        /// </summary>
        /// <param name="isChildNodeOfSpecifiedRoot">删选子节点</param>
        /// <param name="callbackThatMapsElementFromSrcToDes">为你返回根节点</param>
        public void Each(Func<TSrc, TSrc, bool> isChildNodeOfSpecifiedRoot, Action<TSrc> callbackThatMapsElementFromSrcToDes) {
            OtherToTree(SpecifiedRoot, isChildNodeOfSpecifiedRoot, callbackThatMapsElementFromSrcToDes);
        }

        private void OtherToTree(TSrc SpecifiedRoot, Func<TSrc, TSrc, bool> isChildNodeOfSpecifiedRoot, Action<TSrc> callbackThatMapsElementFromSrcToDes) {
            callbackThatMapsElementFromSrcToDes(SpecifiedRoot);

            foreach(var currentElement in _src != null ? _src : _unGenericSrc) {
                if(isChildNodeOfSpecifiedRoot((TSrc)currentElement, SpecifiedRoot)) {
                    OtherToTree((TSrc)currentElement, isChildNodeOfSpecifiedRoot, callbackThatMapsElementFromSrcToDes);
                }
            }
        }
        #endregion   
    }
}

2.6 一个例子：
下面这个例子演示了如何将一个List<Area>对象显示到TreeView树菜单中。

class Area{
    public int Id{get;set;}
    public string Name{get;set;}
    public int PId{get;set;}
}
//Winform窗体的Load事件。
public void Form1_Load(object sender,EventArgs e){
    List<Area> areas=new List<Area>(){
        new Area(){Id=1,Name="北京",PId=0},
        new Area(){Id=2,Name="海淀区",PId=1},
        new Area(){Id=3,Name="朝阳区",PId=1},
        new Area(){Id=4,Name="东城区",PId=1}
    };
    
    TreeMap<Area,TreeNode> treeMap=new TreeMap<Area,TreeNode>();
    treeMap.Each(
        ()
    );
      //将数据从List<Node>_src加载到TreeView中。这里你不再需要写复杂的遍历算法
            Area srcRoot = new Area() { Name = "china", Id = 0 };

            TreeMap<Area> areaToTreeNode = new TreeMap<Area>(areas, srcRoot);

            areaToTreeNode.Each<TreeNode>(
                //筛选条件用于筛选根节点，需要的是，要自己指定一个返回false的值，用于退出递归，否则将会出现死递归。
                (eachElementWithinSrc, specifiedRoot) => {
                    //递归条件。
                    if(eachElementWithinSrc.Id == specifiedRoot.Id)
                        return false;
                    return eachElementWithinSrc.PId == specifiedRoot.Id;
                },
                //将Area映射到TreeNode
                (srcElement, root) => {
                    //首先判断，如果根节点等于null，就创建一个根节点。
                    if(root == null) {
                        root = treeView1.Nodes.Add("china");
                        return root;
                    }

                    //创建一个新节点
                    TreeNode newNodeAsRoot = new TreeNode(srcElement.Name);
                    newNodeAsRoot.Tag = srcElement;

                    //将新节点添加到root下，使其成为root的子节点。
                    root.Nodes.Add(newNodeAsRoot);

                    //返回新创建的节点。
                    return newNodeAsRoot;
                }
             );
}