Java 遞歸遍歷實現linux tree命令方式

Posted on 2021-09-18 by WalkonNet

Java 遞歸遍歷實現linux tree命令

看到介紹java file類的文章，有一個遍歷文件夾的練習，遍歷某個目錄下所有文件，包括子目錄。寫瞭一個用棧實現的遞歸遍歷。

import java.io.File;
import java.util.Stack;
public class TraversalFile {
	
	public static void main(String[] args) {
		File dir = new File("D:/Study/code/Java/TraversalFile");
		traversal(dir,1);
	}
 
	public TraversalFile() {
		// TODO Auto-generated constructor stub
	}
	
	public static void traversal(File file,int depth) {
		printName(file, depth);
		File[] fileArray = file.listFiles();
		
		Stack<File> stack = new Stack<File>();
		for(File f:fileArray) {
			if(f.isFile()) {
				printName(f, depth + 1);				
			}
			else {
				stack.add(f);
			}
		}
		
		while(stack.isEmpty() == false) {
			traversal(stack.pop(),depth + 1);
		}
	}
	
	public static void printName(File f, int signNum) {
		StringBuilder builder = new StringBuilder();
		//signNum個sign
		
		for(int i=0;i<signNum;i++){		   
		   if(i == signNum - 1)
			   builder.append("|--"); 
		   else
			   builder.append("  ");
		}
		
		String str = builder.toString();
		System.out.println(str + f.getName());
	}
}

效果如下

遞歸調用的函數traversal

	public static void traversal(File file,int depth) {
		printName(file, depth);
		File[] fileArray = file.listFiles();
		
		Stack<File> stack = new Stack<File>();
		for(File f:fileArray) {
			if(f.isFile()) {
				printName(f, depth + 1);				
			}
			else {
				stack.add(f);
			}
		}
		
		while(stack.isEmpty() == false) {
			traversal(stack.pop(),depth + 1);
		}
	}

函數首先調用瞭一個printName函數，用來打印當前傳入文件的名字，包括前面的線條，然後遍歷當前文件的子文件，如果是文件類型就打印出來，但是深度+1，深度是用來通過打印字符顯示出文件層次的，如果是目錄就入棧。遍歷結束後就出棧並遞歸調用 traversal，直到所有文件打印完畢。

不使用stack也可以實現遍歷，但是fileArray裡面可能文件和目錄是混著的，顯示出來會比較亂，如果先打印文件，目錄都入棧，之後處理，就可以區別開。

如果不考慮文件和目錄是混著輸出，下面這種寫法更簡潔。

如果是文件就返回，是目錄就遞歸調用。

	public static void traversal(File file,int depth) {
		printName(file, depth);		
		if (file.isFile())
			return;		
		File[] fileArray = file.listFiles();
		for(File f:fileArray) {
			traversal(f, depth + 1);
		}
	}

結果如下

printName函數

	public static void printName(File f, int signNum) {
		StringBuilder builder = new StringBuilder();
		//signNum個sign
		
		for(int i=0;i<signNum;i++){		   
		   if(i == signNum - 1)
			   builder.append("|--"); 
		   else
			   builder.append("  ");
		}
		
		String str = builder.toString();
		System.out.println(str + f.getName());
	}

文件或目錄名前根據層級填充空格，最後用 |–緊接名字。

java實現zTree的遍歷

entity代碼：

public class CategoryVO {
	private Integer id;
	private Integer pId;
	private String name;
	private String url;
	private List<CategoryVO> children=new ArrayList<CategoryVO>();
	//get set 省略
}

數據是這樣的：

在這裡插入圖片描述

實現代碼：

public List<CategoryVO> list(String name,Model model){
		List<CategoryVO> categoryList = CategoryJDBC.getCategoryList();
		
		HashMap<Integer, CategoryVO> tmpMap = new HashMap<>(); // 所有對象存放到map中
		for (CategoryVO categoryVO : categoryList) {
			tmpMap.put(categoryVO.getId(), categoryVO);
		}
		
		ArrayList<CategoryVO> arrayList = new ArrayList<>(); // 結果list，之所以用list，是考慮到有多個根目錄的情況
		for (CategoryVO categoryVO : categoryList) {  // 遍歷所有元素，放到對應的父節點
			if(tmpMap.get(categoryVO.getpId())!=null && categoryVO.getId()!=categoryVO.getpId() ){
				CategoryVO categoryVO2 = tmpMap.get(categoryVO.getpId()); //map中找到父節點
				List<CategoryVO> children = categoryVO2.getChildren(); 
				children.add(categoryVO); // 添加到父節點的children裡
				categoryVO2.setChildren(children); 
				tmpMap.put(categoryVO2.getId(), categoryVO2); //重置添加children後的map
				
			}else{
				arrayList.add(categoryVO);
			}
		}
		return categoryList;
	}

該方法隻用瞭2次遍歷。

第一次，遍歷所有對象，放到tmpMap中。

第二次，遍歷所有對象，通過tmpMap找到每個節點對應的父節點，並添加到父節點children中。然後父節點再放回map。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

Java 遞歸遍歷實現linux tree命令方式

目錄