C# 使用栈遍历磁盘文件和目录，非传统递归方法。

 

    序：前不久使用DEV的TreeList控件，需要根据输入内容定位到对应的节点，由于树的层数不确定，先前采用递归算法遍历Tree，但总觉得别扭，个人比较反感递归函数，好比一个脱缰的野马，难以控制。理论上说，所有递归算法都可以转换为非递归算法，因此决定不使用递归，上网搜了一些资料，最终成功搞定。

    遍历硬盘文件的方法与TreeList是相似的，核心所在就是使用“栈stack”这个数据结构，先把所有驱动器压栈，比如C、D、E、F、G。然后依次遍历每个驱动器，如G:\，先将其出栈，获取该目录下的所有子目录和子文件，将子目录逐个压栈，子文件直接处理。这样处理后，位于栈顶的就是子目录，它里面包含的还有内容，将栈顶目录出栈，再获取该目录下的所有子目录和子文件，同样再将子目录压栈，如此重复，最终实现所有目录和文件的遍历。

    来点实现的代码吧，在Form上拖放一个TreeView、ImageList，其中ImageList有3个图标，分别用于代表磁盘、文件夹、文件。

        /// <summary>

        /// 生成目录树

        /// </summary>

        private void BuildTree()

        {

            //设置图像列表，并暂停重绘

            treeView1.ImageList = imageList1;

            treeView1.BeginUpdate();

            //存放树节点的栈

            Stack<TreeNode> skNode = new Stack<TreeNode>();

            int imageIndex = 0;

            //添加磁盘列表

            string[] drives = Directory.GetLogicalDrives();

            for (int i = 0; i < drives.Length; i++)

            {

                //每个节点的Text存放目录名，Name存放全路径

                TreeNode node = new TreeNode(drives[i], 0, 0);

                node.Name = drives[i];

                treeView1.Nodes.Add(node);

                skNode.Push(node);

            }

            while (skNode.Count > 0)

            {

                //弹出栈顶目录，并获取路径

                TreeNode curNode = skNode.Pop();

                string path = curNode.Name;



                FileInfo fInfo = new FileInfo(path);

                try

                {

                    if ((fInfo.Attributes & FileAttributes.Directory) != 0)

                    {

                        string[] subDirs = null;

                        string[] subFiles = null;

                        try

                        {

                            //获取当前目录下的所有子目录和文件

                            subDirs = Directory.GetDirectories(path);

                            subFiles = Directory.GetFiles(path);

                        }

                        catch

                        { }

                        if (subDirs != null && subFiles != null)

                        {

                            //目录入栈

                            imageIndex = 1;

                            for (int i = 0; i < subDirs.Length; i++)

                            {

                                string dirName = Path.GetFileName(subDirs[i]);

                                TreeNode dirNode = new TreeNode(dirName, 1, 1);

                                dirNode.Name = subDirs[i];

                                curNode.Nodes.Add(dirNode);

                                skNode.Push(dirNode);

                            }

                            //文件无需入栈

                            imageIndex = 2;

                            for (int i = 0; i < subFiles.Length; i++)

                            {

                                string fileName = Path.GetFileName(subFiles[i]);

                                curNode.Nodes.Add(subFiles[i], fileName, 2);

                            }

                        }

                    }

                }

                catch { }

            }

            treeView1.EndUpdate();

        }

虽然效率不是很明显，但是如果去除维护treeview的话，效率会提升1倍多。