奇偶跳

/*
思路：
1. 使用TreeMap和IntStream来实现奇数跳和偶数跳。
2. 维护两个数组odd和even来记录每个索引的奇数跳和偶数跳状态。
3. 通过IntStream倒序遍历数组，从后往前更新奇数跳和偶数跳的状态。
4. 使用TreeMap来找到满足条件的索引。
*/
import java.util.*;
import java.util.stream.IntStream;

public class OddEvenJumpsStream {
    public int oddEvenJumps(int[] A) {
        int n = A.length;
        if (n <= 1) return n;
        boolean[] odd = new boolean[n];
        boolean[] even = new boolean[n];
        odd[n - 1] = true;
        even[n - 1] = true;
        TreeMap<Integer, Integer> map = new TreeMap<>();
        map.put(A[n - 1], n - 1);
        int count = 1;
        IntStream.iterate(n - 2, i -> i >= 0, i -> i - 1).forEach(i -> {
            Integer oddJump = map.ceilingKey(A[i]);
            Integer evenJump = map.floorKey(A[i]);
            if (oddJump != null) {
                odd[i] = even[map.get(oddJump)];
            }
            if (evenJump != null) {
                even[i] = odd[map.get(evenJump)];
            }
            if (odd[i]) count++;
            map.put(A[i], i);
        });
        return count;
    }
}