动态规划——Split Array Largest Sum

题意大概就是,给定一个包含非负整数的序列nums以及一个整数m,要求把序列nums分成m份,并且要让这m个子序列各自的和的最大值最小(minimize the largest sum among these m subarrays)。
Note:
If n is the length of array, assume the following constraints are satisfied:
1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)
Examples:
Input:
nums = [7,2,5,10,8]
m = 2
Output:
18
Explanation:
There are four ways to split nums into two subarrays.
The best way is to split it into [7,2,5] and [10,8],
where the largest sum among the two subarrays is only 18.
状态:dp[j][i]就是简单的缩小题目的规模,把j长的序列分成i个连续的子序列,子序列和的最大值中的最小值。
状态转移方程:每次我们只关注整个序列中最后一个元素加入时对dp值的影响,由于是要分成连续的序列,所以最后一个元素只能与它前面的若干元素组成子序列,需要一个for来枚举包含最后一个元素的子序列的情况,例如我现在要求dp[j][i],在放入最后一个元素nums[j]时,设klen为第i个连续子序列的长度,这个子序列的和为
dp[nums.size][1]-dp[nums.size-klen][1],而前nums-klen个元素组成的i-1个连续子序列和的最大值的最小值为dp[nums.size][i-1]已经在前面的计算过程中完成了计算,易知dp[j][i] = min(max(dp[nums.size][1]-dp[nums.size-klen][1],dp[nums.size][i-1])),这个题可以很明显的看出动态规划的最优子结构
不过为了方便起见,先将所有的dp[j][1]计算出来,计算很简单也好理解,就是for循环叠加,不过如果单纯用int数组的话可能越界,比如:[1,2147483647],会在dp数组中出现-2147483648这样的元素。由于输入样本都是整数,可以使用double数组,返回的结果用int强制转换即可
 
 1     public int splitArray(int[] nums,int m) {
 2         int nlen = nums.length;
 3         int[]num = new int[nlen+1];
 4         double[][]dp = new double[nlen+1][m+1];
 5         double temp = 0;
 6         num[0] = nums.length;
 7         for(int i = 1;i<=nlen;i++)
 8             num[i] = nums[i-1];
 9         for(int i = 0;i<=nlen;i++)
10             dp[i][0] = 0;
11         for(int i = 0;i<=m;i++)
12             dp[0][i] = 0;
13         for(int i = 1;i<=m;i++) {
14             for(int j = i;j<=nlen;j++) {
15                 if(i==1)dp[j][i] = dp[j-1][i]+num[j];
16                 else {
17                     dp[j][i] = dp[nlen][1];
18                     for(int k = i-1;k<j;k++) {
19                         temp = dp[k][i-1]>(dp[j][1]-dp[k][1])?dp[k][i-1]:(dp[j][1]-dp[k][1]);
20                         dp[j][i] = dp[j][i]<temp?dp[j][i]:temp;
21                     }
22                 }
23             }
24         }
25         
26         for(int i = 1;i<=m;i++) {
27             for(int j = 1;j<=nlen;j++) 
28                 System.out.print(dp[j][i]+" ");
29             System.out.println();
30         }
31         
32         return (int) dp[nlen][m];
33     }

猜你喜欢

转载自www.cnblogs.com/messi2017/p/9900027.html