【题目】
给定数组arr和整数num,共返回有多少个子数组满足如下情况:
max(arr[i.j]) - min(arr[i.j]) <= num max(arfi.j])表示子数组ar[ij]中的最大值,min(arli.j])表示子数组arr[i.j]中的最小值。
【要求】
如果数组长度为N,请实现时间复杂度为O(N)的解法。
【题解】
使用两个单调栈,一个栈维持从大到小的排序,头部永远是最大值
一个维持从小到大的排序,头部永远都是最小值
然后使用窗口进行数据移动
当右移后,最大最小差超过num时,计算这段数组可以组成的子数组数量,因为长数组满足要求,
那么里面的短数组更满足要求,
然后进行左移,是最大最小差满足要求
1 #include <iostream> 2 #include <vector> 3 #include <list> 4 5 using namespace std; 6 7 int getSubArrayNum(vector<int>v, int num) 8 { 9 int l, r, res = 0; 10 list<int>maxl, minl; 11 for (l=-1,r=0;l<r && r<v.size();++r) 12 { 13 while (!maxl.empty() && v[r] > v[maxl.back()]) 14 maxl.pop_back(); 15 while (!minl.empty() && v[r] < v[minl.back()]) 16 minl.pop_back(); 17 maxl.push_back(r); 18 minl.push_back(r); 19 if (v[maxl.front()] - v[minl.front()] > num) 20 res += r - l; 21 while(v[maxl.front()] - v[minl.front()] > num) 22 { 23 ++l; 24 if (maxl.front() == l) 25 maxl.pop_front(); 26 if (minl.front() == l) 27 minl.pop_front(); 28 } 29 } 30 res += r - l; 31 return res; 32 } 33 34 35 int main() 36 { 37 vector<int>v = { 8,5,4,6,9,1,4,5,7,8,6,3,4,5,9,8 }; 38 int num = 4; 39 cout << getSubArrayNum(v, num) << endl; 40 return 0; 41 }
原文地址:https://www.cnblogs.com/zzw1024/p/11180966.html
时间: 2024-10-07 05:28:57