在Java中,我們可以通過構建樹節點對象來將一個List轉換為樹結構。下面是一個示例代碼:
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
class TreeNode {
int val;
List<TreeNode> children;
public TreeNode(int val) {
this.val = val;
this.children = new ArrayList<>();
}
}
public class ListToTree {
public static TreeNode listToTree(List<Integer> list) {
Map<Integer, TreeNode> map = new HashMap<>();
TreeNode root = null;
for (Integer val : list) {
TreeNode node = new TreeNode(val);
map.put(val, node);
}
for (Integer val : list) {
TreeNode node = map.get(val);
if (val == 0) {
root = node;
} else {
TreeNode parent = map.get((val - 1) / 2);
parent.children.add(node);
}
}
return root;
}
public static void main(String[] args) {
List<Integer> list = new ArrayList<>();
list.add(0);
list.add(1);
list.add(2);
list.add(3);
list.add(4);
list.add(5);
TreeNode root = listToTree(list);
}
}
在這段代碼中,我們首先創建一個TreeNode
類來表示樹節點對象,其中包含節點值和子節點列表。接著,我們通過遍歷給定的List來創建樹節點對象,并將它們存儲在一個Map中,其中鍵為節點值,值為節點對象。然后,我們再次遍歷List,根據節點值的關系構建樹結構,并返回根節點。最后,我們可以通過調用listToTree
方法將一個List轉換為樹結構,并獲取根節點。