• [codility]Array-closest-ascenders


    http://codility.com/demo/take-sample-test/pi2012

    又是一道单调栈的题目。首先这道题目n^2是最朴素的做法。继续优化,因为和顺序有关就不好排序。然后,看到可以分解成求左方向最值和右方向最值的问题。此时要变成O(n)就要么贪心,要么DP,要么单调栈,要么有更多规律未发现。

    这里有这么一个特点,考虑求左方向最值,当先有4,再有5之后,那么之前的4就没有意义了。所以用单调栈。如果用函数,代码可以更简洁。

    #include <stack>
    vector<int> solution(const vector<int> &A) {
        // write your code in C++98
        int len = A.size();
        vector<int> R_left(len, 0);
        stack<int> left_stack;
        for (int i = 0; i < A.size(); i++) {
            while (!left_stack.empty() && A[left_stack.top()] <= A[i]) {
                left_stack.pop();
            }
            if (!left_stack.empty()) {
                R_left[i] = abs(left_stack.top() - i);
            }
            left_stack.push(i);
        }
        
        vector<int> R_right(len, 0);
        stack<int> right_stack;
        for (int i = A.size() - 1; i >= 0; i--) {
            while (!right_stack.empty() && A[right_stack.top()] <= A[i]) {
                right_stack.pop();
            }
            if (!right_stack.empty()) {
                R_right[i] = abs(right_stack.top() - i);
            }
            right_stack.push(i);
        }
        vector<int> R(len);
        for (int i = 0; i < len; i++) {
            if (R_right[i] == 0 && R_left[i] == 0) R[i] = 0;
            else if (R_right[i] == 0) R[i] = R_left[i];
            else if (R_left[i] == 0) R[i] = R_right[i];
            else R[i] = min(R_left[i], R_right[i]);
        }
        return R;
    }
    

      

  • 相关阅读:
    感知机学习笔记
    NOIP 模拟19
    NOIP 模拟17
    NOIP模拟14-16
    「动态规划」-数位dp专题
    8.5 NOIP 模拟测试 13
    8.3 NOIP 模拟12题解
    8.3 NOIP CE反思
    「分治」-cdq分治
    8.1 NOIP模拟11
  • 原文地址:https://www.cnblogs.com/lautsie/p/3436662.html
Copyright © 2020-2023  润新知