面对分治算法,看这两道题就够了
分治算法
分治,"分而治之"。从字面上理解就是分---治,把大的问题分成小问题,解决一个一个小问题,之后把问题的答案合并起来,就得到大问题的结果。您肯定会在想,这思想这么简单,你不说我也是知道。历史上,秦国通过远交近攻的策略,逐个击破,最后统一六国不也是分治思想的体现吗?
以下用一个二叉树的前序遍历为例,对分治思想在代码上的体现进行说明。
public class PreoderTraversal {
public class TreeNode{
private int val;
public TreeNode left,right;
public TreeNode(int val){
this.val = val;
this.left = this.right = null;
}
}
public ArrayList<Integer> preodertraversal(TreeNode root){
ArrayList<Integer> result = new ArrayList<Integer>();
//退出的条件
if(root == null){
return null;
}
//分:左子树与右子树
ArrayList<Integer> left = preodertraversal(root.left);
ArrayList<Integer> right = preodertraversal(root.right);
//治:把得到的结果合并起来
result.add(root.val);
result.addAll(left);
result.addAll(right);
return result;
}
}
上面的过程可以通过一个递推公式来表示
T(n) = 2T(2/n)+O(1)
2T(2/n) 表示 原来的大问题变成两个原来一半的问题
O(1)表示 对二叉树的每个节点只操作一次。
上面的公式可以推出 上面前序遍历的时间复杂度是 O(n)
从以上代码,可以看出,分治算法在代码实现上有以下两点好处:
1.前序遍历的结果可用通过一个函数内的ArrayList返回不需要创建全局变量,来存放结果。
2.对于拆分后的问题,运算量大,采用多并发,多核来处理,也是很容易的。具体结合上面代码来说,对于left、right结果求解,可以分别启用一个线程。
两道题
对于分治的题目很多,为什么选择下面这两道题目呢?因为足够典型,学会了这两道题,我们保证,您在与同事、面试官聊起分治算法的时候,他们会认为您是懂分治算法。
- 接下来祭出第一道题目
分析:
既然我们使用分治来解决,那就看看问题怎么拆分呢?
这道题目中是求两个节点的公共的祖先,很显然,问题的拆分可以依据:两个节点在二叉树的位置来拆分问题:
都在左子树上、都在右子树上、一个边一个、有一个节点就是根节点
一个大的问题拆分四个问题,逐个解决,求出大问题,下面给出 实现代码
public TreeNode getAncestor(TreeNode root,TreeNode node1,TreeNode node2){
if (root == null)
{
return null;
}
//如果有一个节点就是根节点
if(root == node1 || root == node2){
return root;
}
TreeNode left = getAncestor(root.left,node1,node2);
TreeNode right = getAncestor(root.right,node1,node2);
//节点一边一个
if(left == null && right == null)
{
return root;
}
//节点都在左子树
if (left != null) {
return left;
}
//节点都在右子树
if (right != null) {
return right;
}
return null;
}
如果您还不太明白,没关系,对着分析和代码多看几次,就会打通任督二脉的。
- 第二道(这道题,有点小难度)
为什么说这道题有点难度呢?原因在于二叉树上有负值的存在。而且最关键的是题目只是说遍历二叉树,求最大和,并没有说是从哪里出发,如果从根出发就是求:
root---->anyNode 根到任意节点的最大和
明显这题目的意思是
anyNode---->anyNode 任意节点到任意节点的最大和。
采用分治,怎么拆分呢?
为三种情况:左子树、右子树、左子树-->根-->右子树。不明白,没关系,看下图分析。
分析:
代码实现
public class returnType{
int root2any,any2any;
returnType(int root2any,int any2any){
this.root2any = root2any; //存放上面分析的root-->anyNode
this.any2any = any2any; // anyNode-->anyNode
}
}
public returnType maxSum(TreeNode root){
//如果二叉树不存在,直接设置成最小值
if(root == null){
return new returnType(Integer.MIN_VALUE,Integer.MIN_VALUE);
}
returnType left = maxSum(root.left);
returnType right = maxSum(root.right);
//结合上面的图就是求A+B大还是A+C大呢,
和0做比较就是因为有负数的存在
int root2any =Math.max(0,Math.max(left.root2any,right.root2any))+root.val;
//R=Math.max(D,E)
int any2any = Math.max(left.any2any,right.any2any);
//Math.max(R,A+B+C)
any2any = Math.max(any2any,Math.max(0,left.root2any)+Math.max(0,right.root2any)+ root.val);
return new returnType(root2any,any2any);
}
小福利
分治算法其实在最初的快排和归并排序都接触过了,如果你上面两道题目都理解,下面给出归并排序和快排的代码在重温一下,看下感觉是不是so easy!!
归并排序
private static Comparable[] aux;
public static void sort(Comparable[] list){
aux = new Comparable[list.length];
sort(list,0,list.length-1);
}
public static void sort(Comparable[] list,int lo,int hi){
if(lo < hi){
return;
}
int mid = lo +(hi-lo)/2;
//分
sort(list,lo,mid);
sort(list,mid+1,hi);
//治
meger(list,lo,mid,hi); //这个是归并的具体具体过程,我们这篇介绍分治的重点,在此忽略了
}
快速排序
public static void sort(Comparable[] list){
Collections.shuffle(list); //消除输入的影响
sort(list,0,list.length-1);
}
public static void sort(Comparable[] list,int lo,int hi){
if(lo < hi){
return;
}
int j = patition(list,lo,hi); //快排中重要的切分,典型有三取样切分。找出大小为中间的点
在此忽略了具体实现,有兴趣看相关资料
//分
sort(list,lo,j-1);
sort(list,j+1,hi);
}
快排和归并排序的可以归纳的递推公式
T(n) = 2T(2/n) +O(n)
时间复杂度是 )O(NlogN)