连续子区间的和大于等于某一个数

题目描述

小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;
}