poj 1160 Post Office(四边形不等式优化dp)

转载自：http://www.cnblogs.com/staginner/archive/2012/03/12/2391925.html          

POJ_1160

    我们可以用f[i][j]表示建好i个邮局时覆盖到第j个村庄的最优解，那么就可以得到f[i][j]=min{f[i-1][k]+w[k+1][j]}(k<j)，其中w[x][y]表示建一个邮局覆盖x到y的村庄的距离和，w[x][y]可以事先预处理出来。

    这个题目还可以用四边形不等式去优化，实际上四边形不等式优化难点不在于应用，只是在K[i][j-1]<=k<=K[i+1][j]中去选择更新f[i][j]的k即可，比较复杂的部分就在于对k可以这样选择做出证明。

    一般四边形不等式的证明步骤如下：

    ①证明w为凸，这一步用黑书上的定理w为凸当且仅当w[i][j]+w[i+1][j+1]<=w[i][j+1]+w[i+1][j]，这样只要说明w[i+1][j]-w[i][j]是关于j单调递减的即可。

    ②证明f为凸，这一步要利用①中的定理去证f[i][j]+f[i+1][j+1]<=f[i][j+1]+f[i+1][j]，而证明的方法通常是利用w为凸的结论，先假设f[i][j+1]取得最优解是k为x，f[i+1][j]取得最优解时f[i+1][j]为y，然后分x<y和y<x两种情况，将f[i][j]和f[i+1][j+1]各按k=x或k=y拆开之后，将拆出的w应用四边形不等式，再将各项合并成f[i][j+1]+f[i+1][j]从而完成证明。

    ③证明K[i][j-1]<=K[i][j]<=K[i+1][j]，证明K[i][j-1]<=K[i][j]时，要先假设f[i][j-1]取得最优解时k=y，然后利用x<=y<=j-1<j列一个四边形不等式，然后在不等式两边添加一定的项试图得到f[i][j-1](k=x)+f[i][j](k=y)<=f[i][j-1](k=y)+f[i][j](k=x)，也就是f[i][j-1](k=x)-f[i][j-1](k=y)<= f[i][j](k=x)-f[i][j](k=y)，这时我们就会发现因为f[i][j-1](k=y)<=f[i][j-1](k=x)，那么一定有f[i][j](k=y)<=f[i][j](k=x)，也就是说对于所有小于y的x，都会有f[i][j-1](k=y)&lt;=f[i][j-1](k=x)，那么也都会有f[i][j](k=y)<=f[i][j](k=x)，因此令f[i][j]取得最优解的k一定不小于y，这样就完成了对K[i][j-1]<=K[i][j]的证明。对于K[i][j]<=K[i+1][j]的证明是类似的。

代码如下：复杂度O(np)

#include<cstdio>
#include<cstdio>
#include<cmath>
#include<queue>
#include<stack>
#include<string>
#include<cstring>
#include<iostream>
#include<map>
#include<vector>
#include<algorithm>
#include<set>
#include<cmath>
using namespace std;
const int nn = 3100;
const int inf = 0x3fffffff;
int n,p;
int a[nn];
int sum1[nn];
int sum2[nn];
int dp[nn][35];
int s[nn][35];
int getw(int l,int r)
{
    int mid=(l+r)/2;
    return sum1[r]-sum1[mid-1]-(r-mid+1)*a[mid]+sum2[l]-sum2[mid+1]-(mid-l+1)*(a[n]-a[mid]);
}
int main()
{
    int i,j,k;
    while(scanf("%d%d",&n,&p)!=EOF)
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        sum1[0]=0;
        for(i=1;i<=n;i++)
        {
            sum1[i]=sum1[i-1]+a[i];
        }
        sum2[n+1]=0;
        for(i=n;i>=1;i--)
        {
            sum2[i]=sum2[i+1]+a[n]-a[i];
        }
        for(j=0;j<=p;j++)
        {
            dp[0][j]=0;
            s[0][j]=0;
        }
        for(i=1;i<=n;i++)
        {
            dp[i][0]=inf;
            s[i][0]=0;
        }
        for(j=1;j<=p;j++)
        {
            s[n+1][j]=n-1;
            for(i=n;i>=1;i--)
            {
                dp[i][j]=inf;
                for(k=s[i][j-1];k<=s[i+1][j];k++)
                {
                    if(dp[k][j-1]+getw(k+1,i)<dp[i][j])
                    {
                        dp[i][j]=dp[k][j-1]+getw(k+1,i);
                        s[i][j]=k;
                    }
                }
            }
        }
        printf("%d\n",dp[n][p]);
    }
    return 0;
}