이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "rect.h"
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
using namespace std;
using i64 = long long;
using d64 = long double;
using pi = pair<int, int>;
using pli = pair<i64, i64>;
using ti = tuple<int, int, int>;
using tli = tuple<i64, i64, i64>;
#define iterall(cont) cont.begin(), cont.end()
#define prec(n) setprecision(n) << fixed
vector<pi> findLegalTuple(const vector<int> &v) {
vector<pi> ret;
vector<size_t> seq; // monotonic
size_t sn = 0;
for (size_t i = 0; i < v.size(); i++) {
while (sn >= 1 && v[seq[sn - 1]] < v[i]) {
seq.pop_back(), --sn;
if (sn >= 1 && i - seq[sn - 1] >= 2)
ret.emplace_back(seq[sn - 1], i);
}
if (sn >= 1 && v[seq[sn - 1]] == v[i]) seq.pop_back(), --sn;
seq.emplace_back(i), ++sn;
}
return ret;
}
// 이거 N^3이라서 쫌 이따 바꿔야지
vector<pi> getMaxPos(const vector<int> &v) {
size_t N = v.size();
vector<pi> ret(N, {-1, -1});
{
vector<int> seq; // monotone
size_t sn = 0;
for (int i = 0; i < N; i++) {
while (sn >= 1 && v[seq[sn - 1]] < v[i]) {
ret[seq[sn - 1]].second = i;
seq.pop_back(), --sn;
}
seq.emplace_back(i), ++sn;
}
}
{
vector<int> seq; // monotone
size_t sn = 0;
for (int i = N - 1; i >= 0; i--) {
while (sn >= 1 && v[seq[sn - 1]] < v[i]) {
ret[seq[sn - 1]].first = i;
seq.pop_back(), --sn;
}
seq.emplace_back(i), ++sn;
}
}
return ret;
}
vector<pi> processToIntervals(const vector<pi> &v) {
vector<pi> ret;
for (auto [el, _] : v) {
if (!ret.empty() && ret.back().second == el - 1)
ret.back().second++;
else
ret.emplace_back(el, el);
}
return ret;
}
bool intervalBinarySearch(const vector<pi> &v, pi t) {
if (v.empty()) return false;
const size_t N = v.size();
int r1 = -1;
{
size_t s = 0, e = N - 1;
while (e - s >= 2) {
size_t m = (s + e) >> 1;
if (v[m].first <= t.first && t.first <= v[m].second) {
r1 = m;
break;
}
if (v[m].second < t.first) s = m + 1;
if (v[m].first > t.first) e = m - 1;
}
if (v[s].first <= t.first && t.first <= v[s].second) r1 = s;
if (v[e].first <= t.first && t.first <= v[e].second) r1 = e;
}
int r2 = -1;
{
size_t s = 0, e = N - 1;
while (e - s >= 2) {
size_t m = (s + e) >> 1;
if (v[m].first <= t.second && t.second <= v[m].second) {
r2 = m;
break;
}
if (v[m].second < t.second) s = m + 1;
if (v[m].first > t.second) e = m - 1;
}
if (v[s].first <= t.second && t.second <= v[s].second) r2 = s;
if (v[e].first <= t.second && t.second <= v[e].second) r2 = e;
}
return r1 != -1 && r2 != -1 && r1 == r2;
}
i64 _hs(i64 a, i64 b, i64 c, i64 d) {
i64 x = 2501;
return x * x * x * a + x * x * b + x * c + d;
}
i64 count_rectangles(vector<vector<int>> a) {
const size_t N = a.size();
const size_t M = a[0].size();
// transpose
vector<vector<int>> b(M, vector<int>(N));
for (int i = 0; i < N; i++)
for (int j = 0; j < M; j++) b[j][i] = a[i][j];
// get Max Position
vector<vector<pi>> aM(N, vector<pi>(M));
vector<vector<pi>> bM(M, vector<pi>(N));
for (int i = 0; i < N; i++) aM[i] = getMaxPos(a[i]);
for (int i = 0; i < M; i++) bM[i] = getMaxPos(b[i]);
// find Legal Tuple && assign
vector<vector<vector<pi>>> aL(M, decltype(aL)::value_type(M));
vector<vector<vector<pi>>> bL(N, decltype(bL)::value_type(N));
for (int i = 0; i < N; i++) {
auto Lg = findLegalTuple(a[i]);
for (auto [l, r] : Lg) aL[l][r].emplace_back(i, 0);
}
for (int i = 0; i < M; i++) {
auto Lg = findLegalTuple(b[i]);
for (auto [l, r] : Lg) bL[l][r].emplace_back(i, 0);
}
for (auto &vec : aL)
for (auto &el : vec) el = processToIntervals(el);
for (auto &vec : bL)
for (auto &el : vec) el = processToIntervals(el);
// Main Logic
unordered_set<i64> us;
for (int i = 0; i < N; i++) {
for (int j = 0; j < M; j++) {
auto [L, R] = aM[i][j];
auto [T, B] = bM[j][i];
auto H = _hs(L, R, T, B);
if (L == -1 || R == -1 || T == -1 || B == -1) continue;
if (us.find(H) != us.end()) continue;
if (intervalBinarySearch(aL[L][R], {T + 1, B - 1}) &&
intervalBinarySearch(bL[T][B], {L + 1, R - 1}))
us.insert(H);
}
}
return us.size();
}
컴파일 시 표준 에러 (stderr) 메시지
rect.cpp: In function 'std::vector<std::pair<int, int> > getMaxPos(const std::vector<int>&)':
rect.cpp:47:27: warning: comparison of integer expressions of different signedness: 'int' and 'size_t' {aka 'long unsigned int'} [-Wsign-compare]
47 | for (int i = 0; i < N; i++) {
| ~~^~~
rect.cpp: In function 'i64 count_rectangles(std::vector<std::vector<int> >)':
rect.cpp:141:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
141 | for (int i = 0; i < N; i++)
| ~~^~~
rect.cpp:142:27: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
142 | for (int j = 0; j < M; j++) b[j][i] = a[i][j];
| ~~^~~
rect.cpp:148:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
148 | for (int i = 0; i < N; i++) aM[i] = getMaxPos(a[i]);
| ~~^~~
rect.cpp:149:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
149 | for (int i = 0; i < M; i++) bM[i] = getMaxPos(b[i]);
| ~~^~~
rect.cpp:155:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
155 | for (int i = 0; i < N; i++) {
| ~~^~~
rect.cpp:159:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
159 | for (int i = 0; i < M; i++) {
| ~~^~~
rect.cpp:172:23: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
172 | for (int i = 0; i < N; i++) {
| ~~^~~
rect.cpp:173:27: warning: comparison of integer expressions of different signedness: 'int' and 'const size_t' {aka 'const long unsigned int'} [-Wsign-compare]
173 | for (int j = 0; j < M; j++) {
| ~~^~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |