제출 #1206642

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1206642		countless	Rectangles (IOI19_rect)	C++20	59 / 100	520 ms	1114112 KiB

rect

#include "rect.h"
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

const ll MOD = 998244353;
// const ll INF = 1e18;
const int INF = 5e3;
const ld EPS = 1e-12;

#define endl "\n"
#define sp <<" "<<
#define REP(i, a, b) for(ll i = a; i < b; i++)
#define dbg(x) cout << #x << " = " << x << endl
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define fast_io() ios_base::sync_with_stdio(false); cin.tie(NULL)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define sz(x) ((ll)(x).size())

struct custom_hash {
	static uint64_t splitmix64(uint64_t x) {
		// http://xorshift.di.unimi.it/splitmix64.c
		x += 0x9e3779b97f4a7c15;
		x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
		x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
		return x ^ (x >> 31);
	}

	size_t operator()(uint64_t x) const {
		static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
		return splitmix64(x + FIXED_RANDOM);
	}
};

struct pair_hash {
	size_t operator()(const pair<int, int> &p) const {
		size_t h1 = custom_hash{}(p.first);
		size_t h2 = custom_hash{}(p.second);
		return h1 ^ (h2 << 1);
	}
};

struct rect_hash {
	size_t operator()(const pair<pair<int, int>, pair<int, int>> &p) const {
		size_t h1 = pair_hash{}(p.first);
		size_t h2 = pair_hash{}(p.second);
		return h1 ^ (h2 << 1);
	}
};

using rect = pair<pair<int, int>, pair<int, int>>;

template <typename Key, typename Value>
using hash_map = unordered_map<Key, Value, custom_hash>;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
// uniform_int_distribution<int>(a, b)(rng);
// shuffle(all(a), rng);

struct stuff {
	short coll, colr, rowl, rowr;

	stuff(short coll = -INF, short colr = INF, short rowl = -INF, short rowr = INF) : coll(coll), colr(colr), rowl(rowl), rowr(rowr) {;;}
};

ll count_rectangles(vector<vector<int>> a) {
	short n = a.size(), m = a[0].size();
	short logn = log2(n) + 1, logm = log2(m) + 1;
	ll ans = 0;

	vector<vector<vector<vector<stuff>>>> searchr(logn,
		vector<vector<vector<stuff>>>(n,
			vector<vector<stuff>>(logm,
				vector<stuff>(m)
			)
		)
	);

	stack<short> st;

	// O(nm)
	for (short i = 0; i < n; i++) {
		while (!st.empty()) st.pop();

		for (short j = 0; j < m; j++) {
			while (!st.empty() and a[i][st.top()] <= a[i][j]) {
				st.pop();
			}

			if (!st.empty()) {
				searchr[0][i][0][j].rowl = st.top();
			}

			st.push(j);
		}

		while (!st.empty()) st.pop();

		for (short j = m-1; j >= 0; j--) {
			while (!st.empty() and a[i][st.top()] <= a[i][j]) {
				st.pop();
			}

			if (!st.empty()) {
				searchr[0][i][0][j].rowr = st.top();
			}

			st.push(j);
		}
	}

	for (short i = 0; i < m ; i++) {
		while (!st.empty()) st.pop();

		for (short j = 0; j < n; j++) {
			while (!st.empty() and a[st.top()][i] <= a[j][i]) {
				st.pop();
			}

			if (!st.empty()) {
				searchr[0][j][0][i].coll = st.top();
			}

			st.push(j);
		}

		while (!st.empty()) st.pop();

		for (short j = n-1; j >= 0; j--) {
			while (!st.empty() and a[st.top()][i] <= a[j][i]) {
				st.pop();
			}

			if (!st.empty()) {
				searchr[0][j][0][i].colr = st.top();
			}

			st.push(j);
		}
	}

	auto merge = [&](const stuff &a, const stuff &b) -> stuff {
		stuff res;
		res.coll = min(a.coll, b.coll);
		res.colr = max(a.colr, b.colr);
		res.rowl = min(a.rowl, b.rowl);
		res.rowr = max(a.rowr, b.rowr);
		return res;	
	};

	// O(nm lg n lg m)
	for (short i = 0; i < n; i++) {
		for (short j = 1; j < logm; j++) {
			for (short k = 0; k + (1 << j) - 1 < m; k++) {
				searchr[0][i][j][k] = merge(searchr[0][i][j-1][k], searchr[0][i][j-1][k+(1<<(j-1))]);
			}
		}
	}

	for (short i = 1; i < logn; i++) {
		for (short j = 0; j + (1 << i) - 1 < n; j++) {
			for (short k = 0; k < logm; k++) {
				for (short l = 0; l < m; l++) {
					searchr[i][j][k][l] = merge(searchr[i-1][j][k][l], searchr[i-1][j+(1<<(i-1))][k][l]);
				}
			}
		}
	}

	auto in = [&](short x, short y) -> bool {
		return (0 < x and x < n-1 and 0 < y and y < m-1);
	};

	auto query = [&](short x, short y, short xx, short yy) -> stuff {
		short szx = xx - x + 1, szy = yy - y + 1;
		short lgx = log2(szx), lgy = log2(szy);

		stuff r1 = merge(searchr[lgx][x][lgy][y], searchr[lgx][x][lgy][yy-(1<<lgy)+1]);
		stuff r2 = merge(searchr[lgx][xx-(1<<lgx)+1][lgy][y], searchr[lgx][xx-(1<<lgx)+1][lgy][yy-(1<<lgy)+1]);
		return merge(r1, r2);
	};

	// O(n + m)
	unordered_map<rect, bool, rect_hash> vis;
	auto checkRect = [&](short x, short y, short xx, short yy) -> int {
		if (!in(x,y) or !in(xx, yy)) return 0;

		if (x > xx) swap(x, xx);
		if (y > yy) swap(y, yy);

		if (vis[{{x, y}, {xx, yy}}]) return 0;

		vis[{{x, y}, {xx, yy}}] = true;

		stuff q = query(x, y, xx, yy);

		if (q.coll < x-1 or q.colr > xx+1 or q.rowl < y-1 or q.rowr > yy+1) {
			return 0;
		}

		return 1;
	};

	// O(nm (n+m))
	for (short i = 1; i < n - 1; i++) {
		for (short j = 1; j < m - 1; j++) {
			stuff q = searchr[0][i][0][j];
			if (q.coll == -INF or q.colr == INF or q.rowl == -INF or q.rowr == INF) {
				continue;
			}
			ans += checkRect(q.coll+1, q.rowl+1, q.colr-1, q.rowr-1);

		}
	}

	return ans;
}

// int main() {
// 	int n, m; cin >> n >> m;
// 	vector<vector<int>> a(n, vector<int>(m));
// 	REP(i, 0, n) {
// 		REP(j, 0, m) {
// 			cin >> a[i][j];
// 		}
// 	}

// 	cout << count_rectangles(a) << endl;
// }

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