力扣1562.查找大小为M的最新分组
-
用link数组记录每一段区间的左右端点
- 当且仅当i点为端点时,link[i]有意义
-
int lint[100001] = {0}; class Solution { public: int findLatestStep(vector<int>& arr, int m) { int cnt = 0; memset(lint, -1, sizeof(lint)); int anw = -1; for(int i = 0; i < arr.size(); i++) { int pos = arr[i] - 1; //自己单独一个区间 lint[pos] = pos; int L = pos, R = pos; //左边有1 说明当前i为右端点 if(0 < pos && lint[pos-1] != -1) { //区间长度 = 右端点 - 左端点 + 1 //如果是m 说明m即将消失 if(pos-1 - lint[pos-1] + 1 == m) { cnt--; } //仍然是原来的左端点 L = lint[pos-1]; } //右边有1 说明当前i为左端点 if(pos+1 < arr.size() && lint[pos+1] != -1) { if(lint[pos+1] - (pos+1) + 1 == m) { cnt--; } R = lint[pos+1]; } lint[L] = R; lint[R] = L; //新区间是m长度 if(R-L+1 == m) { cnt++; } //有m的区间 更新答案 if(cnt > 0) { anw = i+1; } } return anw; } };