LeetCode 975. 奇偶跳（set，dp）

题意：

给定一个整数数组 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 上。

（对于某些索引 i，可能无法进行合乎要求的跳跃。）

如果从某一索引开始跳跃一定次数（可能是 0 次或多次），
就可以到达数组的末尾（索引 A.length - 1），那么该索引就会被认为是好的起始索引。

返回好的起始索引的数量。

数据范围:
1 <= A.length <= 20000
0 <= A[i] < 100000

解法：

d[i][0]表示当前在i位置,下一步是偶数步能否到达终点
d[i][1]表示当前在i位置,下一步是奇数步能否到达终点

d[i][0]|=d[j][1],其中j是满足题目条件的后一个位置
d[i][1]|=d[k][0],其中k是满足题目条件的后一个位置

倒着dp即可.

如何快速找满足条件的后一个位置:
这里我用的是set.lower_bound快速查找后一个满足条件的位置.

code：

/*
 * @lc app=leetcode.cn id=975 lang=cpp
 *
 * [975] 奇偶跳
 */

// @lc code=start

const int maxm=1e5+5;
#define PI pair<int,int>
int d[maxm][2];
class Solution {
public:
    int oddEvenJumps(vector<int>& a) {
        int n=a.size();
        memset(d,0,sizeof d);
        d[n-1][0]=d[n-1][1]=1;
        set<PI>s;
        s.insert({a[n-1],n-1});
        s.insert({-999999999,100000});
        s.insert({999999,100000});
        for(int i=n-2;i>=0;i--){
            auto it=s.lower_bound({a[i],0});
            d[i][1]|=d[it->second][0];
            

            if(it->first>a[i])it--;
            d[i][0]|=d[it->second][1];
            
            while(1){
                auto it=s.lower_bound({a[i],0});
                if(it->first==a[i]){
                    s.erase(it);
                }else{
                    break;
                }
            }
            s.insert({a[i],i});
        }
        for(int i=0;i<n;i++){
            cout<<i<<' '<<d[i][0]<<' '<<d[i][1]<<endl;
        }
        int ans=0;
        for(int i=0;i<n;i++){
            if(d[i][1])ans++;
        }
        return ans;
    }
};
// @lc code=end

---