k e t t l e kettle kettle 题解
题目在这里。
题目大意
有一个长度为 n n n序列 a a a,可以做 k k k个操作,每次操作时选择一个 i ( 1 ≤ i ≤ n − 1 ) i(1\leq i\leq n-1) i(1≤i≤n−1),将第 i + 1 i+1 i+1个数加上第 i i i个数,并将第 i i i个数清 0 0 0。
解题方法
这道题的解题方法是贪心。
其实我们可以发现每一次选择一段进行操作就行了,也就是求
max
i
=
k
+
1
n
∑
j
=
i
−
k
i
a
j
\begin{aligned}\max_{i=k+1}^{n}{\sum_{j=i-k}^{i}{a_j}}\end{aligned}
i=k+1maxnj=i−k∑iaj。
因为这是个静态序列,所以我们可以直接用前缀和优化。
设
s
i
=
∑
i
=
1
n
a
i
\begin{aligned}s_i=\sum_{i=1}^{n}{a_i}\end{aligned}
si=i=1∑nai,则
答案为
max
i
=
k
+
1
n
s
i
−
s
i
−
k
−
1
\begin{aligned}\max_{i=k+1}^{n}{s_i-s_{i-k-1}}\end{aligned}
i=k+1maxnsi−si−k−1。
时间复杂度为
O
(
n
)
O(n)
O(n)。
代码
#include<bits/stdc++.h>
#define N 1000001
using namespace std;
int n,k,b[N],ans,f[N];
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&b[i]),f[i]=f[i-1]+b[i];
for(int i=k+1;i<=n;i++) ans=max(ans,f[i]-f[i-k-1]);
printf("%d",ans);
}