首页 > 代码库 > Codility上的练习(12)

Codility上的练习(12)

(1) MinMaxDivision

给定一个非负整数数组,每个整数都是[0..M]之间的,你要把它分成K段,(切K - 1刀),段可以为空,每个元素必须属于一段,每段必须包含0个或者多个连续的元素,要求分好和最大段的和尽量小,返回这个尽可能小的最大和。

数据范围 :N, K [1..10^5], M [0..10^4]

要求复杂度 时间 O(N * log(N + M)) 空间 O(1)。


分析:典型的二分我们可以。二分一个最大段的和,然后我们一段一段地加,超过要加的值,就开始一段新的。这个方法得力于都是非负整数……

简单说一下复杂度,二分问题的框架就是 二分 + 判断。 判断部分显然是O(N)的。 二分的复杂度取决于二分的区间大小。我们的二分区间左端点可以认为是min(A[i]),也可以认为是0,反正区间大点也关系,右端点最大是N * M,那么二分的复杂度是O(log(N * M)) = O(logN + logM) = O(2 * log(max(M, N)) = O(log(max(M, N)) = O(log (M + N)) 所以算上检测的复杂度就达到要求的那个了。

// you can use includes, for example:
// #include <algorithm>

// you can write to stdout for debugging purposes, e.g.
// cout << "this is a debug message" << endl;


bool can(vector<int> &a,int x, int k) {
int sum = 0;
    --k;
    for (int i = 0; i < a.size();) {
        if ((sum += a[i]) > x) {
            if (--k < 0) {
                return false;
            }
            sum = 0;
        }
        else {
            ++i;
        }
        
    }
    return true;
}
int solution(int K, int M, vector<int> &A) {
    // write your code in C++98
    int left = 0,right = -1;
    for (int i = 0; i < A.size(); ++i) {
        right += A[i];
        left = max(left, A[i]);
        
    }
    
    while (left <= right) {
        int mid = (left + right) >> 1;
        if (can(A, mid, K)) {
            right = mid - 1;
        }
        else {
            left = mid + 1;
        }
    }
    return right + 1;
    
}

(2) NailingPlanks

N块木板,可以看作N线段,给定两个长度为N的正整数数组A[],B[],[A[k],B[k]]表示木板(线段)的起点和终点,A[k] <= B[k]。有M个钉子,它们分别在长度为M的正整数数组里。钉子I可以固定住木板K,当且今当A[K]<=C[I]<=B[K]。问按顺序使用钉子,至少使用前多少个钉子可以固定住所有木板?无解返回-1。

数据范围: 木板数N和M的范围[1..30000], A B C数组元素范围为[1..2 * M]

要求复杂度: 时间O((N+M)*log(M)) , 空间O(M)

分析: 一个显然的并且符合要求的算法是二分答案,问题是如何判断,显然我们不能循环木板和钉子。但是我们可以计算从开头到当前位置一共有多少个钉子,这是前缀和的思想。计算前缀和需要O(M),判断需要O(N),二分是O(logM),所以整好是要求的时间复杂度。空间上需要存前缀和O(M)。

代码:

// you can also use includes, for example:
// #include <algorithm>
int solution(vector<int> &A, vector<int> &B, vector<int> &C) {
    // write your code in C++98
    int m = C.size();
    int M = (m << 1) | 1;
    int left = 0, right = m, result = -1;
    while (left <= right) {
        int mid = (left + right) >> 1;
        vector<int> v;
        v.resize(M, 0);
        for (int i = 0; i < mid; ++i) {
            ++v[C[i]];
        }
        for (int i = 1; i < M; ++i) {
            v[i] += v[i - 1];
        }
        bool can = true;
        for (int i = 0; i < A.size(); ++i) {
            if (v[B[i]] - v[A[i] - 1] == 0) {
                can = false;
                break;
            }
        }
        if (can) {
            result = mid;
            right = mid - 1;
        }
        else {
            left = mid + 1;
        }
    }
    return result;
}

更快的算法,如果我们建立一个长度为2 * M的数组,每个位置表示该位置上钉子的最小编号(可能同一个位置有多个钉子,取编号最小的),没有钉子的位置值为无穷大。那么固定第i块木板的最小编号钉子,相当于[A[i],B[i]]区间的最小值。但是我们这个题实际上是求这些最小值的max,首先如果一个木板A的覆盖区间完全包含另外一个木板B,则实际上我们只考虑木板B即可。因为固定木板B同时能固定木板A,并且我们一定要固定木板B,即使A覆盖区间有更小的值,也无法改变最终取最大值的结果。

于是,我们可以建立一个数组plank[x]表示右端点为x的木板的最大左边界,没有木板的话,认为边界是0。我们从左到右遍历木板右边界,假设这之前(更左)的木板已经被固定了,已经固定的区间范围是[left,right) (右开区间),然后对当前这个木板,如果显然它的右边界更大(我们遍历右边界是按当增的顺序),如果该木板start <= left,则它已经被前面固定住了,不影响结果。否则,要求[start,end]之间的最大值。这个问题有点像滑动窗口最大值的问题。本质在于:我们不断查询最大值,每次查询的时候窗口的左边界和右有边界是单调递增的,于是我们可以动态更新窗口维护最大值。这个经典问题可以用单调队列实现,这也是把单调队列发挥到了极致。

结论: 查询窗口最大值的时候,如果窗口向右滑动的过程中,查询时左边界和右边界都是单增的,则可以使用单调队列解决。

时间复杂度 O(N + M)达到了线性。

代码:

// you can also use includes, for example:
// #include <algorithm>
#include <deque>
const int inf = 2000000000;
int solution(vector<int> &A, vector<int> &B, vector<int> &C) {
    // write your code in C++98
    int m = C.size(), M = (m << 1) | 1;
    vector<int> nail(M, inf);
    for (int i = m - 1; i >= 0; --i) {
        nail[C[i]] = i;
    }
    vector<int> plank(M, 0);
    for (int i = 0; i < A.size(); ++i) {
        plank[B[i]] = max(plank[B[i]], A[i]);
    }
    int left = 0, right = 0, r = 0;
    deque<int> q;
    for (int i = 1; i < M; ++i) {
        if (plank[i] > left) {
            left = plank[i];
            while ((!q.empty()) && (q.front() < left)) {
                q.pop_front();
            }
            for (right = max(right, left); right <= i; ++right) {
                while ((!q.empty()) && (nail[q.back()] >= nail[right])) {
                    q.pop_back();
                }
                q.push_back(right);
            }
            r = max(r, nail[q.front()]);
            if (r >= inf) {
                return -1;
            }
        }
    }
    return r + 1;
}



Codility上的练习(12)