【差分贪心】JZOJ_4788 序列

最新推荐文章于 2021-12-14 21:58:36 发布

原创最新推荐文章于 2021-12-14 21:58:36 发布 · 272 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#差分 #贪心

贪心专栏收录该内容

17 篇文章

订阅专栏

博客围绕一道题展开，先阐述题意，接着给出思路。需计算ai表示转换成bi还需的量并进行差分，根据加的数是否超过4分类讨论，若不超4，答案为∑i=1nmax(0,ai)；若超4，让区间(l,r]加4，要使ar+4<al。最后给出代码。

题意

在这里插入图片描述

思路

求出 $a_i$ 表示还需多少才能转换成 $b_i$ ，把它差分。
如果加的数不超过 $4$ ，那么答案就为 $∑i=1nmax(0,ai)\sum_{i=1}^{n}max(0,a_i)$ 。
否则如果让区间 $(l, r]$ 加上4，即 $a_l-4,a_r+4$ ，那么就要使 $a_r+4<a_l$ ，分类讨论。

代码

#include<cstdio>

int t, n, ans;
int a[100002];

int main() {
	scanf("%d", &t);
	while (t--) {
		scanf("%d", &n);
		for (int i = 1; i <= n; i++)
			scanf("%d", &a[i]);
		int x;
		for (int i = 1; i <= n; i++) {
			scanf("%d", &x);
			a[i] = (x - a[i] + 4) % 4;
		}
		ans = 0;
		for (int i = 1; i <= n; i++) {
			a[i] -= a[i + 1];
			if (a[i] > 0) ans += a[i];
		}
		int t2 = 0, t3 = 0;
		for (int i = 1; i <= n; i++) {
			if (a[i] == -3) {
				if (t3) {
					t3--;
					ans -= 2;
				} else if (t2) {
					t2--;
					ans--;
				}
			}
			else if (a[i] == -2) {
				if (t3) {
					t3--;
					t2++;
					ans--;
				}
			}
			else {
				t2 += a[i] == 2;
				t3 += a[i] == 3;
			}
		}
		printf("%d\n", ans);
	}
}