XidianOJ 1099 A simple problem

题目描述

一个长度为N的数组A， 所有数都是整数 ，0 <= A[i] <= 1000000，1 <= i <= N，1 <= N <= 100000，对于 任意i，j ，1 <=  i  <=  j  <= N，[i， j]中所有数为原数组的一个子区间， 现在要求子区间的和小于等于K的子区间有多少个， 0 <=  K  <= 10000。

虽然xry111很SB，但还是在O(N)的时间复杂度内就做出了这题，你呢？

              

 

 

输入

第一行整数T 代表数据组数，1 <= T  <=  12

每组数据第一行 整数 N， K。

接下来一行N个整数，  由空格隔开。

 

输出

输出子区间的和小于等于K的子区间的个数。 每组输出占一行。

--正文

明明是简单的题，却做了半天。。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;typedef long long LL;
long long a[100001];
int n,k,i,j;
int main(){int time,T; scanf("%d",&T);for (time=1;time<=T;time++){int now = 1;LL sum = 0,res = 0;scanf("%d %d",&n,&k);for (i=1;i<=n;i++){scanf("%lld",&a[i]);sum += a[i];if (sum <= k) {res += i - now + 1;}else {while (sum > k){sum = sum - a[now]; now ++;} res += i - now + 1;} }printf("%lld\n",res);}return 0;
}