【线性DP】洛谷P2963 或 POJ 3666 修路

链接

http://poj.org/problem?id=3666
https://www.luogu.org/problemnew/show/P2893

---

大意

给定序列$a$,求出一个严格递增或递减的序列$b$，使得

$$\sum_{i=1}^{n}a[i]-b[i]$$

最小

---

思路

首先对$a$进行离散化存到$b$，至于为什么，我们可以用数学归纳法的方法来证明，但是我不会

然后，我们分单调递增和单调递减两种情况分类讨论，然后进行$xjb$乱$dp$

---

设$f[i,j]$表示完成前$i$个数的构造，$B_i==j$时的最小值

显然可以得到动态转移方程：

$$f[i,j]=min\{f[i-1,k]+|A_i-B_j|\}$$

其中,$k$为上一行$f[i-1]$中的最小值，我们可以开一个二维数组$mf$维护一下

---

代码

#include<cstdio> 
#include<cstring>
#include<algorithm>
#define r(i,a,b) for(register int i=a;i<=b;i++)
using namespace std;int f[2021][2021],mf[2021][2021],a[2021],b[2021],n,ans,m;
inline int dp()
{
    r(i,0,n)
     r(j,0,m) mf[i][j]=f[i][j]=0;//清0
    r(i,1,n)
     r(j,1,m)
    {
        f[i][j]=mf[~-i][j]+abs(a[i]-b[j]);//动态转移
        if(j!=1) mf[i][j]=min(mf[i][~-j],f[i][j]);
        else mf[i][j]=f[i][j];//维护f最小值
    }
    return mf[n][m];//返回
}
signed main()
{
    scanf("%d",&n);
    r(i,1,n) scanf("%d",a+i),b[i]=a[i];
    sort(b+1,b+1+n);
    m=unique(b+1,b+1+n)-(b+1);//离散化
    ans=dp();
    reverse(b+1,b+1+m);//翻转数组
    ans=min(ans,dp());
    printf("%d",ans);
}