LC 1562. 查找大小为 M 的最新分组

class Solution {
   public:
    int findLatestStep(vector<int>& arr, int m) {
        if (m == arr.size()) {
            return m;
        }
        int size = arr.size() + 2; // 因为arr的数从1开始所以需要+1，后面还要判断集合的尾部防止越界所以又+1
        vector<int> v(size, 0);
        int res = -1;
        for (int i = 0; i < arr.size(); i++) {
            int n = arr[i];
            if (v[n - 1] != 0 && v[n + 1] != 0) {
                int q = v[n - 1] + v[n + 1] + 1;
                if (v[n - v[n - 1]] == m || v[n + v[n + 1]] == m) {
                    res = i;
                }
                v[n - v[n - 1]] = q;
                v[n + v[n + 1]] = q;
            } else if (v[n - 1] != 0) {
                if (v[n - 1] == m) {
                    res = i;
                }
                v[n] = v[n - 1] + 1;
                v[n - v[n - 1]]++;
            } else if (v[n + 1] != 0) {
                if (v[n + 1] == m) {
                    res = i;
                }
                v[n] = v[n + 1] + 1;
                v[n + v[n + 1]]++;
            } else {
                v[n] = 1;
            }
        }
        return res;
    }
};