Submission #148228

---

Source Code Expand

#include <iostream>
#include <iomanip>
#include <sstream>
#include <vector>
#include <string>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <algorithm>
#include <functional>
#include <iterator>
#include <limits>
#include <numeric>
#include <utility>
#include <cmath>

using namespace std;

using LL = long long;
using ULL = unsigned long long;
using VI = vector<int>;
using VVI = vector<VI>;
using VS = vector<string>;
using SS = stringstream;
using PII = pair<int,int>;
using VPII = vector< pair<int,int> >;
template < typename T = int > using VT = vector<T>;
template < typename T = int > using VVT = VT< VT<T> >;
template < typename T = int > using LIM = numeric_limits<T>;

template < typename T > inline T fromString( const string &s ){ T res; istringstream iss( s ); iss >> res; return res; };
template < typename T > inline string toString( const T &a ){ ostringstream oss; oss << a; return oss.str(); };

#define REP( i, m, n ) for ( int i = (int)( m ); i < (int)( n ); ++i )
#define FOR( e, c ) for ( auto &e : c )
#define ALL( c ) (c).begin(), (c).end()
#define AALL( a, t ) (t*)a, (t*)a + sizeof( a )
#define DRANGE( c, p ) (c).begin(), (c).begin() + p, (c).end()

#define PB( n ) push_back( n )
#define MP( a, b ) make_pair( ( a ), ( b ) )
#define EXIST( c, e ) ( (c).find( e ) != (c).end() )

#define fst first
#define snd second

#define DUMP( x ) cerr << #x << " = " << ( x ) << endl

// 累積和による区間 Sum クエリ
// 前処理 O( HW )
// クエリ O( 1 )
class ConsecutiveSums2d
{
private:
	const int H, W;
	vector< vector<int> > src, csums;
	bool isSolved;

public:
	ConsecutiveSums2d( const int h, const int w ) : H( h ), W( w ), src( H, vector<int>( W ) ), isSolved( false ) {};
	ConsecutiveSums2d( const vector< vector<int> > &src ) : H( src.size() ), W( src[0].size() ), src( src ), isSolved( false ) {};

	int get( const int y, const int x ) const
	{
		return src[y][x];
	}

	void set( const int y, const int x, const int v )
	{
		src[y][x] = v;
		isSolved = false;
		
		return;
	}

	int sum( const int y1, const int y2, const int x1, const int x2 )
	{
		if ( !isSolved )
		{
			solve();
		}
		return csums[ y2 + 1 ][ x2 + 1 ] - csums[ y2 + 1 ][ x1 ] - csums[ y1 ][ x2 + 1 ] + csums[ y1 ][ x1 ];
	}

private:
	void solve()
	{
		csums.clear();
		csums.resize( H + 1, vector<int>( W + 1 ) );

		for ( int i = 0; i < H; ++i )
		{
			partial_sum( src[i].begin(), src[i].end(), csums[ i + 1 ].begin() + 1 );
		}

		for ( int i = 0; i < H; ++i )
		{
			for ( int j = 0; j <= W; ++j )
			{
				csums[ i + 1 ][j] += csums[i][j];
			}
		}

		isSolved = true;
		return;
	}
};
// consecutivesums2d( H, W )
// consecutivesums2d( SRC )
// get( y, x )
// set( y, x, value )
// sum( y1, y2, x1, x2 )

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

	int n;
	cin >> n;

	VVI board( n, VI( n ) );
	FOR( line, board )
	{
		FOR( d, line )
		{
			cin >> d;
		}
	}

	ConsecutiveSums2d csums( board );

	VI maxes( n * n + 1 );
	REP( y1, 0, n )
	{
		REP( y2, y1, n )
		{
			REP( x1, 0, n )
			{
				REP( x2, 0, n )
				{
					const int s = ( y2 - y1 + 1 ) * ( x2 - x1 + 1 );
					maxes[s] = max( maxes[s], csums.sum( y1, y2, x1, x2 ) );
				}
			}
		}
	}
	partial_sum( ALL( maxes ), maxes.begin(), []( const int a, const int b ){ return max( a, b ); } );

	int q;
	cin >> q;
	REP( i, 0, q )
	{
		int p;
		cin >> p;
		cout << maxes[p] << endl;
	}

	return 0;
}

Submission Info

Judge Result