51nod - 1275 连续子段的差异

给出一个包括N个元素的整数数组A，包括A本身在内，共有 (N+1)*N / 2个非空子段。例如：1 3 2的子段为{1} {3} {2} {1 3} {3 2} {1 3 2}。在这些子段中，如果最大值同最小值的差异不超过K，则认为这是一个合格的子段。给出数组A和K，求有多少符合条件的子段。例如：3 5 7 6 3，K = 2，符合条件的子段包括：{3} {5} {7} {6} {3} {3 5} {5 7} {7 6} {5 7 6}，共9个。

Input

第1行：2个数N, K（1 <= N <= 50000, 0 <= K <= 10^9)
第2 - N + 1行：每行1个数，对应数组的元素Ai(0 <= A[i] <= 10^9)

Output

输出符合条件的子段数量。

Input示例

5 2
3
5
7
6
2

Output示例

9

思路：

从左到右，依次以每个点为起点，找出最大的能满足条件的终点。终点减去起点就是以该点为起点的满足条件的子段个数，最后总数相加即可。

#include <iostream>  
#include <cstring>
#include <queue>
using namespace std;  
 
const int MAXN = 50050;
int input[MAXN];
 
int main()  
{  
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < n; i++)
	{
		cin >> input[i];
	}
 
	deque<int> bufMin, bufMax;
	int j = 0;
	int result = 0;
	for (int i = 0; i < n; i++)
	{
		while (j < n)
		{
			while (!bufMin.empty() && input[bufMin.back()] >= input[j])
			{
				bufMin.pop_back();
			}
			bufMin.push_back(j);
 
			while (!bufMax.empty() && input[bufMax.back()] <= input[j])
			{
				bufMax.pop_back();
			}
			bufMax.push_back(j);
 
			if (input[bufMax.front()] - input[bufMin.front()] <= k)
			{
				j++;
			}
			else
			{
				bufMin.pop_back();
				bufMax.pop_back();
				break;
			}
		}
		
		result += (j - i);
		if (bufMin.front() == i)
		{
			bufMin.pop_front();
		}
		if (bufMax.front() == i)
		{
			bufMax.pop_front();
		}
	}
 
	cout << result << endl;
	return 0;  
}