算法通关 - 二叉树遍历 递归&分治

二叉树遍历的三种方式
  • 前序遍历(根左右)


    前序遍历.png
public static class TreeNode{
  int data;
  TreeNode leftChild;
  TreeNode rightChild;
  
  TreeNode(int data){
    this.data = data;
  }
}

public static void preOrder(TreeNode root){
  if(root == null) return;
  System.out.println(root.data);
  preOrder(root.leftChild);
  preOrder(root.rightChild);
}
  • 中序遍历(左根右)


    中序遍历.png
public static void inOrder(TreeNode root){
  if(root == null) return;
  inOrder(root.leftChild);
  System.out.println(root.data);
  inOrder(root.rightChild);
}
  • 后序遍历(左右根)


    后序遍历.png
public static void postOrder(TreeNode root){
  if(root == null) return;
  postOrder(root.leftChild);
  postOrder(root.rightChild);
  System.out.println(root.data);
}

前序遍历就是根在最前面,中序遍历是根在中间,后序遍历则是根在最后。这样就很容易记住了

递归(Recursion)

递归也是一种循环,但它不是通过定义一个for循环,而是通过函数体自己调用自己来实现一种循环的状态。比如计算n的阶乘n! = 1 * 2 * 3....*n 就可以通过递归来做,代码如下:

public int factorial(int n){
  if(n <= 1) return 1;
  return n * factorial(n - 1);
}

递归的模板代码:

def recursion(level,param1,param2,....):
    //递归终止条件,必须要有,否则就是死循环
    if level > MAX_LEVEL:
        print_result
        return
    //在当前的level处理代码逻辑
    procress_data(level,data...)
    //递归调用
    self.recursion(level + 1,p1,....)
    //按照实际情况,如果需要就进行reverse_state的收尾工作
    reverse_state(level)

斐波那契数列:1,1,2,3,5,8,13,21... f (n) = f (n - 1) + f (n - 2)

递归过程如下:


斐波那契数列.png

我们会发现在递归的过程中很多计算是重复的,比如在计算f (5) 和 f (6)的时候都会计算f (4),这样f (4)就计算了两次,所以有时候递归解决问题并不是高效的。因此就出现了很多基于递归的高级算法,比如下面要说的分治。

分治(Divide&Conquer)

分治的思想就是先把一个大的问题分解成一个个的子问题,然后再对子问题进行一一解决,最后将每个子结果合并得到最终的结果。


分治逻辑图.png

例如,给一个全是小写的从a - j 的字符串,把其中的每个字符变成大写。可以通过for循环对每个字符进行大写操作。也可以通过递归,每次把当前的字符变成大写,把剩下的字符放到递归中去。使用分治的话,就是把字符串分成多个子问题,每个子问题是一个字母,我们要做的就是把每个子问题中的单个字母变成大写,最后把结果合并起来。这样的好处就是可以进行并行计算,提高效率,因为每个子问题是互不相关的。


小写字符串转换为大写字符串.png

递归适用于解决的问题是没有所谓的中间结果,也就是没有重复计算。如果有大量重复计算的话,分治的效率可能并不高,这时候就需要考虑更适合的算法,比如动态规划。

分治代码模板:

def divide_conquer(problem,param1,param2,....):
    //递归终止条件,必须要有,否则就是死循环
    if problem is None:
        print_result
        return
    //准备数据,把当前的大问题拆分成子问题
    data = prepare_data(problem)
    subproblems = split_problem(problem,data)
    //对子问题进行处理,得到每一个子问题的子结果
    subresult1 = self.divide_conquer(subproblems[0],p1,....)
    subresult2 = self.divide_conquer(subproblems[1],p1,....)
    subresult3 = self.divide_conquer(subproblems[2],p1,....)
    ...
    
    //对子结果进行合并
    result = process_result(subresult1,subresult2,subresult3...)
1.实现 pow(x, n) ,即计算 x 的 n 次幂函数(LeetCode - 50)

示例如下:

  • -100.0 < x < 100.0
  • n 是 32 位有符号整数
输入: 2.00000, 10
输出: 1024.00000

输入: 2.10000, 3
输出: 9.26100

输入: 2.00000, -2
输出: 0.25000

方法一:暴力法,直接将 x 连乘 n 次,需要考虑 n > 0 和 n < 0的情况

class Solution {
    public double myPow(double x, int n) {
        long N = n;
        if (N < 0) {
            x = 1 / x;
            N = -N;
        }
        double ans = 1;
        for (long i = 0; i < N; i++)
            ans = ans * x;
        return ans;
    }
}
  • 时间复杂度:O(n). 我们需要将 x 连乘 n 次。
  • 空间复杂度:O(1). 我们只需要一个变量来保存最终 x 的连乘结果。

方法二:快速幂算法(递归)

假定我们已经得到了 x ^ (n / 2) 的结果,并且我们现在想得到 x ^ n的结果。我们令 A 是 x ^ (n / 2)的结果,我们可以根据 n 的奇偶性来分别讨论 x ^ n的值。如果 n 为偶数,我们可以用公式 (x ^ n) ^ 2 = x ^ (2 * n)来得到 x ^ n = A * A。如果 n 为奇数,那么 A * A = x ^ (n - 1)。直观上看,我们需要再乘一次 x ,即 x ^ n = A * A * x 。该方法可以很方便的使用递归实现。我们称这种方法为 "快速幂",因为我们只需最多 O(logn) 次运算来得到 x ^ n

class Solution {
    private double fastPow(double x, long n) {
        if (n == 0) {
            return 1.0;
        }
        double half = fastPow(x, n / 2);
        if (n % 2 == 0) {
            return half * half;
        } else {
            return half * half * x;
        }
    }
    public double myPow(double x, int n) {
        long N = n;
        if (N < 0) {
            x = 1 / x;
            N = -N;
        }

        return fastPow(x, N);
    }
}
  • 时间复杂度:O(logn). 每一次我们使用公式 (x ^ n) ^ 2 = x ^ (2 * n), n 都变为原来的一半。因此我们需要至多 O(logn) 次操作来得到结果。
  • 空间复杂度:O(logn). 每一次计算,我们需要存储 x ^ (n / 2) 的结果。 我们需要计算O(logn) 次,所以空间复杂度为O(logn) 。

方法三:快速幂算法(循环)

思路: x^n 等价于 (x * x) ^ (n/2) 【如果n除以2余1就得在后面乘以一个x补上】, 继续等价于 (x^2 * x^2) ^ (n/4) 【如果n除4余1再乘以一个x^2补上】,用纯文字描述就是随着n不断地除以2减小,x不断地平方来扩大,直到n为1结束,这个时候的x就是最后的结果了,但是每次循环都要对n为奇数的情况做一个填补。因为x是以不断平方自己的方式来到达最后的终点,所以复杂度是logN

class Solution {
    public double myPow(double x, int n) {
        long N = n;
        if (N < 0) {
            x = 1 / x;
            N = -N;
        }
        double ans = 1;
        double current_product = x;
        for (long i = N; i > 0; i /= 2) {
            if ((i % 2) == 1) {
                ans = ans * current_product;
            }
            current_product = current_product * current_product;
        }
        return ans;
    }
}

class Solution {
    public double myPow(double x, int n) {
        long N = n;
        if (N < 0) {
            x = 1 / x;
            N = -N;
        }
        double result = 1.0;
        double base = x;
        while (N != 0) {
            //用(N & 1) != 0 来判断二进制位最后位0或1,也就是判断它的奇偶性
            if ((N & 1) != 0){
                result *= base;
            }
            base *= base;
            N >>= 1;
        }
        return result;
    }
}
  • 时间复杂度:O(logn). 对每一个 n 的二进制位表示,我们都至多需要累乘 1 次,所以总的时间复杂度为 O(logn) 。
  • 空间复杂的:O(1). 我们只需要用到 2 个变量来保存当前的乘积和最终的结果 x 。
2.求众数(LeetCode - 169)

给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。

你可以假设数组是非空的,并且给定的数组总是存在多数元素。

方法一:暴力法

遍历整个数组,然后用另一重循环统计每个数字出现的次数。将出现次数超过n/2的元素返回。

class Solution {
    public int majorityElement(int[] nums) {
        int majorityCount = nums.length/2;

        for (int num : nums) {
            int count = 0;
            for (int elem : nums) {
                if (elem == num) {
                    count += 1;
                }
            }

            if (count > majorityCount) {
                return num;
            }

        }

        return -1;    
    }
}
  • 时间复杂度:O(n^2) 包含两重嵌套的 for 循环,每一层 n 次迭代,所以总的是平方级的时间复杂度。
  • 空间复杂度:O(1) 没有分配任何与输入规模成比例的额外的空间

方法二:哈希表

使用哈希表来存储每个元素,然后用一个循环在线性时间内遍历 nums ,我们只需要返回有最大值的键。

class Solution {
    public int majorityElement(int[] nums) {
        int majority = nums.length / 2;
        Map<Integer, Integer> numCount = new HashMap<>();
        for (int num : nums) {
            int count = numCount.getOrDefault(num, 0) + 1;
            if (count > majority) return num;
            numCount.put(num, count);
        }
        return -1;
    }
}
  • 时间复杂度:O(n) nums 迭代一次,哈希表的插入是常数时间的。所以总时间复杂度为 O(n)
  • 空间复杂度:O(n)

方法三:排序

如果所有数字被单调递增或者单调递减的顺序排了序,那么众数的下标为 (n/2) 当 n 是偶数时,下标为 (n/2) + 1

class Solution {
    public int majorityElement(int[] nums) {
        Arrays.sort(nums);
        return nums[nums.length/2];
    }
}
  • 时间复杂度:O(N*logN)
  • 空间复杂度:O(1)或者 O(n)

我们将 nums 就地排序,如果不能就低排序,我们必须使用线性空间将 nums 数组拷贝,然后再排序。

方法四:分治

如果我们分别知道数组左边一半和右边一半的众数,左右两边众数相等那么就直接找到了,如果不相等就查找两边众数出现次数更多的那一个。

我们使用经典的分治算法递归求解,直到所有的子问题都是长度为 1 的数组。由于传输子数组需要额外的时间和空间,所以我们实际上只传输子区间的左右指针 lo 和 hi 表示相应区间的左右下标。长度为 1 的子数组中唯一的数显然是众数,直接返回即可。如果回溯后某区间的长度大于 1 ,我们必须将左右子区间的值合并。如果它们的众数相同,那么显然这一段区间的众数是它们相同的值。否则,我们需要比较两个众数在整个区间内出现的次数来决定该区间的众数。原问题的答案就是下标为 00 和 nn 之间的众数这一子问题。

class Solution {
    private int countInRange(int[] nums, int num, int lo, int hi) {
        int count = 0;
        for (int i = lo; i <= hi; i++) {
            if (nums[i] == num) {
                count++;
            }
        }
        return count;
    }

    private int majorityEleme***ec(int[] nums, int lo, int hi) {
        // base case; the only element in an array of size 1 is the majority
        // element.
        if (lo == hi) {
            return nums[lo];
        }

        // recurse on left and right halves of this slice.
        int mid = (hi-lo)/2 + lo;
        int left = majorityEleme***ec(nums, lo, mid);
        int right = majorityEleme***ec(nums, mid+1, hi);

        // if the two halves agree on the majority element, return it.
        if (left == right) {
            return left;
        }

        // otherwise, count each element and return the "winner".
        int leftCount = countInRange(nums, left, lo, hi);
        int rightCount = countInRange(nums, right, lo, hi);

        return leftCount > rightCount ? left : right;
    }

    public int majorityElement(int[] nums) {
        return majorityEleme***ec(nums, 0, nums.length-1);
    }
}
  • 时间复杂度:O(N*logN)

  • 空间复杂度:O(logN)

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 162,050评论 4 370
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 68,538评论 1 306
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 111,673评论 0 254
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,622评论 0 218
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 53,047评论 3 295
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,974评论 1 224
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 32,129评论 2 317
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,893评论 0 209
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,654评论 1 250
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,828评论 2 254
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,297评论 1 265
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,619评论 3 262
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,326评论 3 243
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,176评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,975评论 0 201
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 36,118评论 2 285
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,909评论 2 278

推荐阅读更多精彩内容