2019校招真题在线编程 连续子区间和（双指针）

题目描述

小M给你一串含有c个正整数的数组, 想让你帮忙求出有多少个下标的连续区间, 它们的和大于等于x。

输入描述:

第一行两个整数c x（0 < c <= 1000000, 0 <= x <= 100000000)

第二行有c个正整数（每个正整数小于等于100)。

输出描述:

输出一个整数，表示所求的个数。

示例1

输入

复制

3 6
2 4 7

输出

复制

4

说明

对于有3个整数构成的数组而言，总共有6个下标连续的区间，他们的和分别为：

2 = 2

4 = 4

7 = 7

2 + 4 = 6

4 + 7 = 11

2 + 4 + 7 = 13

其中有4个和大于等于6，所以答案等于4。

#include <bits/stdc++.h>

using namespace std;

const int N = (int)1e6 + 10;

int a[N];

int main() {
	int c, x;
	scanf("%d %d", &c, &x);
	for (int i = 0; i < c; i++) {
		scanf("%d", a + i);
	}
	int l = 0, r = 0;
	int sum = a[r];
	long long ans = 0;
	while (l < c) {
		while (sum < x && r < c) {
			r++;
			sum += a[r];
		}
		ans += (c - 1 - r + 1);
		sum -= a[l];
		l++;
	}
	cout << ans << "\n";
	return 0;
}