Submission #430468

#	Time	Username	Problem	Language	Result	Execution time	Memory
430468		egekabas	Rectangles (IOI19_rect)	C++14	72 / 100	4919 ms	957584 KiB

rect

#include "rect.h"
#include <bits/stdc++.h>
#define all(x) (x).begin(), (x).end()
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef pair<ull, ull> pull;
typedef pair<int, int> pii;
typedef pair<ld, ld> pld;
int bit[2509];
int bitsz;
void upd(int idx, int val){
	for(++idx; idx <= bitsz; idx += idx&(-idx))
		bit[idx] += val;
}
int get(int idx){
	int ret = 0;
	for(++idx; idx > 0; idx -= idx&(-idx))
		ret += bit[idx];
	return ret;
}
	
long long count_rectangles(std::vector<std::vector<int> > a) {
	int n = a.size(), m = a[0].size();
	bitsz = max(n, m);

	vector<vector<vector<int>>> pos(m, vector<vector<int>>(m));
	for(int i = 0; i < n; ++i){
		vector<int> cur;
		for(int j = m-1; j >= 0; --j){
			while(cur.size() && a[i][cur.back()] < a[i][j]){
				pos[j][cur.back()].pb(i);
				cur.pop_back();
			}
			if(cur.size())
				pos[j][cur.back()].pb(i);
			while(cur.size() && a[i][cur.back()] <= a[i][j])
				cur.pop_back();
			
			cur.pb(j);
		}
	}
	vector<vector<vector<pii>>> maxval(m, vector<vector<pii>>(n));
	
	for(int x = 0; x < m; ++x)
		for(int y = x+2; y < m; ++y)
			for(int i = 0; i < pos[x][y].size();){
				int j;
				for(j = i+1; j < pos[x][y].size(); ++j)
					if(pos[x][y][j] != pos[x][y][j-1]+1)
						break;
				int l = pos[x][y][i];
				int r = pos[x][y][j-1];
				//cout << x << ' ' << y << ' ' << l << ' ' << r << '\n';
				for(int k = l; k <= r; ++k)
					maxval[x][k].pb({r, y});//////////////////////////
				i = j;
			}
		
	pos = vector<vector<vector<int>>>(n, vector<vector<int>>(n));
	
	for(int i = 0; i < m; ++i){
		vector<int> cur;
		for(int j = n-1; j >= 0; --j){
			while(cur.size() && a[cur.back()][i] < a[j][i]){
				pos[j][cur.back()].pb(i);
				cur.pop_back();
			}
			if(cur.size())
				pos[j][cur.back()].pb(i);
			while(cur.size() && a[cur.back()][i] <= a[j][i])
				cur.pop_back();
			
			cur.pb(j);
		}
	}
	vector<vector<vector<pii>>> fin(m, vector<vector<pii>>(n));
	
	for(int x = 0; x < n; ++x)
		for(int y = x+2; y < n; ++y)
			for(int i = 0; i < pos[x][y].size();){
				int j;
				for(j = i+1; j < pos[x][y].size(); ++j)
					if(pos[x][y][j] != pos[x][y][j-1]+1)
						break;
				int l = pos[x][y][i];
				int r = pos[x][y][j-1];
				for(int k = l; k <= r; ++k)
					fin[k][x].pb({y, r});
	
				i = j;
			}
	ll ans = 0;
	for(int i = 1; i < m; ++i)
		for(int j = 0; j < n-1; ++j){
				sort(all(maxval[i-1][j+1]), greater<pii>());
				sort(all(fin[i][j]), greater<pii>());
				int curidx = 0;
				for(auto u : fin[i][j]){
					while(curidx < maxval[i-1][j+1].size() && maxval[i-1][j+1][curidx].ff >= u.ff-1){
						upd(maxval[i-1][j+1][curidx].ss, +1);
						++curidx;
					}
					ans += get(u.ss+1);
				}
				for(int k = 0; k < curidx; ++k)
					upd(maxval[i-1][j+1][k].ss, -1);
				assert(get(m) == 0);
		}

	return ans;
}

Compilation message (stderr)

rect.cpp: In function 'long long int count_rectangles(std::vector<std::vector<int> >)':
rect.cpp:53:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |    for(int i = 0; i < pos[x][y].size();){
      |                   ~~^~~~~~~~~~~~~~~~~~
rect.cpp:55:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |     for(j = i+1; j < pos[x][y].size(); ++j)
      |                  ~~^~~~~~~~~~~~~~~~~~
rect.cpp:87:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   87 |    for(int i = 0; i < pos[x][y].size();){
      |                   ~~^~~~~~~~~~~~~~~~~~
rect.cpp:89:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   89 |     for(j = i+1; j < pos[x][y].size(); ++j)
      |                  ~~^~~~~~~~~~~~~~~~~~
rect.cpp:106:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  106 |      while(curidx < maxval[i-1][j+1].size() && maxval[i-1][j+1][curidx].ff >= u.ff-1){
      |            ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 7 / 7
• Subtask 3: 12 / 12
• Subtask 4: 22 / 22
• Subtask 5: 10 / 10
• Subtask 6: 13 / 13
• Subtask 7: 0 / 28