# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
204301 | Segtree | 팀들 (IOI15_teams) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include"teams.h"
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
#define chmin(a,b) a=min(a,b)
#define chmax(a,b) a=max(a,b)
#define all(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<n;i++)
#define mod 1000000007
#define mad(a,b) a=(a+b)%mod
#define up(x,y) upper_bound(all(x),y)
#define NS 100010
vector<P> v;
ll dat[NS*2],laste[2*NS],lastd[2*NS],correctA;
vector<ll> as[NS*2];
unordered_set<ll> used;
void add(int K){
correctA=K;
}
ll eval(ll k){
if(k==0)return 0;
used.insert(k);
if(k<NS){
chmax(lastd[k*2],lastd[k]); used.insert(k*2);
chmax(lastd[k*2+1],lastd[k]); used.inert(k*2+1);
}
if(lastd[k]==-1e17){
dat[k]+=up(as[k],correctA)-up(as[k],laste[k]);
}
else{
dat[k]=up(as[k],correctA)-up(as[k],lastd[k]);
}
laste[k]=correctA;
lastd[k]=-1e17;
return dat[k];
}
void erase(ll l,ll r){
l+=NS,r+=NS+1;
for(ll d=20;~d;d--){
eval(l>>d);
eval(r>>d);
}
for(ll a=l,b=r;a<b;a>>=1,b>>=1){
if(a&1){chmax(lastd[a],correctA);a++;}
if(b&1){b--;chmax(lastd[b],correctA);}
}
for(ll a=l,b=r;a+b;a>>=1,b>>=1){
dat[a/2]=eval(a)+eval(a^1);
dat[b/2]=eval(b)+eval(b^1);
}
}
int sum(ll l,ll r){
l+=NS,r+=NS+1;
ll res=0;
for(ll d=20;~d;d--){
eval(l>>d);
eval(r>>d);
}
for(ll a=l,b=r;a<b;a>>=1,b>>=1){
if(a&1){res+=eval(a);a++;}
if(b&1){b--;res+=eval(b);}
}
return res;
}
void init(int N,int A[],int B[]){
rep(i,N)v.push_back(make_pair(B[i],A[i]));
sort(v.begin(),v.end());
rep(i,NS){
if(i<N)as[i+NS].push_back(v[i].second);
}
for(int i=NS-1;i>0;i--){
for(auto t:as[i*2])as[i].push_back(t);
for(auto t:as[i*2+1])as[i].push_back(t);
sort(as[i].begin(),as[i].end());
}
}
int can(int M,int K[]){
used.clear();
correctA=0;
sort(K,K+M);
bool ans=1;
rep(i,M){
add(K[i]);
ll l,r,mid;
l=-1,r=v.size();
while(l<r-1){
mid=(l+r)>>1;
(v[mid].first<K[i]?l:r)=mid;
}
//cerr<<"##"<<sum(0,v.size()-1)<<endl;
erase(0,l);//cerr<<"$"<<l<<endl;
l=-1,r=v.size();
//cerr<<"#"<<sum(0,v.size()-1)<<endl;
while(l<r-1){
mid=(l+r)>>1;
(sum(0,mid)>=K[i]?r:l)=mid;
}
if(r==v.size()){
ans=0;
break;
}
erase(0,r);
}
for(auto i:used)dat[i]=0,laste[i]=lastd[i]=-1e17;
return ans;
}
/*
int main(){
int n,a[110],b[110];
cin>>n;
rep(i,n)cin>>a[i]>>b[i];
init(n,a,b);
while(1){
int m,k[110];
cin>>m;
rep(i,m)cin>>k[i];
cout<<(can(m,k)?"YES":"NO")<<endl;
}
}*/