UVA 12170 Easy Climb(dp+ 单调队列)

题意:给出n个正整数,h1~hn,可以修改除了h1和hn的数,要求相邻的两个数的差距不超过d,并且让修改的费用最小,修改的费用为修改前后的数的差值。

思路:所有可能修改的数是有限的,不是所有数都会修改到。会修改道的数形如hp + kd,那么就将所有的可能数的状态是n^2,dp[i][j]表示处理完前i个数,第i个数为j时的最小花费,复杂度是n^4,不过利用单调队列可以优化到n^3,每次调整前一个数可选的区间就行了。数据有些水,错的代码也能ac……


代码:


#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-8
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn = 100 + 10;
ll dp[maxn][maxn*maxn*3],val[maxn*maxn*3],h[maxn],d;
int q[maxn*maxn*3],head,fail,tot,n;
inline void Insert(int p,int x)
{
    if(dp[p][x] == -1) return ;
    while(head < fail && dp[p][q[fail-1]] >= dp[p][x]) fail--;
    q[fail++] = x;
}
inline void Remove(int x)
{
    while(head < fail && val[q[head]] < val[x] - d) head++;
}
int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%lld",&n,&d);
        tot = 0;
        for(int i = 1;i <= n;++i)
        {
            scanf("%lld",&h[i]);
            for(int j = 0;j < n;++j)
            {
                val[tot++] = h[i] + j*d;
                if(h[i] - j*d >= 0)
                    val[tot++] = h[i] - j*d;
            }
        }
        sort(val,val + tot);
        tot = unique(val,val + tot) - val;
        memset(dp,0xff,sizeof(dp));
        int pos = lower_bound(val,val + tot,h[1]) - val;
        dp[1][pos] = 0;
        for(int i = 2 ;i <= n;++i)
        {
            head = 0;fail = 0;
            pos = 0;
            for(int j = 0;j < tot;++j)
            {
                Remove(j);
                while(pos < tot && val[pos] <= val[j] + d)
                {
                    Insert(i-1,pos);
                    pos++;
                }
                if(head < fail)
                    dp[i][j] = abs(h[i] - val[j]) + dp[i-1][q[head]];
            }
        }
        pos = lower_bound(val,val + tot,h[n]) - val;
        ll ans = dp[n][pos];
        if(ans == -1) puts("impossible");
        else printf("%lld\n",ans);
    }
    return 0;
}


你可能感兴趣的:(dp)