[模拟][线段树]Codeforces Round #424. B Cards Sorting

直接按照题意模拟

找最小值用线段树找就可以了

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <cmath>
#define fi first
#define se second
#define pb push_back

using namespace std;

typedef long long ll;
typedef pair<int,int> par;

inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void rea(int &x){
    char c=nc(); x=0; int f=1;
    for(;c>'9'||c<'0';c=nc())if(c=='-') f=-1;
    for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc()); x*=f;
}

const int N=300010;

int n,a[N],mn[N],p[N],tot[N];

inline void Up(int g){
    int ls=g<<1,rs=g<<1|1;
    if(mn[ls]<=mn[rs]) mn[g]=mn[ls],p[g]=p[ls];
    else mn[g]=mn[rs],p[g]=p[rs];
    tot[g]=tot[ls]+tot[rs];
}

void Build(int g,int l,int r){
    if(l==r){
        mn[g]=a[l]; p[g]=l; tot[g]=1;
        return ;
    }
    int mid=l+r>>1;
    Build(g<<1,l,mid); Build(g<<1|1,mid+1,r);
    Up(g);
}

par Query(int g,int l,int r,int L,int R){
    if(l==L&&r==R) return par(mn[g],p[g]);
    int mid=L+R>>1;
    if(r<=mid) return Query(g<<1,l,r,L,mid);
    else if(l>mid) return Query(g<<1|1,l,r,mid+1,R);
    else return min(Query(g<<1,l,mid,L,mid),Query(g<<1|1,mid+1,r,mid+1,R));
}

void Erase(int g,int x,int l,int r){
    if(l==r){
        mn[g]=1<<30; tot[g]=0; return ;
    }
    int mid=l+r>>1;
    if(x<=mid) Erase(g<<1,x,l,mid);
    else Erase(g<<1|1,x,mid+1,r);
    Up(g);
}

int Check(int g,int l,int r,int L,int R){
    if(l==L&&r==R) return tot[g];
    int mid=L+R>>1;
    if(r<=mid) return Check(g<<1,l,r,L,mid);
    if(l>mid) return Check(g<<1|1,l,r,mid+1,R);
    return Check(g<<1,l,mid,L,mid)+Check(g<<1|1,mid+1,r,mid+1,R);
}

int main(){
    rea(n);
    for(int i=1;i<=n;i++) rea(a[i]);
    Build(1,1,n);
    int cur=0; ll ans=0;
    par fir=Query(1,1,n,1,n);
    ans=fir.se; cur=fir.se;
    Erase(1,fir.se,1,n);
    for(int i=2;i<=n;i++){
        par Left=par(1<<30,0),Right=par(1<<30,0);
        if(cur>1) Left=Query(1,1,cur-1,1,n);
        if(cur<n) Right=Query(1,cur+1,n,1,n);
        if(Left.fi<Right.fi){
            ans+=Check(1,cur,n,1,n)+Check(1,1,Left.se,1,n);
            cur=Left.se;
            Erase(1,cur,1,n);
        }
        else{
            ans+=Check(1,cur,Right.se,1,n);
            cur=Right.se;
            Erase(1,cur,1,n);
        }
    }
    printf("%lld\n",ans);
    return 0;
}