牛客-数学考试——前缀和的k区间问题

Posted 2022-02-05 C+++++++++++++++++++

文章目录

• • 题目
  • 题目解析
  • 解题代码

题目

题目连接

题目解析

总结下上面说的两点：

1. 需要同时找两个的原因。
2. 如何优化到O(n)级别。

解题代码

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
ll dp[200005];
int main()
    int t,n,k;
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--)
        cin>>n>>k;
        for(int i=1;i<=n;i++)
            cin>>dp[i];
            dp[i] += dp[i-1];
        
        ll ret = -1e18;
        ll mx1 = -1e18;
        for(int i=k;i+k<=n;i++)
            mx1 = max(mx1,dp[i]-dp[i-k]);
            ret = max(mx1+dp[i+k]-dp[i],ret);
        
        cout<<ret<<'\\n';