CF940E Cashback 题解

最新推荐文章于 2021-11-23 16:07:55 发布

Plozia

最新推荐文章于 2021-11-23 16:07:55 发布

阅读量129

点赞数

分类专栏：动态规划文章标签：算法动态规划

本文链接：https://blog.csdn.net/BWzhuzehao/article/details/112280299

版权

动态规划专栏收录该内容

34 篇文章 1 订阅

订阅专栏

这道题好像很烦，我第一眼看过去确实没有任何思路。

但是仔细分析题目后，我们会发现：

$c = 1$ 时，答案为 0，但是好像没有这个点？
$c > n$ 时，答案为序列总和。
$c = n$ 时，答案为序列总和减去最小值。
$1 < c < n$ 且 $\times c$ 时，此时序列只能分为一段，而别的序列我们可以看成是长度为 1 的区间。
$1 < c < n$ 且 $\times c \leq n$ 时，我们可以将序列分为多段，而此时别的序列仍然可以分割成长度为 1 的区间。

因此只有两种区间：长度为 1 和长度为 $c$ 。

接下来主要考虑 $1 < c < n$ 的情况。

我们发现，此时问题就变成了一个 dp 问题。

设 $f_i$ 表示到 $a_i$ 截止时的答案，那么：

$\leq i < c$ 时： $f_i = \sum_{j = 1}^{i}a_j$
$\leq i \leq n$ 时： $f_i = \min{(f_{i - c} + sum_i - sum_{i - c} - minn_i,f_{i - 1} +a_i)}$

其中 $s u m$ 是前缀和数组， $minn_i$ 表示 $[i - c + 1, i]$ 这一段区间中的 $a_i$ 的最小值。

然后发现 $c$ 是确定的，于是对 $minn_i$ 直接单调队列优化即可，总复杂度 $O (n)$ 。

Code：

/*
======== Plozia =========
	Author:Plozia
	Problem:CF940E Cashback
	Date:2021/1/6
========= Plozia =========
*/

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int MAXN = 1e5 + 10;
int n, c, q[MAXN], l = 1, r = 0;
LL a[MAXN], f[MAXN], minn[MAXN], sum[MAXN];

LL read()
{
	LL sum = 0, fh = 1; char ch = getchar();
	while (ch < '0' || ch > '9') {if (ch == '-') fh = -1; ch = getchar();}
	while (ch >= '0' && ch <= '9') {sum = (sum << 3) + (sum << 1) + (ch ^ 48); ch = getchar();}
	return sum * fh;
}

int main()
{
	n = read(), c = read();
	for (int i = 1; i <= n; ++i) a[i] = read();
	for (int i = 1; i <= n; ++i)
	{
		while (l <= r && q[l] + c <= i) l++;
		while (l <= r && a[i] <= a[q[r]]) r--;
		q[++r] = i; minn[i] = a[q[l]]; sum[i] = sum[i - 1] + a[i];
	}
	for (int i = 1; i < c; ++i) f[i] = f[i - 1] + a[i];
	for (int i = c; i <= n; ++i) f[i] = min(f[i - c] + sum[i] - sum[i - c] - minn[i], f[i - 1] + a[i]);
	printf("%lld\n", f[n]);
	return 0;
}