前缀和优化dp题集

---

Poj3612 Telephone Wire

题意：

n个电线杆，第i个电线杆的高度为h(i)，连接相邻两个个电线杆的花费为： |高度差|*c
此外可以加高一个电线杆，价高x需要x²的花费

你只能连接相邻的电线杆，问将1到n全部连接所需要的最小花费

数据范围 ：n<=1e5，h(i)<=100

解法：

d[i][j]表示前i个电线杆,第i个长度为j的最小花费
d[i][j]=min{d[i-1][k]+c*|j-k|+(j-h[i])^2}
最外层n,枚举当前长度,枚举上一层长度,总复杂度O(nk^2),需要优化
状态数量是nk个无法优化,考虑优化转移
观察方程:
d[i][j]=min{d[i-1][k]+c*|j-k|+(j-h[i])^2}
d[i][j]=min{d[i-1][k]+c*|j-k|}+(j-h[i])^2

试着将j和k分离,对绝对值分类讨论:
j>k,|j-k|=j-k,式子变为:
d[i][j]=min{d[i-1][k]-ck}+cj+(j-h[i])^2
j<k,|j-k|=k-j,式子变为:
d[i][j]=min{d[i-1][k]+ck}-cj+(j-h[i])^2
其中两个式子中的min部分可以用前缀max与后缀max优化至O(1)
在每次计算之前先处理前缀max与后缀max即可

此外可以滚动d数组的第一维

code：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxm=1e5+5;
const int inf=0x3f3f3f3f;
int l[105],r[105];
int h[maxm];
int d[2][105];//d[i]表示当前电线杆高度为i时的最小花费
int n,c;
int sq(int x){//平方
    return x*x;
}
signed main(){
    while(cin>>n>>c){
        for(int i=1;i<=n;i++)scanf("%d",&h[i]);
        int pos=0;
        //处理d[pos][1]
        for(int i=1;i<h[1];i++)d[pos][i]=inf;
        for(int i=h[1];i<=100;i++){
            d[pos][i]=sq(i-h[1]);
        }
        //
        for(int i=2;i<=n;i++){
            l[0]=r[101]=inf;
            for(int j=1;j<=100;j++){//前缀max
                l[j]=min(l[j-1],d[pos][j]-j*c);
            }
            for(int j=100;j>=1;j--){//后缀max
                r[j]=min(r[j+1],d[pos][j]+j*c);
            }
            pos^=1;
            for(int j=1;j<=100;j++)d[pos][j]=inf;//重置d数组
            for(int j=h[i];j<=100;j++){//枚举当前层高度
                d[pos][j]=sq(j-h[i])+min(l[j]+j*c,r[j]-j*c);
            }
        }
        int ans=inf;
        for(int i=1;i<=100;i++)ans=min(ans,d[pos][i]);
        cout<<ans<<endl;
    }
    return 0;
}

---