题目描述
小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。
链接:https://www.nowcoder.com/questionTerminal/c7db49124acd415f801eb67de09c6d81?f=discussion
来源:牛客网
#include <bits/stdc++.h>
using namespace std;
int main()
{
int c,x,res=0;
cin>>c>>x;
int a[c];
for(int i=0;i<c;i++)
cin>>a[i];
long long ans=0,sum=0,j=0;
for(int i=0;i<c;i++)
{
sum+=a[i];
if(sum>=x)
{
ans+=(c-i);
while(j<=i)
{
sum-=a[j];
j++;
if(sum>=x)
ans+=(c-i);
else
break;
}
}
}
cout<<ans<<endl;
return 0;
}