来看下本题的题目描述:
本题想让我们将二叉树转换为单链表,可以发现,左边的二叉树转换为右边的链表是使用的中序遍历,根 左 右
所以本道题的思路就是可以先用一个集合以中序遍历收集元素,然后再构建单链表。
class Solution {
List<TreeNode> res = new ArrayList();
public void flatten(TreeNode root) {
calc(root);
//这里i从1开始 需要特别注意
for(int i = 1;i<res.size();i++){
//新建头结点,开始构建链表;只取右子树
TreeNode head = res.get(i-1);
TreeNode cur = res.get(i);
head.right=cur;
head.left=null;
}
}
public void calc(TreeNode node ){
if(node ==null)return;
res.add(node);
calc(node.left);
calc(node.right);
}
}