Fork me on GitHub

Codeforces 13C Sequence --DP+离散化

题意:给出一个 n (1 <= n <= 5000)个数的序列 。每个操作可以把 n 个数中的某一个加1 或 减 1。问使这个序列变成非递减的操作数最少是多少 

解法:定义dp[i][j]为将前i个数变为以j为结尾的非递减序列的最少操作次数。

则有: dp[i][j] = min(dp[i][j], min(dp[i][k]) + Cost(原来第i个位置上的数转换到j))  (1 <= k <= j)

即前i个数以j结尾的状态可以由前i-1个数以小于等于j的k结尾的状态转移过来,取一个最小的转移

由于输入过大,需要离散化,将数组的复制b[]排序后用到unique函数,unique函数将数组中相邻元素重复的去掉(实际不去掉,只是放到最后面),然后返回最后一个不与其他元相同的数的地址,所以通过unique(b,b+n)-b可得到该数列实际不同的数有多少个,然后我们上述方程中的j就在这些里面取,Cost其实计算方法为abs(b[j]-a[i])

同时数组不可能存下5000*5000,由于每一个状态都从前一个推来,于是可以用滚动数组。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define lll __int64
using namespace std;
#define N 5007

const lll INF = (1LL<<60);
lll dp[2][N];
lll a[N],b[N];

int main()
{
    int n,i,j;
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {
        scanf("%I64d",&a[i]);
        b[i] = a[i];
    }
    sort(b,b+n);
    int tot = unique(b,b+n) - b;  //离散化成 0~tot
    for(i=0;i<tot;i++)
    {
        dp[0][i] = abs(b[i]-a[0]);
        dp[1][i] = INF;
    }
    int now = 1;
    for(i=1;i<n;i++)
    {
        lll mini = INF;
        for(j=0;j<tot;j++)
        {
            mini = min(dp[now^1][j],mini);
            dp[now][j] = min(dp[now][j],mini+abs(b[j]-a[i]));
        }
        now = 1-now;
        for(j=0;j<tot;j++)
            dp[now][j] = INF;
    }
    lll mini = INF;
    for(i=0;i<tot;i++)
        mini = min(mini,dp[1-now][i]);
    printf("%I64d\n",mini);
    return 0;
}
View Code
posted @ 2014-07-24 19:55  whatbeg  阅读(484)  评论(0编辑  收藏  举报