11080 游泳圈的最大子矩阵和

题目描述

11080 游泳圈的最大子矩阵和
时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: G++;GCC;VC;JAVA
Description
二维数组首尾相连，上下也相连，像个游泳圈或轮胎，又如何求最大子矩阵和？

如游泳圈展开成3行3列的二维矩阵：
-18 10 7
1 -20 2
1 38 -2
那么最大的子矩阵和为：10+7+38-2=53

若：
2 10 7
1 -20 2
1 38 -2
那么最大的子矩阵和为：10+7+2+38-2+1=56

输入格式
游泳圈展开成平面数组，第一行是行数m和列数n，第二行至第m+1行是数组数值，每行n个数。
1<=m,n<=50

输出格式
最大的子矩阵和

输入样例
3 3
2 10 7
1 -20 2
1 38 -2

输出样例
56

提示

1）先编写一维环形（一圈的形状）长度为len的数据中的最大子段和，且子段长不超过n。
————注意要先调试通过这个程序段，要小心编写，用书上动态规划那段程序修改的话
比较麻烦且易出错，用简单算法改倒是很容易做到的。

2）将游泳圈水平方向上扩展n-1列，垂直方向上扩展m-1行，形成2m-1行2n-1列的扩展矩阵，
在扩展矩阵上求解最大不超过m*n的最大矩阵和。

思路

建议参考这篇博客
本题不一样的地方就是要拓展矩阵和设置边界而已。

代码

#include <iostream>
using namespace std;

const int N = 110;

int matrix[N][N];
int tmp[N];

int maxSubArray(int* a, int n)//一维数组的最大连续子序列
{
	int themax = a[0], curmax = a[0];
	for (int i = 1; i < n; i++)
	{
		curmax = max(a[i], curmax + a[i]);
		themax = max(themax, curmax);
	}
	return themax;
}

int main()
{

    int n, m;
    cin >> n >> m;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++)
        {
            cin >> matrix[i][j];
            if (i != n - 1)
            {
                matrix[n + i][j] = matrix[i][j];
                if (j != n - 1)
                    matrix[n + i][m + j] = matrix[i][j];
            }

            if (j != n - 1)
                matrix[i][m + j] = matrix[i][j];
        }
    int tmpsum, cursum, sum = -1000000000;
    for (int i = 0; i < 2 * n - 1; i++)
    {
        fill(tmp, tmp + 2 * m, 0);
        for (int j = i; j < 2 * n - 1 && j - i < n; j++)
        {

            for (int k = 0; k < 2 * m - 1; k++)
            {
                // tmp[k]为从matrix[i][k]到matrix[j][k]的和
                // 求其最大连续子序列和即可知在i行到j行固定的情况下，列的最大连续序列
                tmp[k] += matrix[j][k];
            }
            for (int a = 0; a < 2 * m - 1; a++)
            {
                tmpsum = maxSubArray(tmp+a,m);
                sum = max(sum, tmpsum);
            }
        }
    }
    cout << sum << endl;
}

// 测试数据用例 

// 标准输入数据:
// 9 5
// 2	-2	7	2	-2
// 10	-2	2	10	-2
// 1	38	-12	1	38
// 10	-20	2	10	-20
// 1	38	-12	1	38
// 2	-2	7	2	-2
// 10	-2	2	10	-2
// 1	38	-12	1	38
// 10	-20	2	10	-20

// 标准输出答案:
//    1|246