제출 #859207

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
859207		abcvuitunggio	커다란 상품 (IOI17_prize)	C++17	97.70 / 100	33 ms	11104 KiB

prize

#include "prize.h"
#include <bits/stdc++.h>
using namespace std;
vector <int> a[200000];
int ch[200000],st[800001],f[200001];
void build(int node, int l, int r){
    st[node]=r-l+1;
    if (l==r)
        return;
    int mid=(l+r)>>1;
    build(node<<1,l,mid);
    build(node<<1|1,mid+1,r);
}
void update(int node, int l, int r, int i){
    if (l>i||r<i)
        return;
    if (l==r){
        st[node]=0;
        return;
    }
    int mid=(l+r)>>1;
    update(node<<1,l,mid,i);
    update(node<<1|1,mid+1,r,i);
    st[node]=st[node<<1]+st[node<<1|1];
}
int get(int node, int l, int r, int u, int v){
    if (l>v||r<u||l>r)
        return 0;
    if (l>=u&&r<=v)
        return st[node];
    int mid=(l+r)>>1;
    return get(node<<1,l,mid,u,v)+get(node<<1|1,mid+1,r,u,v);
}
int get(int node, int l, int r, int x){
    if (l==r)
        return l;
    int mid=(l+r)>>1;
    return (st[node<<1]>=x?get(node<<1,l,mid,x):get(node<<1|1,mid+1,r,x-st[node<<1]));
}
void query(int i){
    if (a[i].empty())
        a[i]=ask(i);
}
void update(int i){
    i++;
    for (;i<=200000;i+=i&(-i))
        f[i]++;
}
int get(int i){
    i++;
    int res=0;
    for (;i;i-=i&(-i))
        res+=f[i];
    return res;
}
int find_best(int n){
    if (n<=5000)
        for (int i=0;i<n;i++){
            query(i);
            if (!a[i][0]&&!a[i][1])
                return i;
        }
    int mx=0;
    for (int i=0;i<472;i++){
        query(i);
        if (!a[i][0]&&!a[i][1])
            return i;
        mx=max(mx,a[i][0]+a[i][1]);
    }
    build(1,0,n-1);
    int cur=0;
    priority_queue <int, vector <int>, greater <int>> q;
    while (true){
        while (!q.empty()){
            if (a[q.top()][0]==get(q.top())){
                cur=q.top()+1;
              	q.pop();
            }
            else
                break;
        }
        int l=1,r=get(1,0,n-1,cur,(q.empty()?n-1:q.top()-1)),val=get(1,0,n-1,0,cur-1);
        vector <int> ve;
        while (l<=r){
            int mid=(l+r)>>1;
            int pos=get(1,0,n-1,val+mid);
            query(pos);
            ve.push_back(pos);
            if (!a[pos][0]&&!a[pos][1])
                return pos;
            if (a[pos][0]+a[pos][1]==mx){
                if (a[pos][0]>get(pos)){
                    if (!ch[pos])
                        q.push(pos);
                    ch[pos]=1;
                    r=mid-1;
                }
                else{
                    cur=pos+1;
                    break;
                }
                continue;
            }
            update(pos);
            break;
        }
        for (int i:ve)
            update(1,0,n-1,i);
    }
}

• Subtask 1: 20 / 20
• Subtask 2: 77.7 / 80