遍历的关键点不是遍历思想,而是如何去处理某个节点。
这里采用堆存储,也就是先进后出的存储模式,具体代码为每次处理list最后一个节点。
处理方式为:如果该节点不存在子文件夹,打印该节点的所有文件;如果该节点存在子文件夹,删除该节点的最后一个子文件夹后,将该节点重新入堆,然后将子节点入堆。
以下为具体代码:
package workFiles; import java.util.LinkedList; /** * 模拟文件类,包括文件夹和文件 */ public class File { private LinkedList<File> folders; private LinkedList<String> files; public LinkedList<File> getFolders() { return folders; } public void setFolders(LinkedList<File> folders) { this.folders = folders; } public LinkedList<String> getFiles() { return files; } public void setFiles(LinkedList<String> files) { this.files = files; } }
package workFiles; import java.util.LinkedList; public class WalkTree { public static void main(String[] a){ LinkedList<File> list = new LinkedList<>(); File f = createTree(); list.add(f); while(list.size()!=0){ readList(list); } } //处理堆 static LinkedList<File> readList(LinkedList<File> list){ //获取栈中最后的节点 File f = list.removeLast(); //如果当前节点存在文件夹 if(f.getFolders()!=null&&f.getFolders().size()!=0){ //去掉当前节点的文件夹 File next = f.getFolders().removeLast(); //重新存入当前节点 list.add(f); //存入新节点 list.add(next); } else{ if(f.getFiles()!=null&&f.getFiles().size()!=0){ for(int i=0;i<f.getFiles().size();i++){ System.out.println(f.getFiles().get(i)); } } } return list; } //模拟创建文件夹 static File createTree(){ File A =new File(); File B =new File(); File C =new File(); LinkedList<String> str1 = new LinkedList<String>(); str1.add("A文件1"); A.setFiles(str1); LinkedList<String> str2 = new LinkedList<String>(); str2.add("B文件1"); str2.add("B文件2"); B.setFiles(str2); LinkedList<String> str3 = new LinkedList<String>(); str3.add("C文件1"); str3.add("C文件2"); str3.add("C文件3"); C.setFiles(str3); LinkedList<File> temp = new LinkedList<File>(); temp.add(B); temp.add(C); A.setFolders(temp); return A; } }