이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/extc++.h"
using namespace std;
template <typename T, typename... U>
void dbgh(const T& t, const U&... u) {
cerr << t;
((cerr << " | " << u), ...);
cerr << endl;
}
#ifdef DEBUG
#define dbg(...) \
cerr << "L" << __LINE__ << " [" << #__VA_ARGS__ << "]: "; \
dbgh(__VA_ARGS__)
#else
#define dbg(...)
#define cerr \
if (false) \
cerr
#endif
using ll = long long;
#define endl "\n"
#define long int64_t
#define sz(x) int(std::size(x))
namespace __gnu_pbds {
template <typename T>
using ost =
tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
}
using __gnu_pbds::ost;
template <typename T>
ostream& operator<<(ostream& out, const vector<T>& arr) {
out << "[";
for (int i = 0; i < sz(arr); i++) {
if (i) {
out << ", ";
}
out << arr[i];
}
return out << "]";
}
namespace s1 {
constexpr int DIRS[4][2] {{0, 1}, {0, -1}, {-1, 0}, {1, 0}};
int n, m;
vector<vector<char>> vis;
vector<vector<int>> arr;
bool ibs(int x, int y) {
return 0 <= x && x < n && 0 <= y && y < m;
}
template <typename Cb>
void dfs(int x, int y, const Cb& cb) {
if (!ibs(x, y) || vis[x][y] || arr[x][y]) {
return;
}
cb(x, y);
vis[x][y] = true;
for (auto& [dx, dy] : DIRS) {
dfs(x + dx, y + dy, cb);
}
}
long solve(const vector<vector<int>>& _arr) {
arr = _arr;
n = sz(arr);
m = sz(arr[0]);
vis.resize(n, vector<char>(m));
long ans = 0;
auto c2 = [&](long n) -> long { return n * (n - 1) / 2; };
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (vis[i][j] || arr[i][j]) {
continue;
}
int mnx = 1e9, mxx = 0, mny = 1e9, mxy = 0, cnt = 0;
dfs(i, j, [&](int x, int y) -> void {
mnx = min(mnx, x);
mxx = max(mxx, x);
mny = min(mny, y);
mxy = max(mxy, y);
cnt++;
});
if (cnt == (mxx - mnx + 1) * (mxy - mny + 1) && mnx && mny &&
mxx < n - 1 && mxy < m - 1) {
ans++;
}
}
}
return ans;
}
} // namespace s1
struct PSA {
vector<int> psum;
PSA(const vector<int>& arr) : psum(sz(arr) + 1) {
partial_sum(begin(arr), end(arr), psum.begin() + 1);
}
int query(int l, int r) const {
return psum[r] - psum[l];
}
};
vector<pair<int, int>> solve1(const vector<int>& arr) {
int n = sz(arr);
vector<pair<int, int>> ans, st;
st.emplace_back(arr.back(), n - 1);
for (int i = n - 2; i >= 0; i--) {
while (sz(st)) {
if (i + 1 <= st.back().second - 1) {
dbg(i + 1, st.back().second - 1);
ans.emplace_back(i + 1, st.back().second - 1);
}
if (st.back().first < arr[i]) {
st.pop_back();
} else if (st.back().first == arr[i]) {
st.pop_back();
break;
} else {
break;
}
}
st.emplace_back(arr[i], i);
}
assert(sz(ans) <= 2 * n + 10);
sort(begin(ans), end(ans));
return ans;
}
template <typename T>
vector<T> set_inter(const vector<T>& a, const vector<T>& b) {
vector<T> out;
set_intersection(begin(a), end(a), begin(b), end(b), back_inserter(out));
return out;
}
template <typename T>
vector<T> set_diff(const vector<T>& a, const vector<T>& b) {
vector<T> out;
set_difference(begin(a), end(a), begin(b), end(b), back_inserter(out));
return out;
}
struct DS1 {
int n;
vector<int> arr;
vector<ost<pair<int, int>>> v;
DS1(int n) : n(n), arr(n), v(4 * n) {
build(1, 0, n - 1);
}
void build(int o, int l, int r) {
for (int i = l; i <= r; i++) {
v[o].insert({arr[i], i});
}
if (l == r) {
return;
}
int mid = (l + r) / 2, lc = o * 2, rc = lc + 1;
build(lc, l, mid);
build(rc, mid + 1, r);
}
void update(int o, int l, int r, int ind, int x) {
v[o].erase({arr[ind], ind});
v[o].insert({x, ind});
if (l == r) {
arr[ind] = x;
return;
}
int mid = (l + r) / 2, lc = o * 2, rc = lc + 1;
if (ind <= mid) {
update(lc, l, mid, ind, x);
} else {
update(rc, mid + 1, r, ind, x);
}
}
void update(int ind, int x) {
update(1, 0, n - 1, ind, x);
// arr[ind] = x;
}
int query(int o, int l, int r, int qr, int x) const {
if (r <= qr) {
return sz(v[o]) - int(v[o].order_of_key({x + 1, -1}));
}
int mid = (l + r) / 2, lc = o * 2, rc = lc + 1;
int ans = query(lc, l, mid, qr, x);
if (mid < qr) {
ans += query(rc, mid + 1, r, qr, x);
}
return ans;
}
int query(int ind, int x) const {
return query(1, 0, n - 1, ind - 1, x);
// int ans = 0;
// for (int i = 0; i < ind; i++) {
// ans += arr[i] > x;
// }
// return ans;
}
};
long solve2(int r_shift,
int n,
int m,
const vector<pair<int, int>>& arr,
const vector<array<int, 3>>& queries) {
n -= r_shift;
vector<int> e_arr[m];
for (auto& [ind, x] : arr) {
e_arr[ind].push_back(x - r_shift);
dbg(ind, x - r_shift);
}
vector<pair<int, int>> e_queries[m];
for (auto& [x, ql, qr] : queries) {
e_queries[ql].emplace_back(x - r_shift, qr);
dbg(ql, qr, x - r_shift);
}
long ans = 0;
DS1 ds(n);
vector<int> p_arr;
for (int i = m - 1; i >= 0; i--) {
sort(begin(e_arr[i]), end(e_arr[i]));
for (auto& x : set_diff(p_arr, e_arr[i])) {
ds.update(x, 0);
}
p_arr = e_arr[i];
for (auto& a : e_arr[i]) {
if (!ds.arr[a]) {
ds.update(a, i);
}
}
for (auto& [x, qr] : e_queries[i]) {
dbg(x, n);
ans += ds.query(x + 1, qr - 1);
}
}
return ans;
}
ll count_rectangles(vector<vector<int>> arr) {
int n = sz(arr), m = sz(arr[0]);
{
bool ok = true;
for (auto& a : arr) {
for (auto& b : a) {
ok &= 0 <= b && b <= 1;
}
}
if (ok) {
return s1::solve(arr);
}
}
vector<pair<int, int>> r_res[n], c_res[m];
for (int i = 0; i < n; i++) {
r_res[i] = solve1(arr[i]);
}
for (int i = 0; i < m; i++) {
vector<int> tmp;
for (int j = 0; j < n; j++) {
tmp.push_back(arr[j][i]);
}
c_res[i] = solve1(tmp);
}
vector<int> r_good[m][m];
for (int i = 0; i < n; i++) {
for (auto& [ql, qr] : r_res[i]) {
r_good[ql][qr].push_back(i);
}
}
vector<array<int, 3>> queries[n];
for (int cl = 0; cl < m; cl++) {
for (int cr = cl; cr < m; cr++) {
auto& carr = r_good[cl][cr];
for (int i = 0; i < sz(carr);) {
int ci = i;
for (i++; i < sz(carr) && carr[i - 1] + 1 == carr[i]; i++)
;
int cql = carr[ci], cqr = carr[i - 1];
cql = max(cql, 1);
cqr = min(cqr, n - 2);
for (int j = cql; j <= cqr; j++) {
queries[j].push_back({cqr, cl, cr});
}
}
}
}
long ans = 0;
vector<pair<int, int>> rvals[n];
for (int i = 0; i < m; i++) {
for (auto& [rql, rqr] : c_res[i]) {
rvals[rql].emplace_back(i, rqr);
}
}
for (int r = 0; r < n; r++) {
ans += solve2(r, n, m, rvals[r], queries[r]);
dbg(r, sz(rvals[r]), sz(queries[r]), ans);
}
return ans;
}
컴파일 시 표준 에러 (stderr) 메시지
rect.cpp: In function 'int64_t s1::solve(const std::vector<std::vector<int> >&)':
rect.cpp:81:10: warning: variable 'c2' set but not used [-Wunused-but-set-variable]
81 | auto c2 = [&](long n) -> long { return n * (n - 1) / 2; };
| ^~
# | 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... |