Making the Grade
题目链接
题目大意:给你一个数组,求将其修改为单增或单减(非严格)所需要的最少代价,修改一次的代价为|初值-末值|。
以单增为例:
所有更改后的数字必定在原数组中出现过。
dp[i][j]表示前i个数都满足要求,且第i个数字取j时所要的代价之和,但由于a[i]取值范围到了1E9,故要进行离散化——第i个数字取第j小的数字(sort)
d p [ i ] [ j ] = a b s ( a [ i ] − b [ j ] ) + m i n ( d p [ i − 1 ] [ p ] ) 且 p 取 1 到 j dp[i][j]=abs(a[i]-b[j])+min(dp[i-1][p]) 且p取1到j dp[i][j]=abs(a[i]−b[j])+min(dp[i−1][p])且p取1到j
#include<iostream>
#include<vector>
#include<cstring>
#include<algorithm>
#include<string>
#include<cstdio>
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;
const int maxn = 2e3 + 10;
//所有更改后的数字必定在原数组中出现过
ll dp[maxn][maxn];//前i-1个数符合规则,且第i个数取值为第j小的数时 代价的总和
int a[maxn], b[maxn];
int main()
{
int n;
while (cin >> n)
{
for (int i = 1; i <= n; i++)
{
cin >> a[i];
b[i] = a[i];
}
sort(b + 1, b + 1 + n);
int len = unique(b + 1, b + 1 + n) - (b + 1);
//可以不用unique优化,但速度会变慢
mem(dp, 0);
for (int i = 1; i <= n; i++)
{
ll mn = 0x3f3f3f3f3f3f3f;
for (int j = 1; j <= len; j++)
{
mn = min(mn, dp[i - 1][j]);
dp[i][j] = abs(a[i]-b[j]) + mn;
//b[i]为原数字,a[j]为目标数字
}
}
ll ans = 0x3f3f3f3f3f3f3f;
for (int i = 1; i <= len; i++)
{
ans = min(ans, dp[n][i]);
}
cout << ans << "\n";
}
}