您的位置:首页 > 其它

一维和二维子数组之和最大值

2013-12-02 15:10 169 查看
一、约定

所谓子数组,是连续的。
只求和,不返回子数组的具体位置。
元素是整数,所以数组可能包含正整数,0,负数。
二、一维数组子数组之和的最大值

最直接的求法——暴力求解
           记sum[ i,...j ]为数组A中第i个元素到第j个元素的和(其中0<=i<=j<n)遍历所有可能的sum[ i,...j ]。

/**
* 常规解法求一位最大字段和
* @author DaiSong
* @Date 2013年12月2日
*/
public class OneDimensionalWithNormalSolution {

/**方法一,复杂度O(N^3)
* @param a
* @param n
* @return
*/
public static int MaxSum1(int[] a,int n){

int maximum=Integer.MIN_VALUE;
int sum;
for(int i=0;i<n;i++){
for(int j=i;j<n;j++){
sum=0;
for(int k=i;k<=j;k++){
sum+=a[k];
}
if(sum>maximum){
maximum=sum;
}
}
}
return maximum;
}

/**
* 方法二,改进:将算法的最后一个for循环省略,避免重复计算,复杂度O(N^2).
* @param a
* @param n
* @return
*/
public static int MaxSum2(int[] a,int n){
int maximum=Integer.MIN_VALUE;
int sum;
for(int i=0;i<n;i++){
sum=0;
for(int j=i;j<n;j++){
sum+=a[j];
if(sum>maximum){
maximum=sum;
}
}
}
return maximum;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
int a[]={0,-2,3,5,-1,2};
int b[]={1,-2,3,5,-3,2};
int c[]={-9,-2,-3,-5,-3};
//MaxSum1 Test Result
System.out.println(MaxSum1(a, 6)+" "+MaxSum1(b, 6)+" "+MaxSum1(c, 5));
//MaxSum2 Test Result
System.out.println(MaxSum2(a, 6)+" "+MaxSum2(b, 6)+" "+MaxSum2(c, 5));
}
}


       2.递归法

          将数组(A[0],...A[n-1])分为长度相等的两端数组(A[0],...A[n/2-1])和(A[n/2],...,A[N-1]),分别求出这两端数组各自的最大字段和,则数组(A[0],...A[n-1])的最大字段和为以下三种情况的最大值:

(A[0],...A[n-1])的最大字段和与(A[0],...A[n/2-1])的最大字段和相同。
(A[0],...A[n-1])的最大字段和与(A[n/2],...,A[N-1])的最大字段和相同。
(A[0],...A[n-1])的最大字段和跨过其中间两个元素A[n/2-1]到A[n/2]。

/**
* 分治策略求一维最大字段和,时间复杂度为O(N*log2N),(以2为底)
* @author DaiSong
* @Date 2013年12月2日
*/
public class OneDimensionalWithDivideAndConquer {

public static int FindMaxSubArray(int[] a ,int low,int high){
int leftSum=0,rightSum=0,crossSum=0;
if(low==high){
return a[low];
}
int mid=(low+high)/2;
leftSum=FindMaxSubArray(a,low,mid);
rightSum=FindMaxSubArray(a,mid+1,high);
crossSum=FindMaxCrossSubArray(a,low,mid,high);
return Math.max(Math.max(leftSum, rightSum),crossSum);
}

/**
* 找到跨越终点的子数组的最大值
* @param a
* @param low
* @param mid
* @param high
* @return
*/
public static int FindMaxCrossSubArray(int[] a,int low ,int mid,int high){
int leftSum=Integer.MIN_VALUE;
int rightSum=Integer.MIN_VALUE;
int sum=0;
for(int i=mid;i>=low;i--){
sum+=a[i];
if(sum>leftSum){
leftSum=sum;
}
}
sum=0;
for(int i=mid+1;i<=high;i++){
sum+=a[i];
if(sum>rightSum){
rightSum=sum;
}
}
return leftSum+rightSum;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
int a[]={0,-2,3,5,-1,2};
int b[]={1,-2,3,5,-3,2};
int c[]={-9,-2,-3,-5,-3};
System.out.println(FindMaxSubArray(a,0,5));
System.out.println(FindMaxSubArray(b,0,5));
System.out.println(FindMaxSubArray(c,0,4));
}

}


    3.DP

       考虑数组第一个元素A[0],以及最大的一段数组(A[ i ],...,A[ j ])之间的关系。有以下几种情况:

当0=i=j是,元素本身构成和的最大的一段。
当0=i<j,和最大的一段以A[0]开始。
当0<i时,元素A[0]跟和最大的一段没有关系。

/**
* DP求解一维最大字段和问题。
* @author DaiSong
* @Date 2013年12月2日
*/
public class OneDimensionalWithDP {

/**
* 方法一,逆序。时间复杂度O(N),空间复杂度O(N).
* @param a
* @param n
* @return
*/
public static int MaxSumDp1(int[] a,int n){
int[] start=new int
;
int[] all=new int
;
start[n-1]=all[n-1]=a[n-1];
for(int i=n-2;i>=0;i--){
start[i]=Math.max(a[i], a[i]+start[i+1]);
all[i]=Math.max(start[i],all[i+1]);
}
return all[0];
}

/**
* 方法二,验证正序和逆序没有差别
* @param a
* @param n
* @return
*/
public static int MaxSumDp2(int[] a,int n){
int[] start=new int
;
int[] all=new int
;
start[0]=all[0]=a[0];
for(int i=1;i<n;i++){
start[i]=Math.max(a[i], a[i]+start[i-1]);
all[i]=Math.max(start[i],all[i-1]);
}
return all[n-1];
}

/**
* 方法三,空间复杂度进一步改进为O(N).
* @param a
* @param n
* @return
*/
public static int MaxSumDp3(int[] a,int n){
int start,all;
start=all=a[0];
for(int i=1;i<n;i++){
start=Math.max(a[i], a[i]+start);
all=Math.max(start,all);
}
return all;
}

/**
* 方法四,方法三的另一种写法。
* @param a
* @param n
* @return
*/
public static int MaxSumDp4(int[] a,int n){
int start,all;
start=all=a[0];
for(int i=1;i<n;i++){
if(start<0){
start=0;
}
start+=a[i];
if(start>all){
all=start;
}
}
return all;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
int a[]={0,-2,3,5,-1,2};
int b[]={1,-2,3,5,-3,2};
int c[]={-9,-2,-3,-5,-3};
System.out.println(MaxSumDp1(a,6)+" "+MaxSumDp1(b,6)+" "+MaxSumDp1(c,5));
System.out.println(MaxSumDp2(a,6)+" "+MaxSumDp2(b,6)+" "+MaxSumDp2(c,5));
System.out.println(MaxSumDp3(a,6)+" "+MaxSumDp3(b,6)+" "+MaxSumDp3(c,5));
System.out.println(MaxSumDp4(a,6)+" "+MaxSumDp4(b,6)+" "+MaxSumDp4(c,5));
}
}


三、二维数组的最大子数组和的最大值

      1.暴力求解

         枚举矩阵的四个点,再就矩阵内的数的和。矩阵求和由于存在重复计算,可以预处理用数组存起来。

/**
* 暴力解法及其优化求解二维最大子段和。
* @author DaiSong
* @Date 2013年12月2日
*/
public class TwoDimensionalWithNormalSolution {
static int MAX = 501;
static int[][] ps = new int[MAX][MAX];

/**
* 方法一,暴力求解,时间复杂度为O(N^3*M^3)
*
* @param a
* @param m
* @param n
* @return
*/
public static int MaxSum1(int a[][], int m, int n) {

int max = Integer.MIN_VALUE;
int sum;
for (int min_i = 0; min_i < n; min_i++) {
for (int max_i = min_i; max_i < n; max_i++) {
for (int min_j = 0; min_j < m; min_j++) {
for (int max_j = min_j; max_j < m; max_j++) {
// 求区域矩阵的和
sum = 0;
for (int i = min_i; i <= max_i; i++) {
for (int j = min_j; j <= max_j; j++) {
sum += a[i][j];
}
}
if (sum > max) {
max = sum;
}
}
}
}
}
return max;
}

/**
* 方法二,改进:考虑到区域的和需要频繁计算,做了预处理。用ps
[m]存放i=1..n,j=1..m区域和。
* 时间复杂度为O(N^2*M^2)
* @param a
* @param n
* @param m
* @return
*/
public static int MaxSum2(int a[][], int n, int m) {

int max = Integer.MIN_VALUE;
int sum;
PieceSum(a, n, m);
for (int min_i = 1; min_i <= n; min_i++) {
for (int max_i = min_i; max_i <= n; max_i++) {
for (int min_j = 1; min_j <= m; min_j++) {
for (int max_j = min_j; max_j <= m; max_j++) {
// 求区域矩阵的和
sum = ps[max_i][max_j]-ps[min_i-1][max_j]-ps[max_i][min_j-1]+ps[min_i-1][min_j-1];
if (sum > max) {
max = sum;
}
}
}
}
}
return max;
}

/**
* 预处理求出区域和,时间复杂度为O(N*M)
* @param a
* @param n
* @param m
*/
public static void PieceSum(int[][] a, int n, int m) {
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
ps[i][j] = ps[i - 1][j] + ps[i][j - 1] - ps[i - 1][j - 1]
+ a[i][j];
}
}
}

public static void main(String[] args) {
// TODO Auto-generated method stub
int a[][] = { { -1, -4, 3 }, { 3, 4, -1 }, { -5, -2, 8 } };
System.out.println(MaxSum1(a, 3, 3));
int b[][] = { {0,0,0,0},{ 0, -1, -4, 3 }, { 0, 3, 4, -1 }, { 0, -5, -2, 8 } };
System.out.println(MaxSum2(b, 3, 3));
}
}


         2.DP

          把二维数组压缩为一维数组,再求和。

/**
* 最大子阵,压缩矩阵为一维数组,转化为求最大字段和问题,运用动态规划求解。
* @author DaiSong
* @Date 2013年12月2日
*/
public class TwoDimensionalWithDP {

/**转化为一位数组,求一维数组的最大字段和。时间复杂度O(N*M*Min(N,M))
* @param a
* @param n
* @param m
* @return
*/
public static int TwoMaxSum(int a[][],int n,int m){
int minMax;
int  Max = Integer.MIN_VALUE;
for (int i=0; i<n; i++){
minMax = OneMaxSum(a[i], m);
if (minMax > Max)
Max = minMax;
for (int j=i+1; j<n; j++){
for (int k=0; k<n; k++){
a[i][k] += a[j][k];
}
minMax = OneMaxSum(a[i], n);
if (minMax > Max)
Max = minMax;
}
}
return Max;
}

/**一维最大字段和
* @param a
* @param n
* @return
*/
public static int OneMaxSum(int[] a,int n){
int start,all;
start=all=a[0];
for(int i=1;i<n;i++){
start=Math.max(a[i], a[i]+start);
all=Math.max(start,all);
}
return all;
}

public static void main(String[] args) {
int[][] a ={{-1,-4,3},{3,4,-1},{-5,-2,8}};
System.out.print(TwoMaxSum(a,3,3));
}
}


参考资料:《编程之美》,《算法导论》。

www.gavinboo.com同步发布。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: