DP之石子堆合并问题

(1)相邻:在一个圆形操场的四周摆放着n堆石子(n<= 100),现要将石子有次序地合并成一堆。规定每次只能选取相邻的两堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。选择一种合并石子的方案,使得做n-1次合并,得分的总和最小。

 1 /*
 2      * 在一个圆形操场的四周摆放着n堆石子(n<= 100),现要将石子有次序地合并成一堆。
 3      * 规定每次只能选取相邻的两堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。
 4      *选择一种合并石子的方案,使得做n-1次合并,得分的总和最小。
 5      */
 6     public static void test(int[]a) {
 7         //b[i][j]表示i-j相加之和,求出每一斜列的最小值就是这一步相加
 8         int[][]b=new int[a.length][a.length];
 9         //赋初值
10         for(int i=0;i<a.length;i++) {
11             b[i][i]=a[i];
12             System.out.print(a[i]+" ");
13         }
14         int sum=0;
15         System.out.println("score:"+sum);
16         for(int r=1;r<=a.length-1;r++) {
17             int min=b[1][r]+b[0][0];
18             int index1=0,index2=r;
19             b[0][r]=min;
20             for(int i=1;i<a.length-r;i++) {
21                 int j=i+r;
22                 b[i][j]=b[i+1][j]+b[i][i];
23                 if(b[i][j]<min) {
24                     min=b[i][j];
25                     index1=i;
26                     index2=j;
27                 }
28             }
29             
30             for(int i=0;i<a.length;i++) {
31                 if(i<index1||i>index2) {
32                     System.out.print(a[i]+" ");
33                 }else if(i==index1){//出现最小值
34                     System.out.print(min+" ");
35                 }
36             }
37             sum+=min;
38             System.out.println("score:"+sum);
39         }
40     }

 思路就是类似矩阵连乘,关键就是最优解不好表示或者我写的有问题。。。(明天再来改!!!今天累了)Ok得出结论了就是我的想法有问题。

下面是标准DP:

设dp[i][j]表示第i到第j堆石子合并的最优值(合并i-j的花费的最小力气),sum[i][j]表示第i到第j堆石子的总数量。那么就有状态转移公式: 
这里写图片描述

因为每一个石子堆无非是和左边合并或者先和右边合并。对第i到第j的每一个k(子问题)计算他们的最小值加上每堆石子的总数量因为最后一步总是把所有的都合并。

我们可以从最小问题想起来:假如这个数组只有三个数,那么最小力气就是min(左边两个,右边两个)+三个的总量(最后一步)

有四个数时,将它们分为1-3,2-4两个子问题再继续分...

 1 public static void testA(int[]a) {
 2         class Point{
 3             int start;
 4             int end;
 5             public Point(int start, int end) {
 6                 this.start = start;
 7                 this.end = end;
 8             }
 9         }
10         //b[i][j]表示第i堆到第j堆的最优合并值
11         int[][]b=new int[a.length][a.length];
12         //s表示i到j的和
13         int[][]s=new int[a.length][a.length];
14         //保存最优解 0左 1下
15         int[][]l=new int[a.length][a.length];
16         Stack<Point> stack=new Stack<Point>();
17         //赋初值
18         for(int i=0;i<a.length;i++) {
19             b[i][i]=0;
20             s[i][i]=a[i];
21         }
22         
23         for(int r=1;r<=a.length-1;r++) {
24             for(int i=0;i<a.length-r;i++) {
25                 int j=i+r;
26                 s[i][j]=s[i][j-1]+s[j][j];
27                 if(b[i][j-1]<b[i+1][j]) {//左边小
28                     b[i][j]=b[i][j-1]+s[i][j];
29                     l[i][j]=0;
30                 }else {//下边小
31                     b[i][j]=b[i+1][j]+s[i][j];
32                     l[i][j]=1;
33                 }
34             }
35         }
36         int x=0,y=a.length-1;
37         for(int i=0;i<a.length;i++) {
38             stack.add(new Point(x,y));
39             if(l[x][y]==0)//左边
40                 y--;
41             else
42                 x++;
43         }
44         
45         while(!stack.isEmpty()) {
46             Point p=stack.pop();
47             for(int i=0;i<a.length;i++) {
48                 if(i<p.start||i>p.end)
49                     System.out.print(a[i]+" ");
50                 else if(i==p.start)
51                     System.out.print(s[p.start][p.end]+" ");
52             }
53             System.out.println("score:"+b[p.start][p.end]);
54         }
55     }

 (2)环形:就是不是相邻的是成环的

最简单的就是穷举相邻的找出最小的。这个时候把上面函数的返回值设为int然后计算。

第二种方法是将两边各扩展一位。因为只和左右相关边界位置应与头尾相连,然后进行相同的计算,最后在几种方案中挑出最小的。

 1 public static void testB(int[]a) {
 2         class Point{
 3             int start;
 4             int end;
 5             public Point(int start, int end) {
 6                 this.start = start;
 7                 this.end = end;
 8             }
 9         }
10         //首尾相连
11         int[]b=new int[a.length+2];
12         b[0]=a[a.length-1];
13         b[b.length-1]=a[0];
14         for(int i=0;i<a.length;i++) {
15             b[i+1]=a[i];
16         }
17         
18         //i-j的最优
19         int[][]d=new int[b.length][b.length];
20         //sum
21         int[][]s=new int[b.length][b.length];
22         //最优解0左 1下
23         int[][]l=new int[b.length][b.length];
24         Stack<Point> stack=new Stack<Point>();
25         //赋初值
26         for(int i=0;i<b.length;i++) {
27             d[i][i]=0;
28             s[i][i]=b[i];
29         }
30         //可以不填满,只填到需要的就行
31         int min=-1;
32         Point end=new Point(-1,-1);
33         for(int r=1;r<=a.length-1;r++) {
34             for(int i=0;i<b.length-r;i++) {
35                 int j=i+r;
36                 s[i][j]=s[i][j-1]+s[j][j];
37                 if(d[i][j-1]<d[i+1][j]) {//左边
38                     d[i][j]=d[i][j-1]+s[i][j];
39                     l[i][j]=0;
40                 }else {//下边
41                     d[i][j]=d[i+1][j]+s[i][j];
42                     l[i][j]=1;
43                 }
44                 if(r==a.length-1) {//找到最小的那个的值并记录解
45                     if(min==-1)
46                         {
47                         min=d[i][j];
48                         end.start=i;
49                         end.end=j;
50                         }
51                     else if(min>d[i][j])
52                         {
53                         min=d[i][j];
54                         end.start=i;
55                         end.end=j;
56                         }
57                 }
58             }
59         }
60         int x=end.start,y=end.end;
61         for(int i=0;i<a.length;i++) {
62             stack.add(new Point(x,y));
63             if(l[x][y]==0)//左边
64                 y--;
65             else
66                 x++;
67         }
68         while(!stack.isEmpty()) {
69             Point p=stack.pop();
70             for(int i=0;i<a.length;i++) {
71                 if(i<p.start||i>p.end)
72                     System.out.print(b[i]+" ");
73                 else if(i==p.start)
74                     System.out.print(s[p.start][p.end]+" ");
75             }
76             System.out.println("score:"+d[p.start][p.end]);
77         }
78     }

 (3)总结

这个是典型的DP问题,因为已经处理过之前的矩阵连乘问题,所以一开始我陷入了固定思维只想着填表,而没有去找子问题一步一步来。本末倒置。导致思路很不清晰但是代码似乎更简短。在处理环形时,选取了在两边扩展一位再每次分组计算最后取出一部分。

posted @ 2020-04-20 21:21  小帆敲代码  阅读(319)  评论(0编辑  收藏  举报