一维情况
首先设置一个数组dp[],令dp[i]表示以A[i]作为末尾的连续序列的最大和。于是,通过设置这么一个数组,最大连续子序列和便是数组dp中的最大值。
由于dp[i]是以A[i]作为末尾的连续序列的最大和,因此只有两种情况:
- 最大和的连续序列只有一个元素,即A[i]本身,也就是说dp[i]=A[i]。
- 最大和的连续序列有多个元素,即从前面某个A[j]开始,一直到A[i]结束,也就是dp[i] = A[j]+...+A[i-1]+A[i]。如何获得 A[j]+...+A[i-1]呢?不难发现dp[i-1] = A[j]+...+A[i-1]。
由于只有这两种情况,于是得到状态转移方程dp[i] = max(A[i], dp[i-1]+A[i])。只需要将i从小到大枚举并依次遍历,即可得到整个dp数组。接着输出该数组中的最大值,即求得最大连续子序列的和。
代码
#include <iostream>
using namespace std;
const int MAXN = 10001;
int main(){
int n;
while(cin>>n && n){
int arr[MAXN],dp[3][MAXN],maxinum=-1,maxiIndexLeft=-1,maxiIndexRight=-1;
for(int i=0;i<n;i++){
cin>>arr[i];
if(i==0){
dp[0][i] = arr[i];
dp[1][i] = 0; // 首
dp[2][i] = 0; // 尾
}
else {
if(arr[i]>dp[0][i-1]+arr[i]){
dp[0][i] = arr[i];
dp[1][i]=dp[2][i]=i;
}else{
dp[0][i] = dp[0][i-1]+arr[i];
dp[1][i] = dp[1][i-1];
dp[2][i] = i;
}
}
if(maxinum<dp[0][i]){
maxinum = dp[0][i];
maxiIndexLeft = dp[1][i];
maxiIndexRight = dp[2][i];
}
}
if(maxiIndexLeft==-1){
cout<<0<<" "<<arr[0]<<" "<<arr[n-1]<<endl;
}else{
cout<<maxinum<<" "<<arr[maxiIndexLeft]<<" "<<arr[maxiIndexRight]<<endl;
}
}
return 0;
}
二维情况——求一个二维矩阵的最大子矩阵和
假设原二维矩阵中最大子矩阵所在的行是从i到j,那么只会出现下面这两种情况:
- 当i==j时,求最大子矩阵和就转换成了求第i行元素的最大连续子序列和。
- 当i!=j时,把从第i行到第j行的所有行的元素加起来,得到只有一行的一维数组,这个一维数组的最大连续子序列和便是最大子矩阵和。
只需从小到大枚举并依次遍历i和j,接着输出所有子矩阵中的最大值,即求得最大子矩阵和。
当i!=j时,不必将原始矩阵中的第i行到第j行的所有行的元素累加起来得到一维数组,而事先用一个辅助二维矩阵记录原始矩阵从上到下加起来的累加矩阵,于是求从第i行到第j行的一维数组只需将辅助矩阵进行一次减法便可得到,而不需要逐行进行多次累加。
代码
#include <iostream>
using namespace std;
const int MAXN=101;
int total[MAXN][MAXN]; // 辅助矩阵
int arr[MAXN]; // 一维数组
int dp[MAXN];
int MaxSubsequence(int n){
int maximum = 0;
for(int i=0;i<n;i++){
if(i==0) dp[i] = arr[i];
else dp[i] = max(arr[i],dp[i-1]+arr[i]);
maximum = max(maximum,dp[i]);
}
return maximum;
}
int MaxSubmatrix(int n){
int maximal = 0;
for(int i=0;i<n;i++){
for(int j=i;j<n;j++){
for(int k=0;k<n;k++){ // 获得一维数组
if(i==0) arr[k] = total[j][k];
else arr[k] = total[j][k]-total[i-1][k];
}
maximal = max(maximal, MaxSubsequence(n));
}
}
return maximal;
}
int main(){
int n,elem;
while(cin>>n){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cin>>elem;
if(i==0) total[i][j]=elem;
else total[i][j]=total[i-1][j]+elem;
}
}
cout<<MaxSubmatrix(n)<<endl;
}
}