单调栈——数组中找出每个数后面比它大的数中最小的那个 odd even jump

数组中找出每个数后面比它大的数中最小的那个；

数组中找出每个数后面比它小的数中最大的那个。

---

给定一个整数数组A。从某一些起始索引，你可以做一系列的跳跃。其中的（第1，第3，第5 ......）跳跃称为奇数跳跃，（第2，第4，第6 ......）跳跃称为偶数跳跃。

你可以从索引i 以下列方式跳转到索引 j（i <j）：

在奇数跳跃（即跳跃1,3,5，...）期间，跳转到索引j，使得A [i] <= A [j]并且A [j]是可能的最小值。如果有多个这样的索引j，则只能跳转到最小的索引 j。 在偶数跳跃（即跳跃2,4,6，...）期间，跳转到索引j，使得A [i]> = A [j]，A [j]是最大的可能值。如果有多个这样的索引j，则只能跳转到最小的索引 j。（可能存在某些索引，不存在合法的跳跃）
如果从该索引开始，您可以通过跳转一些次数（大于等于0次）到达数组的末尾（索引A.length - 1），那么这是一个有效的起始索引

 

---

 

方法一单调栈：

class Solution:
    """
    @param A: An integer array A
    @return: Return the number of good starting indexes
    """
    def oddEvenJumps(self, A):
        # Write your code here
        stack, array, upper = [], sorted([(a, i) for i, a in enumerate(A)]), [-1 for i in range(len(A))]
        for num, i in array:
            while len(stack) > 0 and stack[-1] < i:
                upper[stack.pop()] = i
            stack.append(i)
            
        stack, array, lower = [], sorted([(-a, i) for i, a in enumerate(A)]), [-1 for i in range(len(A))]
        for num, i in array:
            while len(stack) > 0 and stack[-1] < i:
                lower[stack.pop()] = i
            stack.append(i)
            
        odd, even = [0 for i in range(len(A))], [0 for i in range(len(A))]
        odd[len(A) - 1], even[len(A) - 1] = 1, 1
        
        ans = 0
        for i in range(len(A) - 1, -1, -1):
            if upper[i] != -1:
                odd[i] = even[upper[i]]
            if lower[i] != -1:
                even[i] = odd[lower[i]]
            ans += odd[i]
        
        return ans

 

方法二：

用treemap来做，给定一个数（key），去map中找upper_bound第一个大于key的，以及lower_bound第一个大于等于key的。

class Solution {
public:
    /**
     * @param A: An integer array A
     * @return: Return the number of good starting indexes
     */
    int oddEvenJumps(vector<int> &A) {
        // Write your code here
        vector<int> upper(A.size(), -1);
        vector<int> lower(A.size(), -1);
        
        map<int, int> mapNumToIndex;
        
        for(int i = A.size() - 1; i >= 0; i--){
            map<int, int>::iterator lowerBound = mapNumToIndex.lower_bound(A[i]);
            
            if(lowerBound != mapNumToIndex.end()) {
                upper[i] = lowerBound->second;
            }
            
            if(lowerBound != mapNumToIndex.end() && lowerBound->first == A[i]) lower[i] = lowerBound->second;
            else if(lowerBound != mapNumToIndex.begin()) lower[i] = (--lowerBound)->second;
            
            mapNumToIndex[A[i]] = i;
        }
        
        vector<int> odd(A.size(), 0);
        vector<int> even(A.size(), 0);
        odd[A.size() - 1] = even[A.size() - 1] = 1;
        
        int ans = 1;
        for(int i = A.size() - 2; i >= 0; i--){
            if(upper[i] != -1) odd[i] = even[upper[i]];
            if(lower[i] != -1) even[i] = odd[lower[i]];
            ans += odd[i];
        }
        return ans;
    }
};