计蒜客 - T3337

一、题意： 给定n个数字，参数m，k，在n个数字中选择k个数字，选择第i个数字的花费为
j为上一次选择的数字的下标，问如何选择才能使得花费最小。
二、思路： 我们可以用dp来解决，设dp[i][j]为选择到i个数并且j为第i个数的花费，则可以有状态转移方程为
*dp[j][i] = min(dp[j][i], dp[t][i - 1] + a[j] * a[j] + max(p, m - j + t)max(p, m - j + t)); t为k-1次选择的第k-1个数字的下标。
在这里需要做一个小小的优化，假设选择了a[i]和a[j]两个数字，则总的花费最小为a[i]*a[i]+a[j]*a[j]+min(max(m-i+j,0)²，max(m-i+j,0)²)
所以总花费的大小取决于m-i+j的大小，也就是a[i]和a[j]的选择先后顺序，很显然i>j时可以获得花费的最小值。
cpp代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#define int long long 
using namespace std;
const int inf = 1e9 + 7;
int a[1009];
int dp[1009][1009];
signed main()
{
	int n, m, k;
	scanf("%lld %lld %lld", &n, &m, &k);
	for (int i = 1; i <= n; i++)scanf("%lld", &a[i]);
	for (int i = 1; i <= n; i++)for (int j = 1; j <= n; j++)dp[i][j] = 1e18;
	int p = 0;
	for (int i = 1; i <= n; i++) {
		dp[i][1] = a[i] * a[i] + max(p, m - i)*max(p, m - i);
	}
	for (int i = 2; i <= k; i++) {
		for (int j = n; j >=1; j--) {
			for (int t = 1; t<j; t++) {
				dp[j][i] = min(dp[j][i], dp[t][i - 1] + a[j] * a[j] + max(p, m - j + t)*max(p, m - j + t));
			}
		}
	}
	int ans = 1e18;
	for (int i = 1; i <= n; i++)
		ans = min(ans, dp[i][k]);
	printf("%lld", ans);
	return 0;
}