首页 > 代码库 > POJ 3666 Making the Grade [DP]

POJ 3666 Making the Grade [DP]

题意:

给定一个序列,以最小代价将其变成单调不增或单调不减序列,这里的代价看题目公式。

思路:

很容易想到是DP。

1.

对前i个序列,构成的最优解其实就是与两个参数有关。一个是这个序列处理后的最大值mx,和这个序列处理的代价值cost。

显然最大值mx最小最好(这样第i+1个值可以不花代价直接接在其后面的可能性更大),cost最小也最好(题意要求),但是两者往往是鱼和熊掌。

用dp[i][j]表示:前i个数构成的序列,这个序列最大值为j,dp[i][j]的值代表相应的cost。

所以状态转移方程如下:

dp[i][j]=abs(j-w[i])+min(dp[i-1][k]);(k<=j)

这个表格是根据转移方程写出来的dp数组。

再仔细看一下转移方程:dp[i][j]=abs(j-w[i])+min(dp[i-1][k]);(k<=j)

右边没填充的是因为填充的数字肯定比前面的数字大,无用,因为在求min( dp[i-1][k] )时,是求最小值,既然更大,则最小值时无需考虑。

又从表格中可以看出:

dp[i][j]=abs(j-w[i])+min(dp[i-1][k]);(k<=j)这里的k无需从1遍历到j。

只要在对j进行for循环的时候不断更新一个dp[i-1][j]的最小值mn=min(mn,dp[i-1][j]),

然后对dp[i][j]=abs(j-w[i])+mn即可;

这样改进之后即可从本来的时候时间复杂度O(NMM)改进为O(NM);


但是,这里的m是A[i]的最大值,显然TLE。

所以必须用离散化思想改进,因为N=2000。远小于A[i]的最大值。

离散化:将序列排序一下,然后用位置的前后关系来制定其值,这样时间复杂度变成O(N^2).


最后是这题数据有bug,只需要求不减序列即可。


#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;
long long n;
long long a[2222];
long long b[2222];
long long dp[2222][2222];
long long Abs(long long a,long long b)
{
	return a>b?a-b:b-a;
}
void solve()
{
	long long minn=(1<<30);
	for(long long i=1;i<=n;i++)
	{
		minn=dp[i-1][1];
		for(long long j=1;j<=n;j++)
		{
			minn=min(minn,dp[i-1][j]);
			dp[i][j]=minn+Abs(a[i],b[j]);
		}
	}
	long long ans=0x7fffffffffffffffL;
	for(int i=1;i<=n;i++)
	{
		ans=min(ans,dp[n][i]);
	}
	printf("%lld\n",ans);
}
int main()
{
    //freopen("/home/rainto96/in.txt","r",stdin);
	scanf("%lld",&n);
	for(long long i=1;i<=n;i++)
	{
		scanf("%lld",a+i);
		b[i]=a[i];
	}
	sort(b+1,b+1+n);
	solve();
	return 0;
}