洛谷P5638 光骓者的荣耀(前缀和、dp)

洛谷P5638 光骓者的荣耀


这道题可以看成大小为n数组的和减去中间某一段连续的k个数,并且使得结果最小。

最直白的方法当然逐一尝试,用数组的和减去每一段连续的k个数,最后输出一个最小值,但是这样会超时。

所以这里介绍一个动态规划的方法,一个简单的dp,万物皆可dp

思路还是一样的,大小为n数组的和减去中间某一段连续的k个数,任何一段连续k个数都可能是我们需要减去的一段,所以我们从头开始,每一段都尝试,所以对于从头开始的每一部分数(也就是前缀)都有两种可能:要么减去这一段,要么保留这一段。如果保留的话,值就是前缀和,如果减去这一段的话就是抛去这一段以后的前缀和。所以说我们需要用前缀和来计算。
数组a[ ]存放数据,数组b[ ]存放前缀和,数组dp[ ]存放结果。
可以得到dp方程:dp[i]=min(dp[i-1]+a[i],b[i-k]) dp[ i ]为前i+1个数的最优值
我们假设k=3,假如从头开始的3个数(a[0],a[1],a[2])都不选,那么dp[0]~dp[2]就都为0,从a[3]开始,假设我们认定了前3个数是我们要减掉的最优的方法,那么a[3]加上dp[2]就应该是前4个数的最优的结果,如果a[3]加上dp[2]要大于第1个元素的值,就说明减掉前三个数不是最优解,最优解是减掉a[1],[2],a[3],留下a[0],a[0]的值其实也就是前缀和。这时dp[3]就可以更新了。以此类推。再看一下那个dp方程应该就可以理解了。

上代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[1000005],dp[1000005],b[1000005];
int main()
{
    int n,k;
    ll sum=0;
    scanf("%d %d",&n,&k);
    for(int i=0;i<n-1;i++){
        scanf("%lld",&a[i]);
        sum+=a[i];
        b[i]=sum;
    }
    if(k==0){
        printf("%lld\n",sum);
        return 0;
    }
    for(int i=0;i<k;i++) dp[i]=0;
    for(int i=k;i<n;i++) dp[i]=min(dp[i-1]+a[i],b[i-k]);
    printf("%lld\n",dp[n-1]);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值