AcWing每日一题 3499.序列最大收益（DP）

序列最大收益

原题链接

给定一个长度为 m 的整数序列 a1,a2,…,am。

序列中每个元素的值 ai 均满足 1≤ai≤n。

当一个值为 i 的元素和一个值为 j 的元素相邻时，可以产生的收益为 wi,j。

现在，我们可以从序列中删除最多 k 个元素，删除一些元素后，原本不相邻的元素可能会变得相邻。

序列的收益和为所有相邻元素对产生的收益之和，例如一个长度为 3 的整数序列 1,3,2 的收益和为 w1,3+w3,2。

请问，通过利用删除操作，能够得到的序列的最大收益和是多少？
数据范围
对于 30% 的数据，1≤n,k,m≤20。
对于 100% 的数据，1≤n,k,m≤200，0≤wi,j≤107，1≤ai≤n。
数据保证 wi,j=wj,i，wi,i=0。

输入格式
第一行包含三个整数 n,k,m。

第二行包含 m 个整数 a1,a2,…,am。

接下来 n 行，每行包含 n 个整数，其中第 i 行第 j 列的数表示 wi,j。

输出格式
输出序列的最大收益和。

输入样例：
4 1 3
1 4 2
0 3 0 1
3 0 0 0
0 0 0 0
1 0 0 0
输出样例：
3
样例解释
初始序列收益和为 w1,4+w4,2=1+0=1。

删除中间的 4 后，序列 1,2 的收益和为 w1,2=3。

---

很容易发现这道题是一道DP题，但是难点在于状态转移方程怎么写
这里我们用dp[i][j]表示前i个数删除j个数且保留第i个数的最大收益（注意：这里保留了第i个数，没有删除）

然后考虑状态转移

我们用u来表示i前面第一个没被删除的数

因为总共删除了j个数
所以[1,u]区间内就需要删除j-(i-u-1)个数
[1,u]区间也就有u-(j-(i-u-1))=u-(j-i+u+1)=i-j-1个数

那么状态转移方程就是
dp[i][j]=max(dp[i][j],dp[u][j-(i-u-1)]+w[a[u]][a[i]])

再来确定各个变量的范围

i：1<=i<=m //这个毋庸置疑
j：0<=j<=i-1&&j<=k
//因为要保留i，所以j到i-1，最多删除k个，所以小于等于k
u：i-j-1<=u<=i-1
//首先因为[1,u]中有i-j-1个元素，u最小为i-j-1，上限还是因为要保留一个数作为i，所以u小于等于i-1

AC代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 203;
int n, k, m;
int a[maxn];
ll w[maxn][maxn];
ll dp[maxn][maxn];
int main(){
	scanf("%d %d %d",&n,&k,&m);
	for(int i=1;i<=m;++i)
		scanf("%d",&a[i]);	
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			scanf("%lld",&w[i][j]);
	for(int i=1;i<=m;++i){
		for(int j=0;j<=i-1&&j<=k;++j){
			for(int u=i-(j+1);u<=i-1;++u){
				dp[i][j]=max(dp[i][j],dp[u][j-(i-u-1)]+w[a[i]][a[u]]);
			}
		}
	}
	ll ans=0;
	for(int i=0;i<=k;++i)ans=max(ans,dp[m][i]);
	printf("%lld",ans);
	return 0;
}

---

码风改变计划失败,我放弃了QWQ