//题目:
//给定数组arr和整数num,共返回有多少个子数组满足如下情况:
//max(arr[i...j]) - min(arr[i...j]) <= num
//max(arr[i...j])表示子数组arr[i...j]中的最大值,min(arr[i...j])表示子数组arr[i...j]中的最小值。
//给定数组arr和整数num,共返回有多少个子数组满足如下情况:
//max(arr[i...j]) - min(arr[i...j]) <= num
//max(arr[i...j])表示子数组arr[i...j]中的最大值,min(arr[i...j])表示子数组arr[i...j]中的最小值。
//要求:如果数组长度为N,请实现时间复杂度为O(N)的解法。
//思路:
//1.生成qmax和qmin,同时生成两个整型变量i和j,表示子数组的范围,即arr[i..j],生成整型变量res,表示所有满足条件的子数组数量。
// 2.令j不断向右移动(j++),表示arr[i..j]一直向右扩大,不断更新qmax和qmin结构。一旦出现arr[i..j]不满足条件的情况,j向右扩的过程停止,
// 此时arr[i..j-1]、arr[i..j-2]、arr[i..j-3]、……、arr[i..i]都是满足条件的。也就是说满足条件的个数为j - i,即令res = j - i 。
// 3.完成步骤2,再令i向右移动一个位置,并对qmax和qmin进行更新,此时是arr[i+1..j]窗口的最大值和最小值的更新结构。然后反复重复步骤2。
// 4.根据步骤2和3,依次求出以arr[0]、arr[1]……、arr[N]作为第一个元素的子数组中满足条件的数量分别有多少个,累积起来的数量就是最终的结果。
// 整个过程由于小标值最多进qmax和qmin一次和出一次,所以过程的时间复杂度为O(N)。
#include<iostream>
using namespace std ;
#include <deque>
#include <cassert>
int GetNum(int *arr, int size, int num)
{
assert(arr && size > 0 && num > 0);
deque<int> qmax; //用双端队列维持连续子数组的最大值。
deque<int> qmin; //用双端队列维持连续子数组的最小值。
int res = 0; //子数组的个数。
int i = 0;
int j = 0;
while (i < size)
{
while (j < size)
{
while (!qmax.empty() && arr[qmax.back()] <= arr[j]) //求子数组的最大值。
{
qmax.pop_back();
}
qmax.push_back(j);
while (!qmin.empty() && arr[qmin.back()]>= arr[j])//子数组的最小值。
{
qmin.pop_back();
}
qmin.push_back(j);
if (arr[qmax.front()] - arr[qmin.front()] > num)
{
break;
}
j++;
}
res += j-i; //算出以i为起始点的连续子数组的个数。
// 因为已经算出以i为起始点的连续子数组的个数,下次就是以i+1为起始位置了,所以如果max/min中含有i,则需要移除
if (qmin.front() == i)
{
qmin.pop_front();
}
if (qmax.front() == i)
{
qmax.pop_front();
}
i++;
}
return res;
}
int main()
{
int arr[] = {77,5,1};
int num = 8;
int len = sizeof(arr)/sizeof(arr[0]);
int ret = GetNum(arr,len, num);
cout << ret <<endl;
cout << "hello..." <<endl;
system("pause");
return 0;
}