题目描述 来源
常州大学组织了新生寒假训练一共N天,每天训练可以获得的训练效果是Ei。但是如果连续训练超过K天,萌新们会受不了而被劝退。
现在负责人想知道,如何安排能保证萌新不会被劝退并且能获得最大的训练效果。
输入描述:
第一行:两个用空格隔开的整数:N和K,1≤N≤100000,1≤K≤N
第二行到N+1行:第i+1行有一个整数,表示第N天的训练效果是Ei,(0 <= Ei <= 1,000,000,000)
输出描述:
第一行:单个整数,表示最大的能力之和
示例1
输入
5 2 1 2 3 4 5
输出
12
说明
(除了第三天以外每天都在训练,总训练效果为1+2+4+5=12)
备注:
1≤n≤100,000
代码来源
#include<bits/stdc++.h> using namespace std; #define LL long long const int N = 1e5+11; const LL inf = 0x3f3f3f3f3f3f3f3f ; int q[N],st,ed;//q数组中的元素满足递增规律,作为单调栈 LL dp[N],a[N]; int main(){ int n,k; while(scanf("%d%d",&n,&k)!=EOF){ LL sum=0; for(int i=1;i<=n;i++) scanf("%lld",&a[i]),sum+=a[i]; for(int i=1;i<=n;i++){ while(i-q[st]>k+1) st++; dp[i]=a[i]+dp[q[st]]; while(st<=ed && dp[q[ed]]>dp[i]) --ed; q[++ed]=i; } LL ans=inf; for(int i=n-k;i<=n;i++) ans=min(ans,dp[i]); printf("%lld\n",sum-ans); } return 0; }