题目:连续子序列


解题思路:

        首先,不能使用暴力枚举,时间为O(n2),超时。以下为正确做法:

        假设找到一段区间(其和>=m),如上图黄色部分,那么该区间加上i后面的元素形成的新区间和都>=m,因此以该区间为基础就有n-i+1个区间符合要求。

        那么我们只需要从1开始找到每一个恰好大于等于m的黄色区间,再依次把每一个黄色区间为基础的区间的个数相加就得到答案。


AC代码:

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e5+9;
int a[N];
ll m;
// 依次找出区间和>=m的滑动窗口,j++ 
int main()
{
     ll sum = 0,ans = 0;
    int n, j = 1;cin >> n >> m;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        sum += a[i];
        if(sum >= m)
        {
            ans += (n-i+1);
            while(j <= i && sum >= m){  // 数组从1开始序号递增,所以当序号i>=j时区间合法 
            sum -= a[j];
            j++;
            if(sum >= m)ans += (n-i+1);
            } 
        }
    }
    cout << ans << '\n';    
    return 0;
}




知识点:

        双指针,滑动窗口

上一篇:跟我学C++中级篇——优化的整体分析-一、软件的优化


下一篇:sql-labs靶场第十八关测试报告