石子合并问题 (区间dp)

叁歲伎倆 2022-09-28 05:53 282阅读 0赞

石子合并问题是最经典的DP问题。首先它有如下3种题型:

(1)有N堆石子,现要将石子有序的合并成一堆,规定如下:每次只能移动任意的2堆石子合并,合并花费为新合成的一堆石子的数量。求将这N堆石子合并成一堆的总花费最小(或最大)。

分析:当然这种情况是最简单的情况,合并的是任意两堆,直接贪心即可,每次选择最小的两堆合并。本问题实际上就是哈夫曼的变形。

(2) 有N堆石子,现要将石子有序的合并成一堆,规定如下:每次只能移动相邻的2堆石子合并,合并花费为新合成的一堆石子的数量。求将这N堆石子合并成一堆的总花费最小(或最大)。

分析:我们熟悉矩阵连乘,知道矩阵连乘也是每次合并相邻的两个矩阵,那么石子合并可以用矩阵连乘的方式来解决。

设dp[i][j]表示第i到第j堆石子合并的最优值,sum[i][j]表示第i到第j堆石子的总数量。那么就有状态转移公式:

当 i = j 时,dp[ i ][ j ] = 0;

当i != j 时,dp[ i ][ j ] = min(dp[i][j], dp[ i ][ k ]+dp[ k+1 ][ j ] ) + sum[ i ][ j ].

代码:复杂度为O(n^3).

  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int a[1010];
  4. int dp[1010][1010];
  5. int sum[1010][1010];
  6. int main()
  7. {
  8. ios::sync_with_stdio(false);
  9. int n;
  10. cin>>n;
  11. for(int i=1;i<=n;i++){
  12. cin>>a[i];
  13. }
  14. for(int i=1;i<n;i++)
  15. {
  16. sum[i][i]=a[i];
  17. for(int j=i+1;j<=n;j++)
  18. {
  19. sum[i][j]=sum[i][j-1]+a[j];//合并区间i到j的结果
  20. }
  21. }
  22. for(int r=2;r<=n;r++)//区间长度为 r的最小值
  23. {
  24. for(int i=1;i<=n-(r-1);i++)//枚举区间长度的开头
  25. {
  26. int j=i+(r-1);//区间长度固定,根据开头求出结尾
  27. dp[i][j]=999999999;
  28. for(int k=i;k<j;k++)//枚举出合并这段区间以前的最小花费
  29. {
  30. dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]);
  31. }
  32. dp[i][j] += sum[i][j];//加上这一段合并的花费
  33. }
  34. }
  35. cout<<dp[1][n]<<endl;
  36. return 0;
  37. }

因为上面的复杂度为O(n^3),那么我们可以利用平行四边形优化,优化为O(n^2).

代码:

  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int a[1010];
  4. int dp[1010][1010];
  5. int sum[1010][1010];
  6. int p[1010][1010];
  7. int main()
  8. {
  9. ios::sync_with_stdio(false);
  10. int n;
  11. cin>>n;
  12. for(int i=1;i<=n;i++){
  13. cin>>a[i];
  14. }
  15. for(int i=1;i<n;i++)
  16. {
  17. sum[i][i]=a[i];
  18. for(int j=i+1;j<=n;j++)
  19. {
  20. sum[i][j]=sum[i][j-1]+a[j];//合并区间i到j的结果
  21. }
  22. }
  23. for(int i=1;i<=n;i++){
  24. p[i][i]=i;
  25. }
  26. for(int len=1;len<n;len++)
  27. {
  28. for(int i=1;i+len<=n;i++)
  29. {
  30. int end = i+len;
  31. int tmp=0x3ffffff;
  32. int k=0;
  33. for(int j = p[i][end-1]; j <= p[i+1][end]; j++)
  34. {
  35. if(dp[i][j] + dp[j+1][end] + sum[i][end] < tmp)
  36. {
  37. tmp=dp[i][j] + dp[j+1][end] + sum[i][end];
  38. k=j;
  39. }
  40. }
  41. dp[i][end] = tmp;
  42. p[i][end] = k;
  43. }
  44. }
  45. /*
  46. for(int r=2;r<=n;r++)//区间长度为 r的最小值
  47. {
  48. for(int i=1;i<=n-(r-1);i++)//枚举区间长度的开头
  49. {
  50. int j=i+(r-1);//区间长度固定,根据开头求出结尾
  51. dp[i][j]=999999999;
  52. for(int k=i;k<j;k++)//枚举出合并这段区间以前的最小花费
  53. {
  54. dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]);
  55. }
  56. dp[i][j] += sum[i][j];//加上这一段合并的花费
  57. }
  58. }
  59. */
  60. cout<<dp[1][n]<<endl;
  61. return 0;
  62. }

(3) 问题(2)的是在石子排列是直线情况下的解法,如果把石子改为环形排列,又怎么做呢?

分析:状态转移方程为:

当 j =0 时,dp[i][j]=0.

当 j > 0&& 0<=k < j, dp[i][j]=min(dp[i][j], dp[i][k] + dp[(i+k+1)%n][j-k-1] + sum(i,j).

20140109151533250

代码:

  1. #include<bits/sdc++.h>
  2. using namespace std;
  3. const int INF = 1 << 30;
  4. const int N = 205;
  5. int mins[N][N];
  6. int maxs[N][N];
  7. int sum[N],a[N];
  8. int minval,maxval;
  9. int n;
  10. int getsum(int i,int j)
  11. {
  12. if(i+j >= n) return getsum(i,n-i-1) + getsum(0,(i+j)%n);
  13. else return sum[i+j] - (i>0 ? sum[i-1]:0);
  14. }
  15. void solve(int a[],int n)
  16. {
  17. for(int i=0;i<n;i++) mins[i][0] = maxs[i][0] = 0;
  18. for(int j=1;j<n;j++)
  19. {
  20. for(int i=0;i<n;i++)
  21. {
  22. mins[i][j] = INF;
  23. maxs[i][j] = 0;
  24. for(int k=0;k<j;k++)
  25. {
  26. mins[i][j] = min(mins[i][j],mins[i][k] + mins[(i+k+1)%n][j-k-1] + getsum(i,j));
  27. maxs[i][j] = max(maxs[i][j],maxs[i][k] + maxs[(i+k+1)%n][j-k-1] + getsum(i,j));
  28. }
  29. }
  30. }
  31. minval = mins[0][n-1];
  32. maxval = maxs[0][n-1];
  33. for(int i=0;i<n;i++)
  34. {
  35. minval = min(minval,mins[i][n-1]);
  36. maxval = max(maxval,maxs[i][n-1]);
  37. }
  38. }
  39. int main()
  40. {
  41. while(~scanf("%d",&n))
  42. {
  43. for(int i=0;i<n;i++) scanf("%d",&a[i]);
  44. sum[0] = a[0];
  45. for(int i=1;i<n;i++) sum[i] = sum[i-1] + a[i];
  46. solve(a,n);
  47. printf("%d %d\n",minval,maxval);
  48. }
  49. return 0;
  50. }

发表评论

表情:
评论列表 (有 0 条评论,282人围观)

还没有评论,来说两句吧...

相关阅读

    相关 石子合并问题区间dp

    石子合并问题是最经典的DP问题。首先它有如下3种题型: (1)有N堆石子,现要将石子有序的合并成一堆,规定如下:每次只能移动任意的2堆石子合并,合并花费为新合成的

    相关 282 石子合并区间dp

    1. 问题描述: 设有 N 堆石子排成一排,其编号为 1,2,3,…,N。每堆石子有一定的质量,可以用一个整数来描述,现在要将这 N 堆石子合并成为一堆。每次只能合并相邻的两

    相关 ACM DP 石子合并问题

    滴,集训第二十一天打卡。 可能是对组队不太满意,都不大高兴做新的训练... 所以最近一直在磨DP,翻一下博客,发现最近都是DP啊... 这个石子合并的,我做的训练数据n是