NOI / 2.6基本算法之动态规划 1768:最大子矩阵 双指针 dp

描述

已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵，你的任务是找到最大的非空(大小至少是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]。

输出

输出最大子矩阵的大小。

样例输入

4
0 -2 -7 0 9 2 -6 2
-4 1 -4  1 -1

8  0 -2

样例输出

15

双指针：

#include<bits/stdc++.h>
using namespace std;
#define N 105
#define INF 0x3f3f3f3f
int n, a[N][N], b[N], s[N][N];//b[j]:子矩阵第j列的加和 s[i][j]:第j列的第1行到第i行的元素加和 
int mx = -INF;
int main()
{
    cin >> n;
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= n; ++j)
        { 
            cin >> a[i][j];
            s[i][j] = s[i-1][j] + a[i][j];
        }
    for(int i1 = 1; i1 <= n; ++i1)//子矩阵第一行为原矩阵的第i1行，最后一行为原矩阵的第i2行 
        for(int i2 = i1; i2 <= n; ++i2)
        {
            for(int j = 1; j <= n; ++j)
                b[j] = s[i2][j] - s[i1-1][j];
            int sum = 0;
            for(int j = 1; j <= n; ++j)//双指针法求最大子段和
            {
                if(sum < 0)
                    sum = b[j];
                else
                    sum += b[j];
                mx = max(mx, sum);
            }
        }
    cout << mx;
    return 0;
}

dp:

#include<stdio.h>
#include<string.h>
#include<stdlib.h>

#define maxn 105
#define inf 0x3f3f3f3f

int n, ans;
int pre[maxn][maxn], a[maxn][maxn], f[maxn];

int _max(int a, int b){
    return a > b ? a : b;
}
int main()
{
    scanf("%d", &n);
    ans = -inf;
    int i, j, k;
    for(i = 1; i <= n; i++){
        for(j = 1; j <= n; j++){
            scanf("%d", &a[i][j]);
            pre[i][j] = pre[i - 1][j] + a[i][j];
        }
    }
    for(i = 1; i <= n; i++){
        for(j = i + 1; j <= n; j++){
            for(k = 1; k <= n; k++){
                f[k] = pre[j][k] - pre[i - 1][k];
            }
            for(k = 1; k <= n; k++){
                f[k] = _max(f[k - 1] + f[k], f[k]);
                ans = _max(ans, f[k]);
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}

#include<bits/stdc++.h>
using namespace std;
#define N 105
#define INF 0x3f3f3f3f
int n, a[N][N], b[N], s[N][N];//b[j]:子矩阵第j列的加和 s[i][j]:第j列的第1行到第i行的元素加和
int mx = -INF;
int main()
{
    cin >> n;
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= n; ++j)
        {
            cin >> a[i][j];
            s[i][j] = s[i-1][j] + a[i][j];
        }
    for(int i1 = 1; i1 <= n; ++i1)//子矩阵第一行为原矩阵的第i1行，最后一行为原矩阵的第i2行
        for(int i2 = i1; i2 <= n; ++i2)
        {
            for(int j = 1; j <= n; ++j)
                b[j] = s[i2][j] - s[i1-1][j];
            int dp[N] = {};//线性动规求最大子段和
            for(int j = 1; j <= n; ++j)
            {
                //p[j] 表示以第 j 列为结束点的最大子段和。
                dp[j] = max(b[j], dp[j-1]+b[j]);
                //对于每个 j（从 1 到 n）
                // 我们计算以该列结尾的最大子段和，并更新 mx（全局最大子段和）。
                mx = max(mx, dp[j]);
            }
        }
    cout << mx;
    return 0;
}