二叉树算法题总结（1）

二叉树算法题总结（1）

一.前言

树的问题永远逃不开树的递归遍历框架：

/* 二叉树遍历框架 */
void traverse(TreeNode root) {
    // 前序遍历，对根节点的操作位置在这里
    traverse(root.left)
    // 中序遍历，对根节点的操作位置在这里
    traverse(root.right)
    // 后序遍历，对根节点的操作位置在这里
}

二.二叉树的重要性

举个例子，经典排序算法：【快速排序】和【归并排序】；快速排序就是个二叉树的前序遍历，归并排序就是二叉树的后序遍历。

快排的逻辑：对数组nums排序，先找一个分界点p，通过交换元素使得nums[lo…p-1]都小于等于nums[p],且 nums[p+1…hi] 都大于 nums[p]，然后递归地去 nums[lo…p-1] 和 nums[p+1…hi] 中寻找新的分界点，最后整个数组就被排序了。

void sort(int[] nums, int lo, int hi) {
    /****** 前序遍历位置 ******/
    // 通过交换元素构建分界点 p
    int p = partition(nums, lo, hi);
    /************************/

    sort(nums, lo, p - 1);
    sort(nums, p + 1, hi);
}

先构造分界点，再去左右子数组构造分界点，这就是二叉树的前序遍历。

归并排序的逻辑：先对nums[lo…mid]排序，再对nums[mid…hi]排序，最后把两个有序子数组合并。

void sort(int[] nums, int lo, int hi) {
    int mid = (lo + hi) / 2;
    sort(nums, lo, mid);
    sort(nums, mid + 1, hi);

    /****** 后序遍历位置 ******/
    // 合并两个排好序的子数组
    merge(nums, lo, mid, hi);
    /************************/
}

先左右，再合并（类似有序链表），这就是二叉树的后序遍历框架，也可以看做是分治算法。

三.写递归算法的技巧

写递归算法的关键是要明确函数的「定义」是什么，然后相信这个定义，利用这个定义推导最终结果，绝不要跳入递归的细节。

写树相关的算法，简单说就是，先搞清楚当前 root 节点该做什么，然后根据函数定义递归调用子节点，递归调用会让孩子节点做相同的事情。

四.算法实践

I.翻转二叉树

输入一个二叉树节点root，把整棵树镜像翻转

void traverse(TreeNode root){
	if(root == null)
		return null;
	traverse(root.left);
	traverse(root.right);
	TreeNode tmp = root.left;
	root.left = root.right;
	root.right = tmp;
	return root;
}

II.填充二叉树节点的右侧指针

Node connect(Node root){
	if(root == null || root.left==null)
		return root;
	root.left.next = root.right;
	connect(root.left);
	connect(root.right);
	return root;
}

以上做法只能够实现每个子树下左节点指向右节点，不同子树下做不到。

因此，可以设计一个辅助函数，任务【将每一层二叉树节点连接起来】可以细化为【将每两个相邻的节点连接起来】

Node connect(Node root){
	if(root ==null)
		return null;
	connectwo(root.left,root.right);
	return root;
	}
void connectwo(Node n1,Node n2){
	if(n1 == null || n2==null)
		return;
	n1.next =n2;
	connectwo(n1.left,n1.right);
	connectwo(n2.left,n2.right);
	connectwo(n1.right,n2.left);
}

III.将二叉树展开为链表

void flatten(TreeNode root){
	if(root == null) return;
	flatten(root.left);
	flatten(root.right);
	TreeNode left = root.left;
	TreeNode right = root.right;
	root.left =null;
	root.right = left;
	TreeNode p =root;
	while(p.right !=null){
		p=p.right;
	}
	p.right =right;
}

五.总结

递归算法的关键是要明确函数的定义，利用这个定义，而不要跳进递归的细节中去。

写二叉树的算法题，都是基于递归的，要先明确root在干什么（也就是函数的定义），再确定使用哪种遍历。

二叉树的难点在于弄明白每个节点在做什么，这个需要通过刷题来积累。