最大的和(元素和）

题目描述
给定一个长度为 n 的正整数数列 a1,a2,…,an。

初始时，数列中的每个元素要么处于可选状态，要么处于不可选状态。

你可以选择一个长度恰好为 k 的区间 [i,i+k−1]，使得 ai∼ai+k−1 这 k 个元素的状态全部变为可选。

请问，在经过此操作后，所有处于可选状态的元素之和最大是多少。

输入格式
第一行包含两个整数 n 和 k。

第二行包含 n 个整数 ai。

第三行包含一个长度为 n 的 01 序列，如果第 i 个数为 1，表示 ai 的初始状态为可选，如果第 i 个数为 0，表示 ai 的初始状态为不可选。

输出格式
一行一个整数，表示答案。

数据范围
对于 30% 的数据，1≤k≤n≤1000
对于 100% 的数据，1≤k≤n≤105,1≤ai≤105

样例
输入样例1：
3 1
2 5 4
0 0 1
输出样例1：
9
输入样例2：
4 3
10 5 4 7
0 1 1 0
输出样例2：
19

思路一：前缀和
首先找出初始状态为不选的区间和最大值，然后加上状态为选的数，即为本题答案
听起来很简单，实际上有个小坑点，就是注意区间长度k的下标;
如果是从下标i开始计算，那么区间最后一个数应该是i+k-1(下标)
我的脑袋瓜子表示为啥搞不懂，不应该是i+k吗？hhh
这个包括了i本身，所以是i+k-1;
噢噢，明白了。开心到起飞！！！

时间复杂度 O(n)
参考文献
C++ 代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6;
typedef long long LL;
LL a[N],b[N],c[N];
bool st[N];
int n,m;
int main()
{
    cin>>n>>m;
    for (int i = 1; i <= n; i ++ )cin>>a[i];
    for (int i = 1; i <= n; i ++ )cin>>st[i];
    for (int i = 1; i <= n; i ++ )
    {
        if(st[i])b[i]=a[i];
        else c[i]=a[i];
    }
    for (int i = 1; i <= n; i ++ )b[i]+=b[i-1],c[i]+=c[i-1];
    LL res=0;
    for (int i = 1; i+m-1<=n; i ++ )
    {
        int j=i+m-1;
        res=max(res,c[j]-c[i-1]);
    }
    return cout<<res+b[n],0;
}

思路二：滑动窗口
选的数是必须加上去的，至于未选的数，是利用滑动窗口进行计算
求滑动窗口内最大的区间和即可；
当然要注意滑动过程中删除最左边的数
我的脑袋瓜子表示理解了，万岁！！

时间复杂度 时间复杂度O(n)
参考文献
C++ 代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6;
typedef long long LL;
LL a[N],b[N];
LL n,m,sum,res,ret;
int main()
{
    cin>>n>>m;
    for (int i = 1; i <= n; i ++ )cin>>a[i];
    for (int i = 1; i <= n; i ++ )
    {
        cin>>b[i];
        if(b[i])sum+=a[i];
    }
    for (int i = 1; i <= n; i ++ )
    {
        if(!b[i])res+=a[i];
        if(i>m&&!b[i-m])res-=a[i-m];
        ret=max(ret,res);
    }
    return cout<<ret+sum,0;
}

Java代码

import java.util.*;
import java.io.*;
public class Main{
    static long res,sum,ret;
    static int a[]=new int [100010];
    static int b[]=new int [100010];
    public static void main(String[] args){
        Scanner cin = new Scanner(System.in);
        int n=cin.nextInt(),m=cin.nextInt();
        for(int i=1;i<=n;i++)a[i]=cin.nextInt();
        for(int i=1;i<=n;i++)
        {
            b[i]=cin.nextInt();
            if(b[i]==1)sum+=a[i];
        }
        for (int i = 1; i <= n; i ++ )
        { 
            if(b[i]==0)res+=a[i];
            if(i>m&&b[i-m]==0)res-=a[i-m];
            ret=Math.max(ret,res);
        }
        System.out.print(ret+sum);
        return ;
    }
}

Go语言

package main
import "fmt"
var a[100010]int
var b[100010]int
func main(){
    var n,m,res,ret,ans int
    fmt.Scanf("%d %d",&n,&m)
    a:=make([]int,100010)
    b:=make([]int,100010)
    for i:=1 ; i<=n ; i++ {
        fmt.Scanf("%d",&a[i])
    }
    for j:=1 ; j<=n ; j++ {
        fmt.Scanf("%d",&b[j])
        if(b[j]==1){
            res+=a[j]
        }
    }
    for i:=1;i<=n;i++ {
       if(b[i]==0){
           ret+=a[i]
       }
        if(i>m&&b[i-m]==0){
            ret-=a[i-m]
        }
        ans=max(ret,ans);
    }
    fmt.Printf("%d",ans+res)
}
func max(num1, num2 int) int {
   var result int
   if (num1 > num2) {
      result = num1
   } else {
      result = num2
   }
   return result
}

欢迎留言点赞