【LeetCode】分治法 divide and conquer （共17题）

链接：https://leetcode.com/tag/divide-and-conquer/

 

【4】Median of Two Sorted Arrays 

【23】Merge k Sorted Lists 

 

【53】Maximum Subarray (2019年1月23日， 谷歌tag复习)

最大子段和。

题解：

follow up 是divide and conquer

If you have figured out the O(n) solution, try coding another solution using the divide and conquer approach, which is more subtle. 

 

【169】Majority Element 

 

【215】Kth Largest Element in an Array (2018年12月11日，wiggle sort 专题，需要复习)

用 O(n) 的时间复杂度找到数组中第 K 大的元素。重复元素也计入 K。

题解：本题可以用 heap 解答，时间复杂度是 O(nlogK)。其实还可以用 quick select 解答（也就是快排的partition(2-way partition)），平均复杂度是 O(n)，最坏是 O(n^2)。为了搞平均，一开始要把数组 random_shuffle 一下，尽可能避免 worst case。 

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        random_shuffle(nums.begin(), nums.end());
        const int n = nums.size();
        int start(0), end(n-1), index(n-k);        
        while (start < end) {
            int idx = partition(nums, start, end);
            if (idx < index) {
                start = idx + 1;
            } else if (idx > index) {
                end = idx - 1;
            } else {
                return nums[idx];
            }
        }
        return nums[start];
    }
    int partition(vector<int>& nums, int start, int end) {
        int pivot = start;
        while (start < end) {
            while (nums[start] <= nums[pivot]) {
                start++;
            }
            while (nums[end] > nums[pivot]) {
                end--;
            }
            if (start > end) {break;}
            swap(nums[start], nums[end]);
        }
        swap(nums[end], nums[pivot]);
        return end;
    }

};

 

【218】The Skyline Problem 

【240】Search a 2D Matrix II 

【241】Different Ways to Add Parentheses （2018年11月15日，算法群）

给了一个字符串算式，里面含有 “+”，“-”，“*” 这三种运算符，可以在算式的任何一个地方加括号，整个算式能加的括号数不限。问这个算式所有可能的答案。

Example 1:
Input: "2-1-1"
Output: [0, 2]
Explanation: 
((2-1)-1) = 0 
(2-(1-1)) = 2

Example 2:
Input: "2*3-4*5"
Output: [-34, -14, -10, -10, 10]
Explanation: 
(2*(3-(4*5))) = -34 
((2*3)-(4*5)) = -14 
((2*(3-4))*5) = -10 
(2*((3-4)*5)) = -10 
(((2*3)-4)*5) = 10

 题解：我们可以在任意地方加括号，每个运算符的两边都可以加个括号看成是一个子问题，先把子问题的所有解求出来，然后把两个子问题的解集合做笛卡尔积，形成大问题的解集合。

审题很重要，我一度以为如果运算符是 ‘-’ 的话，那么后面的算式的 加号要变成减号，减号要变成加号，这个题意里面是没有的。

class Solution {
public:
    vector<int> diffWaysToCompute(string input) {
        return calPart(input);
    }
    vector<int> calPart(string s) {
        const int n = s.size();
        if (record.find(s) != record.end()) {return record[s];}
        //纯数字的情况
        if (s.find("+") == string::npos && s.find("-") == string::npos && s.find("*") == string::npos) {
            vector<int> ret{stoi(s)};
            return ret;
        }
        //含有运算符的情况
        vector<int> res;
        for (int i = 0; i < n; ++i) {
            if (!isdigit(s[i])) {
                string front = s.substr(0, i), back = s.substr(i+1);
                vector<int> resFront = calPart(front), resBack = calPart(back);
                for (auto f : resFront) {
                    for (auto b : resBack) {
                        int tempres = 0;
                        if (s[i] == '+') {
                            tempres = f + b;
                        } else if (s[i] == '-') {
                            tempres = f - b;
                        } else if (s[i] == '*') {
                            tempres = f * b;
                        }
                        res.push_back(tempres);
                    }
                }
            }
        }
        //sort(res.begin(), res.end());
        record[s] = res;
        return res;
    }
    unordered_map<string, vector<int>> record;
};

 

【282】Expression Add Operators (2019年3月16日，dfs，打卡题)

Given a string that contains only digits 0-9 and a target value, return all possibilities to add binary operators (not unary) +, -, or *between the digits so they evaluate to the target value.

 

Example 1:
Input: num = "123", target = 6
Output: ["1+2+3", "1*2*3"] 

Example 2:
Input: num = "232", target = 8
Output: ["2*3+2", "2+3*2"]

Example 3:
Input: num = "105", target = 5
Output: ["1*0+5","10-5"]

Example 4:
Input: num = "00", target = 0
Output: ["0+0", "0-0", "0*0"]

Example 5:
Input: num = "3456237490", target = 9191
Output: []

 

题解：如果本题只有加号和减号，那么就是一个直接的dfs。但是本题还有一个乘号，乘号的可以提前优先级运算。所以我们需要保存最后一个单项式的值。

比如说我们已经计算了 1 + 2 - 3 ___  5 要填写 3 和 5 之间的符号的时候， 如果我们想填个乘号，那么，我们需要最后一项 -3，所以我们用一个变量保存当前表达式的最后一个参数，作为 dfs 的参数。

class Solution {
public:
    vector<string> addOperators(string num, int target) {
        vector<string> res;
        string temp;
        dfs(num, temp, res, (long)target, 0, 0LL, 0LL);
        return res;
    }
    void dfs(const string num, string temp, vector<string>& res, long target, int start, long curRes, long lastVal) {
        if (start == num.size()) {
            if (target == curRes) {res.push_back(temp);}
            return;
        }
        for (int i = start; i < num.size(); ++i) {
            string strCur = num.substr(start, i - start + 1);
            if (strCur.size() > 1 && strCur[0] == '0') {break;} //leading zeros
            long iCur = stol(strCur);
            if (start == 0) {
                dfs(num, strCur, res, target, i+1, iCur, iCur);
            } else {
                dfs(num, temp + "+" + strCur, res, target, i + 1, curRes + iCur, iCur);
                dfs(num, temp + "-" + strCur, res, target, i + 1, curRes - iCur, -iCur);
                dfs(num, temp + "*" + strCur, res, target, i + 1, curRes - lastVal + lastVal * iCur, lastVal * iCur);
            }
        }
    }
};

 

【312】Burst Balloons 

 

【315】Count of Smaller Numbers After Self (2019年2月12日，归并排序)

给了一个数组，要求返回一个数组，返回数组中的元素是原数组中每个元素右边比它小的个数。

Input: [5,2,6,1]

Output: [2,1,1,0] 

Explanation:

To the right of 5 there are 2 smaller elements (2 and 1).

To the right of 2 there is only 1 smaller element (1).

To the right of 6 there is 1 smaller element (1).

To the right of 1 there is 0 smaller element.

题解：这个题可以用线段数，树状数组，分治法来解。下面说一下分治法怎么解。

在每一轮中，我们可以把整个数组分成左右两半，利用归并排序的思想，把左区间和右区间变的有序之后，这样我们就可以把左区间内的任意一个元素nums[i]，在右区间内用 lower_bound 找到有多少个元素小于它，然后把这个值加到res[i]上。

一共需要三个数组：nums, sorted, count。原来的数据存在nums, 归并排序后的数组存在sortedNums, count[i]对应的是nums[i]的 number of smaller elements to the right.

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        const int n = nums.size();
        vector<int> sorted(nums.begin(), nums.end()), res(n, 0);
        if (n == 0) {return res;}
        mergesort(nums, sorted, 0, n-1, res);
        return res;
    }
    void mergesort(vector<int>& nums, vector<int>& sorted, int begin, int end, vector<int>& res) {
        if (begin == end) { return; }
        int mid = (begin + end) / 2;
        mergesort(nums, sorted, begin, mid, res);
        mergesort(nums, sorted, mid + 1, end, res);
        for (int i = begin; i <= mid; ++i) {
            int value = nums[i];
            auto iter = lower_bound(sorted.begin() + mid + 1, sorted.begin() + end + 1, value);
            res[i] += distance(sorted.begin() + mid + 1, iter);
        }
        //merge
        vector<int> arr1(sorted.begin() + begin, sorted.begin() + mid + 1), 
                    arr2(sorted.begin() + mid + 1, sorted.begin() + end + 1);
        int p1 = 0, p2 = 0;
        for (int idx = begin; idx <= end; ++idx) {
            if (p1 == arr1.size()) {
                sorted[idx] = arr2[p2++];
            } else if (p2 == arr2.size()) {
                sorted[idx] = arr1[p1++];
            } else {
                if (arr1[p1] < arr2[p2]) {
                    sorted[idx] = arr1[p1++];
                } else {
                    sorted[idx] = arr2[p2++];
                }
            }
        }
    }
};

 

【327】Count of Range Sum （2019年2月14日，谷歌tag，归并排序）

给了一个数组nums，和一个范围 [lower, upper]，返回有多少个子数组的和在这个范围之内。

题解：

【426】Convert Binary Search Tree to Sorted Doubly Linked List 

 

【493】Reverse Pairs (2019年2月19日，归并排序)

给定一个数组，求它的逆序对个数。本题的逆序对的定义和别的题不同：if i < j and nums[i] > 2*nums[j].

题解：我们这个题目同315，用分治法解题。我们把这整个数组分成两个区间，然后分别对这两个左右区间做归并排序。然后对右边区间的每一个元素 nums[j], target = nums[j] * 2。然后用 upper_bound() 求出左区间内第一个比 target 大的元素，从这个元素开始 到左区间结束，这些元素都能和 target 组成逆序对。所以把这些元素加到结果上。

class Solution {
public:
    typedef long long LL;
    int reversePairs(vector<int>& nums) {
        if (nums.empty()) {return 0;}
        int n = nums.size();
        vector<LL> arr(n);
        for (int i = 0; i < n; ++i) {
            arr[i] = nums[i];
        }
        divideAndConquer(arr, 0, n-1);
        return res;
    }
    int res = 0;
    void divideAndConquer(vector<LL>& nums, int start, int end) {
        if (start == end) {return;}
        int mid = start + (end - start) / 2;
        divideAndConquer(nums, start, mid);
        divideAndConquer(nums, mid + 1, end);
        for (int j = mid + 1; j <= end; ++j) {
            LL target = nums[j] * 2;
            auto iter = upper_bound(nums.begin() + start, nums.begin() + mid + 1, target);
            res += distance(iter, nums.begin() + mid + 1);
        }
        // print(nums, start, end);
        inplace_merge(nums.begin() + start, nums.begin() + mid + 1, nums.begin() + end + 1);
        // print(nums, start, end);
    }
    void print(vector<LL>& nums, int start, int end) {
        for (int i = start; i <= end; ++i) {
            printf("%lld ", nums[i]);
        }
        printf("\n");
    }
};

  

【514】Freedom Trail 

【903】Valid Permutations for DI Sequence 

【932】Beautiful Array