前缀和

直接说例题把

数学考试

链接:https://ac.nowcoder.com/acm/problem/15553
来源:牛客网

今天qwb要参加一个数学考试,这套试卷一共有n道题,每道题qwb能获得的分数为ai,qwb并不打算把这些题全做完,
他想选总共2k道题来做,并且期望他能获得的分数尽可能的大,他准备选2个不连续的长度为k的区间,
即[L,L+1,L+2,…,L+k-1],[R,R+1,R+2,…,R+k-1](R >= L+k)。
输入描述:

第一行一个整数T(T<=10),代表有T组数据
接下来一行两个整数n,k,(1<=n<=200,000),(1<=k,2k <= n)
接下来一行n个整数a1,a2,…,an,(-100,000<=ai<=100,000)

输出描述:

输出一个整数,qwb能获得的最大分数

输入
2
6 3
1 1 1 1 1 1
8 2
-1 0 2 -1 -1 2 3 -1

输出
6
7

思路:运用前缀和,记录到i前面的连续k个的最大值和到i后面连续的k个最大值
最后取L[i]+R[i+1]的最大值

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+5;
int a[N];
ll b[N];//b记录前缀和
ll L[N],R[N];//L记录到i前面连续k个最大值,R记录到i后连续k个最大值
int main() {
    int t;
    cin>>t;
    while(t--){
        int n,k;
        cin>>n>>k;
        memset(b,0,sizeof(b));
        for(int i=1;i<=n;i++){
            cin>>a[i];
            L[i]=R[i]=-1e18;
            b[i]=b[i-1]+a[i];
        }
        for(int i=k;i<=n-k;i++){
            L[i]=max(L[i-1],b[i]-b[i-k]);
        }
        for(int i=n-k+1;i>=k+1;i--){
            R[i]=max(R[i+1],b[i+k-1]-b[i-1]);
        }
        ll ans=-1e18;
        for(int i=k;i<=n-k;i++){
            ans=max(ans,L[i]+R[i+1]);
        }
        cout<<ans<<endl;
    }
    return 0;
}

在网上还看到其他的思路但是我没怎么懂,这里还是记录一下别人的思路:

因为区间大小是固定为k的,所以显然需要前缀和处理一下
处理之后我们去维护前缀中长度为k的最大值ma,枚举第二个长度为k的起点,那么答案就是max(ma+当前长度为k的序列和)
复杂度为O(n)
极限数据计算 n为2e5,k=1e5 所有ai都是1e5 那么最大值是2e5*1e5>(1<<31)-1 超过了int所能表示的范围 所以需要开long long

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
ll a[N];
int main(){
    int t;cin>>t;
    while(t--){
        int n,k;cin>>n>>k;
        for(int i=1;i<=n;i++) cin>>a[i],a[i]+=a[i-1];
        ll ma=-1e18,ans=-1e18;
        for(int i=k;i+k<=n;i++){
            ma=max(ma,a[i]-a[i-k]);
            ans=max(ans,ma+a[i+k]-a[i]);
        }
        cout<<ans<<endl;
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值