当前位置:编程学习 > C/C++ >>

Hdu 2829 Lawrence (DP_四边形优化、斜率优化)

题目大意:给定一个长度为n的序列,至多将序列分成m段,每段序列都有权值,权值为序列内两个数两两相乘之和。m<=n<=1000.

解题思路:经典的DP优化题,可以用四边形不等式优化也可以用斜率优化,我三种方法实现,两种斜率优化,一种四边形不等式,其中复杂度都为n*m,但是常熟略有差异。
    状态转移方程很好想,dp[i][j] = min(dp[i][j],dp[k][j-1]+cost[k+1][j])(1<=k<i),这种方程普通写法是n*n*m,当n为1000时运算量为10亿级别,必须优化。
   
     第一种:四边形不等式优化,这种方法是最简单的,主要是减少枚举k的次数。cost[i][j]是某段区间的权值,当区间变大,权值也随之变大,区间变小,权值也随之变小,此时就可以用四边形不等式优化。
     我们设s[i][j]为dp[i][j]的前导状态,即dp[i][j] = dp[s[i][j][j-1] + cost[s[i][j]+1][j].之后我们枚举k的时候只要枚举s[i][j-1]<=k<=s[i+1][j],此时j必须从小到大遍历,i必须从大到小。
     用这种方法我的代码跑了140ms。

    第二种:斜率优化.其实是借鉴大牛大思路,Here,我只是抛砖引玉而已。这种方法的dp和suma数组必须为64位整数,因为平方和会超过32位整数。
    用这种方法我的代码跑了350ms。

    第三种:斜率优化.其实是借鉴大牛大思路,Here,我只是抛砖引玉而已。其实这题可以作为模板题,斜率优化大抵如此吧。
    用这种方法我的代码跑了109ms。

测试数据:
Input:
4 1
4 5 1 2
4 2
4 5 1 2
5 3
1 2 1 2 1
6 4
7 5 3 6 8 9
10 3
1 4 2 7 5 6 8 5 6 9

OutPut:
17
2
92
15
187

C艹代码:
[cpp]
//四边形不等式 
#include <stdio.h> 
#include <string.h> 
#define MAX 1100 
#define INF (1<<30) 
 
 
int n,m,sum[MAX],cost[MAX][MAX]; 
int arr[MAX],dp[MAX][MAX],s[MAX][MAX]; 
 
 
void Initial() { 
 
    int i, j, k; 
 
 
    for (i = 1; i <= n; ++i) 
        for (j = 1; j <= n; ++j) 
            if (j < i) cost[i][j] = 0; 
            else cost[i][j] = cost[i][j - 1] + arr[j] * (sum[j - 1] - sum[i - 1]); 
    for (i = 0; i <= n; ++i) { 
 
        dp[i][0] = cost[1][i]; 
        s[i][0] = 0,s[n+1][i] = n; 
    } 

int Solve_DP() { 
 
    int i,j,k; 
 
 
    for (j = 1; j <= m; ++j) 
        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] + cost[k+1][i] < dp[i][j]) { 
 
                    s[i][j] = k; 
                    dp[i][j] = dp[k][j-1] + cost[k+1][i]; 
                } 
        } 
 
 
    return dp[n][m]; 

 
 
int main() 

    int i,j,k; 
 
 
    while (scanf("%d%d",&n,&m),n+m) { 
 
        for (i = 1; i <= n; ++i) 
            scanf("%d",&arr[i]),sum[i] = arr[i] + sum[i-1]; 
 
 
        Initial(); 
        int ans = Solve_DP(); 
        printf("%I64d\n",ans); 
    } 

[cpp] 
//sum[i] = arr[1] + .. arr[i]^2 
//sum2[i] = arr[1]^2 + .. arr[i]^2; 
//dp[i][j] = min{dp[k][j-1] -sum[i] * sum[k] + (suma[k] - sum[k]^2)/2 + (sum[k]^2 - suma[k])/2}; 
//斜率优化二 
#include <stdio.h> 
#include <string.h> 
#define MAX 1100 
#define INF (1<<30) 
#define int64 __int64//long long 
 
 
struct point { 
 
    int64 x,y; 
}pot[MAX]; 
int head,tail,qu[MAX]; 
int n,m,arr[MAX]; 
int64 sum[MAX],sum2[MAX],dp[MAX][MAX]; 
 
 
void Initial() { 
 
    for (int i = 1; i <= n; ++i) { 
 
        sum[i] = arr[i] + sum[i-1]; 
        sum2[i] = arr[i] * arr[i] + sum2[i-1]; 
        dp[i][0] = dp[i-1][0] + arr[i] * sum[i-1]; 
    } 

int CheckIt(point p0,point p1,point p2) { 
 
    return (p0.x-p1.x) * (p0.y-p2.y) - (p0.y-p1.y) * (p0.x-p2.x) <= 0; 

int NotBest(point p0,point p1,int k) { 
 
    return p0.y - k * p0.x > p1.y - k * p1.x; 

int Solve_DP() { 
 
    int i,j,k; 
 
 
    for (j = 1; j <= m; ++j) { 
 
        head = 0,tail = 0; 
        qu[tail] = 0; 
        for (i = j + 1; i <= n; ++i) { 
 
            pot[i].x = sum[i-1]; 
            pot[i].y = dp[i-1][j-1] + (sum[i-1] * sum[i-1] + sum2[i-1]) / 2; 
            while (head <= tail - 1 && 
                    CheckIt(pot[qu[tail-1]],pot[qu[tail]],pot[i])) tail--; 
 
 
            qu[++tail] = i; 
            while (head + 1 <= tail && 
                    NotBest(pot[qu[head]],pot[qu[head+1]],sum[i])) head++; 
            k = qu[head]; 
            //dp[i][j] = y - k * x + c 
            dp[i][j] = pot[k].y - sum[i] * pot[k].x + (sum[i] * sum[i] - sum2[i]) / 2; 
        } 
 &nbs

补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,