【LGR-161-Div.3】洛谷基础赛 #4 T3 - P9688 Colo.

是阿荣的荣

已于 2023-10-11 20:08:38 修改

阅读量203

点赞数

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

于 2023-10-11 19:55:31 首次发布

本文链接：https://blog.csdn.net/m0_55529377/article/details/133777053

版权

动态规划专栏收录该内容

5 篇文章 0 订阅

订阅专栏

下午没课，vp了一遍赛题，T4最高80分，就是A不了..

em..基础赛都AK不了.. 说明基础不行哇 - -...

T1、2太简单了，不想写题解，写T3吧。。（bushi,再写下去作业就写不完了>_<

赛题url：Colo. - 洛谷；

题目描述

小 F 和小 Y 经常在一起玩耍，因为小 F 是一个画家，他喜欢在一个长度为 n，宽度为 1 的网格图上画画，从左往右第 i 个方格被涂成了一种颜色 ai。

你觉得他的随意涂鸦太难看了，想要保留恰好 k 种颜色（你不能保留没在网格图上出现的颜色），使得网格图上没被涂成任何一种你喜欢的颜色的网格都被剪掉，最后会剩下一些网格，你希望这些网格从左到右颜色的编号是单调不下降的。

此外，小 Y 使用的第 i种颜色有一个价值 bi，小 Y 看到了你裁剪后的网格图很是高兴，于是决定付给你你选择的颜色的价值总和。

你需要求出你能够获得的最大的价值是多少。

输入格式

第一行两个整数 n,k，表示小 Y 画画的网格图的大小和你需要保留颜色的种类数。
第二行 n 个整数 ai，表示小 Y 画出来的网格图从左往右第 i 个格子的颜色。
第三行 n 个整数 bi，表示第 i 种颜色的价值。

输出格式

一行一个整数，表示你能够获得的最大价值；特别地，如果无法找到选择颜色的方法满足要求，输出 −1。

样例1输入

5 2
1 2 1 3 2
5 3 1 100 100

样例1输出

样例2输入

10 3
1 3 4 2 9 3 4 2 5 1
1 5 2 3 9 8 1 2 3 10

样例2输出

-1

思路：

DP（想了好久，because ben)。。

状态表示：保留前 i 种颜色，且以颜色 j 作为最右端能取得的最大价值；

> 记录数组a中的每个颜色第一次出现的位置和最后一次出现的位置，不妨设为first[a[i]]和last[a[i]]，以及1-n中的数字在数组a中出现的次数，设为nums[i]，数组b为保留颜色 i 能取到的价值。

> 令 i ∈ (1, k](保留1个颜色能取得的最大价值等于b[i]) , j ∈ [1, n]，x ∈ [1,j) ;

> j 从1到n枚举，当nums[j]存在时，若ョx 使得 $last[x] < first[j]$ ，且上一个状态以x为右端点的最大价值不为0，则当前以j为右端点的状态可以由dp[i-1][x]加上当前j的价值b[j]求得；

> 状态转移：

                                  $dp[i][j] = max(dp[i][j], dp[i-1][x] + b[j])$

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int n,k; 
int a[510],b[510],last[510],first[510],nums[510];
LL dp[510][510],ans;
int main(){
	ios::sync_with_stdio(false);
	cin >> n >> k;
	for (int i = 1; i <= n; i ++){
		cin >> a[i];
		last[a[i]] = i;
		nums[a[i]] ++;
		first[a[i]] = (nums[a[i]] == 1?i:first[a[i]]);
	}
	for (int i = 1; i <= n; i ++){
		cin >> b[i];
		dp[1][i] = b[i];
	}
	for (int i = 2; i <= k; i ++){
		for (int j = 1; j <= n; j++){
			if (!nums[j]) continue;
			for (int x = 1; x < j; x ++){
				if (nums[x] && last[x] < first[j] && dp[i - 1][x]){
					dp[i][j] = max(dp[i][j],dp[i-1][x] + b[j]);
				}
			}
		}
	}
	for (int i = 1; i <= n; i++) ans = max(ans, dp[k][i]);
	if (ans == 0) cout << -1 << endl;
	else cout << ans << endl;
	return 0;
}

结果：