前言:
21考研,记录一下准备复试路上写下的垃圾代码。本来啃《算法笔记》,但是感觉太多了做不完,改做王道机试指南。
题目描述:
已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵,你的任务是找到最大的非空(大小至少是1 * 1)子矩阵。 比如,如下4 * 4的矩阵 0 -2 -7 0 9 2 -6 2 -4 1 -4 1 -1 8 0 -2 的最大子矩阵是 9 2 -4 1 -1 8 这个子矩阵的大小是15。
输入描述
输入是一个N * N的矩阵。输入的第一行给出N (0 < N <= 100)。
再后面的若干行中,依次(首先从左到右给出第一行的N个整数,再从左到右给出第二行的N个整数……)给出矩阵中的N2个整数,整数之间由空白字符分隔(空格或者空行)。
已知矩阵中整数的范围都在[-127, 127]。
输出描述:
测试数据可能有多组,对于每组测试数据,输出最大子矩阵的大小。
解答
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<map>
#include<string>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 110;
int main()
{
int n;
int m[maxn][maxn];
while (cin >> n) {
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
scanf("%d", &m[i][j]);
int dp[maxn][maxn];
int dprow[maxn];
for(int i=0;i<n;i++)
for (int j = i; j < n; j++) {
if (i == j) {
dprow[0] = m[i][0];
for (int k = 1; k < n; k++)
dprow[k] = max(m[i][k], m[i][k] + dprow[k - 1]);
}
else {
int temp[maxn] = {0};
for (int k = 0; k < n; k++)
for (int l = i; l <= j; l++)
temp[k] += m[l][k];
dprow[0] = temp[0];
for (int g = 1; g < n; g++)
dprow[g] = max(temp[g], temp[g] + dprow[g - 1]);
}
int max = -130;
for (int k = 0; k < n; k++)
if (max < dprow[k])
max = dprow[k];
dp[i][j] = max;
}
int res = -130;
for (int i = 0; i < n; i++)
for (int j = i; j < n; j++)
if (res < dp[i][j])
res = dp[i][j];
cout << res << endl;
}
return 0;
}