| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 927729 | MrDeboo | 자리 배치 (IOI18_seats) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
using ordered_set = tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>;
vector<pair<int,int>>v;
bool hehe=0;
map<int,ordered_set>a,b;
int slv(int in){
pair<pair<int,int>,pair<int,int>>pr={{v[in].first,v[in].first},{v[in].second,v[in].second}};
for(int i=in;i>=0;i--){
pr.first.first=min(pr.first.first,v[i].first);
pr.first.second=max(pr.first.second,v[i].first);
pr.second.first=min(pr.second.first,v[i].second);
pr.second.second=max(pr.second.second,v[i].second);
}
return (in+1==(pr.second.second-pr.second.first+1)*(pr.first.second-pr.first.first+1));
}
void give_initial_chart(int H, int W, std::vector<int> R, std::vector<int> C) {
if(H==1)hehe=1;
v.resize(H*W);
for(int i=0;i<H*W;i++){
v[i]={R[i],C[i]};
if(i)a[C[i]+i].insert(C[i]);
if(i)b[C[i]-i].insert(C[i]);
}
}
void rmv(int in){
if(in==0)return;
a[v[in].second+i].erase(v[in].second);
b[v[in].second-i].erase(v[in].second);
}
void ad(int in){
if(in==0)return;
a[v[in].second+i].insert(v[in].second);
b[v[in].second-i].insert(v[in].second);
}
int clc(){
int g=v[0].second;
int ans=0;
ans+=a[g].order_of_key(g);
ans+=b[g].size()-b[g].order_of_key(g);
}
int swap_seats(int a, int b){
if(hehe){
rmv(a);
rmv(b);
swap(v[a],v[b]);
ad(a);
ad(b);
return clc();
}
int ans=0;
swap(v[a],v[b]);
pair<pair<int,int>,pair<int,int>>pr={{v[0].first,v[0].first},{v[0].second,v[0].second}};
for(int i=0;i<v.size();i++){
pr.first.first=min(pr.first.first,v[i].first);
pr.first.second=max(pr.first.second,v[i].first);
pr.second.first=min(pr.second.first,v[i].second);
pr.second.second=max(pr.second.second,v[i].second);
if(i+1==(pr.second.second-pr.second.first+1)*(pr.first.second-pr.first.first+1))ans++;
}
return ans;
}