# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
143513 |
2019-08-14T12:30:15 Z |
icypiggy |
Rectangles (IOI19_rect) |
C++14 |
|
614 ms |
636136 KB |
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#include <stack>
using namespace std;
const int n_max = 2600; // make it not lag
vector<pair<int,bool>> h_seg[n_max][n_max];
vector<pair<int,bool>> v_seg[n_max][n_max];
vector<pair<int,int>> h_map[n_max][n_max];
vector<pair<int,int>> v_map[n_max][n_max];
int get_lower(set<int> &s, int val) {
auto it = s.lower_bound(val);
it--;
return *it;
}
void gen_h_seg(vector<int> &x, int r) {
/*
vector<pair<int,int>> ret;
vector<pair<int,int>> y;
for(int i=0; i<x.size(); i++) {
y.push_back(make_pair(-x[i],i));
}
sort(y.begin(), y.end());
set<int> s;
s.insert(-2e8);
s.insert(2e8);
for(int i=0; i<y.size(); i++) {
int z = get_lower(s,y[i].second);
if(z!=-2e8 && y[i].second!=z+1)
h_seg[r][z+1].push_back(make_pair(y[i].second-1, true));
z = *(s.upper_bound(y[i].second));
if((i+1==y.size() || (y[i].first!=y[i+1].first || z<y[i+1].second)) && z!=2e8 && y[i].second+1!=z) {
h_seg[r][y[i].second+1].push_back(make_pair(z-1,true));
}
s.insert(y[i].second);
}*/
stack<int> s;
s.push(0);
for(int i=1; i<x.size(); i++) {
while(!s.empty() && x[s.top()]<x[i]) {
if(s.top()!=i-1) {
h_seg[r][s.top()+1].push_back(make_pair(i-1, true));
//cout << r << " " << s.top()+1 << " " << i-1 << "\n";
}
s.pop();
}
if(!s.empty() && s.top()!=i-1) {
h_seg[r][s.top()+1].push_back(make_pair(i-1, true));
//cout << r << " " << s.top()+1 << " " << i-1 << "\n";
}
s.push(i);
}
}
struct fenwick {
int BIT[n_max], a[n_max], n;
fenwick(): n(0) {}
void reset() {
for(int i=0; i<n_max; i++) {
BIT[i] = 0;
a[i] = 0;
}
}
void update(int x, int delta)
{
for(; x <= n; x += x&-x)
BIT[x] += delta;
}
int query(int x) {
int sum = 0;
for(; x > 0; x -= x&-x)
sum += BIT[x];
return sum;
}
int query(int x, int y) {
return query(y)-query(x-1);
}
};
void gen_v_seg(vector<int> &x, int r) {
stack<int> s;
s.push(0);
for(int i=1; i<x.size(); i++) {
while(!s.empty() && x[s.top()]<x[i]) {
if(s.top()!=i-1) {
v_seg[r][s.top()+1].push_back(make_pair(i-1, true));
//cout << r << " " << s.top()+1 << " " << i-1 << "\n";
}
s.pop();
}
if(!s.empty() && s.top()!=i-1) {
v_seg[r][s.top()+1].push_back(make_pair(i-1, true));
//cout << r << " " << s.top()+1 << " " << i-1 << "\n";
}
s.push(i);
}
}
long long count_rectangles(vector<vector<int>> a) {
for(int i=0; i<a.size(); i++) {
gen_h_seg(a[i],i);
}
for(int i=0; i<a[0].size(); i++) {
vector<int> tmp;
for(int j=0; j<a.size(); j++) {
tmp.push_back(a[j][i]);
}
gen_v_seg(tmp, i);
}
for(int i=0; i<a.size(); i++) {
for(int j=0; j<a[0].size(); j++) {
sort(h_seg[i][j].begin(), h_seg[i][j].end());
sort(v_seg[j][i].begin(), v_seg[j][i].end());
}
}
for(int i=0; i<a.size(); i++) {
for(int j=0; j<a[0].size(); j++) {
for(auto &p: h_seg[i][j]) {
if(p.second) {
int down = i;
while(down+1!=a.size()) {
auto it = lower_bound(h_seg[down+1][j].begin(), h_seg[down+1][j].end(), make_pair(p.first,true));
if(it==h_seg[down+1][j].end()) break;
if(*it==make_pair(p.first, true)) {
it->second = false;
down++;
} else {
break;
}
}
h_map[i][j].push_back(make_pair(p.first, down));
}
}
}
}
for(int i=0; i<a[0].size(); i++) {
for(int j=0; j<a.size(); j++) {
for(auto &p: v_seg[i][j]) {
if(p.second) {
int down = i;
while(down+1!=a[0].size()) {
auto it = lower_bound(v_seg[down+1][j].begin(), v_seg[down+1][j].end(), make_pair(p.first,true));
if(it==v_seg[down+1][j].end()) break;
if(*it==make_pair(p.first, true)) {
it->second = false;
down++;
} else {
break;
}
}
v_map[i][j].push_back(make_pair(p.first, down));
//cout << "i,j,p.first,down(v): " << i << " " << j << " " << p.first << " " << down << "\n";
}
}
}
}
for(int i=a[0].size()-1; i>=0; i--) {
for(int j=a.size()-1; j>=0; j--) {
for(auto p: v_map[i][j]) {
for(int k=i+1; k<=p.second; k++) {
v_map[k][j].push_back(p);
//cout << "k,j,p: " << k << " " << j << " " << p.first << " " << p.second << "\n";
}
}
}
}
fenwick f[a.size()];
for(int i=0; i<a.size(); i++) {
f[i].n = a[0].size()+10;
f[i].reset();
}
long long ans = 0;
for(int i=0; i<a[0].size(); i++) {
vector<pair<int,int>> updates;
for(int j=0; j<a.size(); j++) {
//cout << "coordinate: " << i << " " << j << "\n";
for(auto p: h_map[j][i]) {
for(int k=j; k<=p.second; k++) {
f[k].update(p.first, 1);
//cout << "updated: " << k << "-th fenwick, position " << p.first << "increased by 1.\n";
updates.push_back(make_pair(k, p.first));
}
}
for(auto p: v_map[i][j]) {
int tmp = f[p.first].query(i,p.second);
ans += tmp;
//cout << "ans changed by " << tmp << " queried: " << p.first << " " << i << " " << p.second << "\n";
}
}
for(auto p: updates) {
f[p.first].update(p.second, -1);
//cout << "undo increment: " << p.first << " " << p.second << "\n";
}
}
//cout << "ans: " << ans << "\n";
return ans;
}
Compilation message
rect.cpp: In function 'void gen_h_seg(std::vector<int>&, int)':
rect.cpp:40:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=1; i<x.size(); i++) {
~^~~~~~~~~
rect.cpp: In function 'void gen_v_seg(std::vector<int>&, int)':
rect.cpp:82:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=1; i<x.size(); i++) {
~^~~~~~~~~
rect.cpp: In function 'long long int count_rectangles(std::vector<std::vector<int> >)':
rect.cpp:99:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a.size(); i++) {
~^~~~~~~~~
rect.cpp:102:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a[0].size(); i++) {
~^~~~~~~~~~~~
rect.cpp:104:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int j=0; j<a.size(); j++) {
~^~~~~~~~~
rect.cpp:109:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a.size(); i++) {
~^~~~~~~~~
rect.cpp:110:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int j=0; j<a[0].size(); j++) {
~^~~~~~~~~~~~
rect.cpp:115:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a.size(); i++) {
~^~~~~~~~~
rect.cpp:116:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int j=0; j<a[0].size(); j++) {
~^~~~~~~~~~~~
rect.cpp:121:33: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while(down+1!=a.size()) {
~~~~~~^~~~~~~~~~
rect.cpp:136:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a[0].size(); i++) {
~^~~~~~~~~~~~
rect.cpp:137:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int j=0; j<a.size(); j++) {
~^~~~~~~~~
rect.cpp:142:33: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while(down+1!=a[0].size()) {
~~~~~~^~~~~~~~~~~~~
rect.cpp:169:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a.size(); i++) {
~^~~~~~~~~
rect.cpp:175:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<a[0].size(); i++) {
~^~~~~~~~~~~~
rect.cpp:177:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int j=0; j<a.size(); j++) {
~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
609 ms |
635504 KB |
Output is correct |
2 |
Correct |
574 ms |
636136 KB |
Output is correct |
3 |
Correct |
578 ms |
636004 KB |
Output is correct |
4 |
Correct |
579 ms |
636000 KB |
Output is correct |
5 |
Correct |
585 ms |
635916 KB |
Output is correct |
6 |
Correct |
581 ms |
635964 KB |
Output is correct |
7 |
Correct |
614 ms |
635916 KB |
Output is correct |
8 |
Correct |
589 ms |
635696 KB |
Output is correct |
9 |
Incorrect |
578 ms |
635924 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
609 ms |
635504 KB |
Output is correct |
2 |
Correct |
574 ms |
636136 KB |
Output is correct |
3 |
Correct |
578 ms |
636004 KB |
Output is correct |
4 |
Correct |
579 ms |
636000 KB |
Output is correct |
5 |
Correct |
585 ms |
635916 KB |
Output is correct |
6 |
Correct |
581 ms |
635964 KB |
Output is correct |
7 |
Correct |
614 ms |
635916 KB |
Output is correct |
8 |
Correct |
589 ms |
635696 KB |
Output is correct |
9 |
Incorrect |
578 ms |
635924 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
609 ms |
635504 KB |
Output is correct |
2 |
Correct |
574 ms |
636136 KB |
Output is correct |
3 |
Correct |
578 ms |
636004 KB |
Output is correct |
4 |
Correct |
579 ms |
636000 KB |
Output is correct |
5 |
Correct |
585 ms |
635916 KB |
Output is correct |
6 |
Correct |
581 ms |
635964 KB |
Output is correct |
7 |
Correct |
614 ms |
635916 KB |
Output is correct |
8 |
Correct |
589 ms |
635696 KB |
Output is correct |
9 |
Incorrect |
578 ms |
635924 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
609 ms |
635504 KB |
Output is correct |
2 |
Correct |
574 ms |
636136 KB |
Output is correct |
3 |
Correct |
578 ms |
636004 KB |
Output is correct |
4 |
Correct |
579 ms |
636000 KB |
Output is correct |
5 |
Correct |
585 ms |
635916 KB |
Output is correct |
6 |
Correct |
581 ms |
635964 KB |
Output is correct |
7 |
Correct |
614 ms |
635916 KB |
Output is correct |
8 |
Correct |
589 ms |
635696 KB |
Output is correct |
9 |
Incorrect |
578 ms |
635924 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
608 ms |
635932 KB |
Output is correct |
2 |
Correct |
573 ms |
635772 KB |
Output is correct |
3 |
Correct |
575 ms |
635404 KB |
Output is correct |
4 |
Correct |
575 ms |
635384 KB |
Output is correct |
5 |
Correct |
578 ms |
635768 KB |
Output is correct |
6 |
Correct |
575 ms |
635796 KB |
Output is correct |
7 |
Correct |
573 ms |
635704 KB |
Output is correct |
8 |
Correct |
584 ms |
635684 KB |
Output is correct |
9 |
Incorrect |
608 ms |
635740 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
579 ms |
635516 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
609 ms |
635504 KB |
Output is correct |
2 |
Correct |
574 ms |
636136 KB |
Output is correct |
3 |
Correct |
578 ms |
636004 KB |
Output is correct |
4 |
Correct |
579 ms |
636000 KB |
Output is correct |
5 |
Correct |
585 ms |
635916 KB |
Output is correct |
6 |
Correct |
581 ms |
635964 KB |
Output is correct |
7 |
Correct |
614 ms |
635916 KB |
Output is correct |
8 |
Correct |
589 ms |
635696 KB |
Output is correct |
9 |
Incorrect |
578 ms |
635924 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |