问题描述:
给定n个整数组成的序列,现在要求将序列分割为m段,每段子序列中的数在原序列中连续排列。如何分割才能使这m段子序列的和的最大值达到最小?
输入格式:
第一行给出n,m,表示有n个数分成m段,随后一行给出n个数,以空格分隔
输入样例:
9 3
9 8 7 6 5 4 3 2 1
输出样例:
17
解释:
9 8 | 7 6 | 5 4 3 2 1,9个数分成3段所有情况里这种分法的最大子段和(17)最小。
思路:
动态规划的一种模板,包括矩阵连乘,石子归并等问题。先是考虑分的段数,然后考虑起点/终点,最后是每段中分割点的位置。
本题中使用 dp[i][j]表示前i个数分成j段里最小的最大子段和,则dp[i][j]=min{dp[i][1]-dp[k][1],dp[k][j-1] (1<=k<i)};
代码:
1 #include<bits/stdc++.h> 2 using namespace std; 3 #define N 99 4 #define inf 0x3f3f3f3f 5 int a[N],dp[N][N];//dp[i][j]表示前i个数分j段所有情况里最大子段和最小的 6 int main() 7 { 8 int n,m; 9 cin>>n>>m; 10 for(int i=1; i<=n; i++) 11 cin>>a[i]; 12 for(int i=1; i<=n; i++) 13 dp[i][1]=dp[i-1][1]+a[i]; 14 for(int r=2; r<=m; r++){ 15 for(int i=1; i<=n; i++){ 16 int mins=inf; 17 for(int k=1; k<i; k++){ 18 int t=max(dp[i][1]-dp[k][1],dp[k][r-1]); 19 if(t<mins) mins=t; 20 } 21 dp[i][r]=mins; 22 } 23 } 24 cout<<dp[n][m]; 25 return 0; 26 }