[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;
}

  

原文地址:https://www.cnblogs.com/lautsie/p/3436662.html