SRM610 Div1 Easy TheMatrix

問題概要

盤面上の市松模様の面積の最大値

解法

横向きに累積和を取って縦の範囲決めてO(N^3)
制約読み違えてて辛かった.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
class TheMatrix{
public:
    int MaxArea(vs b){
        int h=b.size();
        int w=b[0].size();
        vvi p(h,vi(w,1));
        REP(i,h){
            REP(j,w-1){
                if(b[i][j]!=b[i][j+1]){
                    p[i][j+1]=p[i][j]+1;
                }
            }
        }
        int ans=0;
        REP(i,h){
            REP(j,w){
                int mi=p[i][j];
                FOR(k,i,h){
                    chmin(mi,p[k][j]);
                    chmax(ans,mi*(k-i+1));
                    if(k+1<h&&b[k][j]==b[k+1][j]){
                        break;
                    }
                }
            }
        }
        return ans;
    }
};