2020复旦大学机试E序列

这道题是一个动态规划，dp[i][j]的意思就是第i个位置，填j这个数字，然后去计算，j就遍历0-9。

#include <stdio.h>
#include <iostream>
#include <vector>
#pragma warning(disable:4996);
using namespace std;
#define N 10
const int maxn = 100010;
const int INF = 0x3f3f3f3f;
int a[maxn];
int main()
{
	int n;
	scanf("%d", &n);
	vector<vector<int> > dp(N,vector<int>(N,INF));
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &a[i]);
	}
	
	//递归边界
	for (int i = 0; i <= 9; i++)
	{
		dp[1][i] = abs(i-a[1]);
	}
	//开始递归
	//dp[i][j]表示到第i位的时候，填j可以使得的权值最大是多少
	for (int i = 2; i <= n; i++)
	{
		for (int j = 0; j <= 9; j++)
		{
			for (int k = 0; k <= 9; k++)
			{
				//这一位填什么比较好
				dp[i][j] = min(dp[i][j],dp[i-1][k]+abs(a[i]-j)+(j-k)*(j-k));
			}
		}
	}
	int ans = INF;
	for (int i = 0; i <= 9; i++)
	{
		ans = min(ans,dp[n][i]);
	}
	printf("%d", ans);
}