首页 > 代码库 > 石子合并(四边形不等式优化dp)

石子合并(四边形不等式优化dp)

该来的总是要来的————————

 

经典问题,石子合并。

  对于 f[i][j]= min{f[i][k]+f[k+1][j]+w[i][j]}

 

From 黑书

凸四边形不等式:w[a][c]+w[b][d]<=w[b][c]+w[a][d](a<b<c<d)

区间包含关系单调: w[b][c]<=w[a][d](a<b<c<d)

定理1:  如果w同时满足四边形不等式和决策单调性 ,则f也满足四边形不等式

定理2:  若f满足四边形不等式,则决策s满足 s[i][j-1]<=s[i][j]<=s[i+1][j]

定理3: w为凸当且仅当w[i][j]+w[i+1][j+1]<=w[i+1][j]+w[i][j+1]   

 

简要证明:

若w[a][c]+w[b][d]<=w[b][c]+w[a][d],归纳证明f[a][c]+f[b][d]<=f[b][c]+f[a][d]

  设f[a][d]最优决策是在s取到,f[b][c]最优决策在t取到,设s<t,反之同理

  可知a<s<t<c<d

    f[a][c]+f[b][d]<=f[a][s]+f[s+1][c]+w[a][c] + f[b][t]+f[t+1][d]+w[b][d]

            =f[a][s]+f[s+1][c]+w[a][d] + f[b][t]+f[t+1][d]+w[b][c]

           <=f[a][s]+w[a][d]+f[s+1][d] + f[b][t]+w[b][c]+f[t+1][c]         归纳得到 sc+td<sd+tc  起始条件即定理3

           =f[a][d]+f[b][c]

得证.

若f[a][c]+f[b][d]<=f[b][c]+f[a][d],则s[i][j-1]<=s[i][j]<=s[i+1][j]

  仅证s[i][j-1]<=s[i][j],右边同理

  记f_k[i][j]=f[i][k]+f[k+1][j]+w[i][j]

  记s点为[i,j]最优点,t点为[i,j+1]最优点,

  则只需证明 在[i,j+1]决策时, 取s点能够比取在k∈[i,s-1]的点更优即可

    即证明 f_s[i,j+1]<=f_k[i,j+1]

  又因为f_s[i,j]<=f_k[i,j]

     只需证明 0 <= f_k[i,j] - f_s[i,j] <= f_k[i,j+1] - f_s[i,j+1]

      可发现右边即 f_k[i,j] + f_s[i,j+1] <= f_k[i,j+1] + f_s[i,j]  

      展开后即: f[k][j] + f[s][j+1] <= f[k][j+1] + f[s][j]

      正是 k<s<j<j+1 的四边形不等式

得证.

 

一般利用定理3证明凸函数,然后利用定理2的结论 s[i][j-1]<=s[i][j]<=s[i+1][j]

  就能够使得复杂度由O(n^3)降低为O(n^2)

详细证明参见《动态规划算法的优化技巧》--毛子青(会因为论文用i,j,i‘,j‘搞得雾水,但是慢慢推一下就能够出来)

 

#include <cstdio>#include <cstring>#define N 1005int s[N][N],f[N][N],sum[N],n;int main(){    while(scanf("%d",&n)!=EOF)    {        memset(f,127,sizeof(f));        sum[0]=0;        for(int i=1; i<=n; i++){            scanf("%d",&sum[i]);            sum[i]+=sum[i-1];            f[i][i]=0;            s[i][i+1]=i;        }        for(int i=1; i<=n; i++)            f[i][i+1]=sum[i+1]-sum[i-1];        for(int i=n-2; i>=1; i--)            for(int j=i+2; j<=n; j++)                for(int k=s[i][j-1]; k<=s[i+1][j]; k++)                if(f[i][j]>f[i][k]+f[k+1][j]+sum[j]-sum[i-1])                {                    f[i][j]=f[i][k]+f[k+1][j]+sum[j]-sum[i-1];                    s[i][j]=k;                }            printf("%d\n",f[1][n]);    }    return 0;}