poj1160 Post Office 四边形不等式优化

四边形不等式的动态规划优化算法不多说,只对它做一个简单的介绍:这个是优化动态规划算法时间复杂度的方法之一,空间复杂度只会增加几个常数,不用考虑。方法说实话,验证是极其麻烦的,所以很多时候我都是要看题解再知道,这里也就不多说了。
直接上代码:

#include
#include
#include
#include
#include
#include
#include
#include

using namespace std;

#define REP(i, a, b) for(i = (a);i <= (b); ++i)

const int INF = 1000000000;
const int maxn = 500;

int n, m;
int a[maxn], d[maxn][maxn], dp[maxn][maxn];

int main() {
    int i, j, k;
    scanf("%d", &n);
    scanf("%d", &m);
    REP(i, 1, n)
        scanf("%d", &a[i]);
    REP(i, 1, n)
        REP(j, i + 1, n) {
            int mid = (i + j) >> 1;
            REP(k, i, j)
                d[i][j] += abs(a[k] - a[mid]);
        }
    REP(i, 1, m)
        REP(j, 1, n)
            dp[i][j] = INF;
    REP(i, 1, n)
        dp[1][i] = d[1][i];
    REP(i, 2, m)
        REP(j, i, n)
            REP(k, 1, j)
                if(dp[i][j] > dp[i - 1][k] + d[k + 1][j])  
                    dp[i][j] = dp[i - 1][k] + d[k + 1][j]; 
    printf("%d\n", dp[m][n]);
    return 0;
}

你可能感兴趣的:(poj,动态规划)