HDU-3516 Tree Construction(区间dp+四边形优化)

H - Tree Construction

  HDU - 3516 

/*
题意：有N个点，每个点的坐标为(Xi,Yi),对于j<k,Xj<Xk,Yj>Yk,现在用只能向上或向右走的线将所有点连接成一棵树，点为树的叶，问线的最小长度
题解：区间dp+四边形不等式优化

对于[i,j]可被分成[i,k]和[k+1,j]
假设区间[i,k]和[k+1,j]分别连成了一棵树，那么只要在用长度w(i,j)=(X(k+1)-Xi)+(Yk-Yj)即可将两个区间合并

考虑到N<=1000，用一般的O(N^3)的方法会超时，这里需要用四边形不等式优化

#include<bits/stdc++.h>
using namespace std;
const int MX = 1005;
const int INF = 0x3f3f3f3f;
int dp[MX][MX],a[MX],s[MX][MX],x[MX],y[MX];
int cal(int i,int j,int k){
    if(k>=j) return INF;
    return x[k+1]-x[i]+y[k]-y[j];
}
int main(){
    int n;
    //freopen("in.txt","r",stdin);
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++) scanf("%d%d",&x[i],&y[i]);
        for(int i=1;i<=n;i++) {
            dp[i][i]=0;
            s[i][i]=i;
        }
        for(int l=1;l<n;l++){
            for(int i=1;i+l<=n;i++){
                int j=i+l;
                dp[i][j]=INF;
                for(int k=s[i][j-1];k<=s[i+1][j];k++){
                    if(dp[i][k]+dp[k+1][j]+cal(i,j,k)<=dp[i][j]){
                        dp[i][j]=dp[i][k]+dp[k+1][j]+cal(i,j,k);
                        s[i][j]=k;
                    }
                }
            }
        }
        printf("%d\n",dp[1][n]);
    }
    return 0;
}