C++ 基礎題 Q-6-5 二維最大子矩陣

題目連結:d074. Q_6_5 二維最大子矩陣 - JMJudge

擱置了半年,如今終於解開。與 Q-2-12有異曲同工之妙,但複雜度變O(n2m)。

#include<bits/stdc++.h>
using namespace std;
#define r first
#define c second
#define INF 100000000000
typedef long long ll;

ll a[205][205],p[205][205];

int main(){
    ios::sync_with_stdio(0);cin.tie(0);

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

    ll ans=0;
    for(int i=0;i<=n;i++){
        for(int j=i+1;j<=n;j++){
            ll psum=0;
            for(int k=0;k<m;k++){
                psum+=p[j][k]-p[i][k];
                if(psum<0) psum=0;
                ans=max(psum,ans);
            }
        }
    }
    cout<<ans;
    return 0;
}