cf round #424 div2 E. Cards Sorting(线段树)

题目链接：http://codeforces.com/contest/831/problem/E

分析：题意很简单，一开始想写个树状数组然后找逆序数，发现找下一个并不好处理。。还是用线段树，直接找最小值为下一个，然后再把这个最小值改成无穷大，直接往下推就可以了。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+5,inf=1e5+5;
int n;
class TreeArray{
    int c[maxn],n;
public:
    void Init(int n){
        this->n=n;
        memset(c,0,sizeof(c));
    }
    void add(int k,int num){
        while(k<=n){
            c[k]+=num;
            k+=k&-k;
        }
    }
    int read(int k){
        int sum=0;
        while(k){
            sum+=c[k];
            k-=k&-k;
        }
        return sum;
    }
    int read(int l,int r){
        return read(r)-read(l-1);
    }
}ta;
class segTree{
public:
    int a[maxn],s[maxn*4];
    void build(int node,int begin,int end){
        if(begin==end){
            s[node]=begin;
        }else{
            build(2*node,begin,(begin+end)/2);
            build(2*node+1,(begin+end)/2+1,end);
            if(a[s[2*node]]<=a[s[2*node+1]])
                s[node]=s[2*node];
            else
                s[node]=s[2*node+1];
        }
    }
    void Updata(int node,int begin,int end,int idx,int num){
        if(begin==end){
            a[idx]=num;
            s[node]=begin;return;
        }
        int m=(begin+end)/2;
        if(idx<=m)
            Updata(2*node,begin,m,idx,num);
        else
            Updata(2*node+1,m+1,end,idx,num);
        if(a[s[2*node]]<=a[s[2*node+1]])
            s[node]=s[2*node];
        else
            s[node]=s[2*node+1];
    }
    int query(int node,int begin,int end,int left,int right){
        if(left>end||right<begin)
            return -1;
        if(begin>=left&&end<=right)
            return s[node];
        int q1,q2;
        q1=query(2*node,begin,(begin+end)/2,left,right);
        q2=query(2*node+1,(begin+end)/2+1,end,left,right);
        if(q1==-1)
            return q2;
        if(q2==-1)
            return q1;
        if(a[q1]<=a[q2])
            return q1;
        return q2;
    }
}st;
int main(){
    scanf("%d",&n);
    ta.Init(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&st.a[i]);
        ta.add(i,1);
    }
    st.build(1,1,n);
    long long ans=0;
    int s,t;
    s=st.query(1,1,n,1,n);
    ans+=s;
    st.Updata(1,1,n,s,inf);
    ta.add(s,-1);
    for(int i=1;i<n;i++){
        int a,b;
        if(s<n){
            a=st.query(1,1,n,s+1,n);
        }
        if(s>1){
            b=st.query(1,1,n,1,s-1);
        }
        if(s==n)
            t=b;
        else if(s==1)
            t=a;
        else if(st.a[a]<=st.a[b])
            t=a;
        else
            t=b;
        if(t>s){
            ans+=ta.read(s,t);
        }else{
            ans+=ta.read(s,n)+ta.read(1,t);
        }
        st.Updata(1,1,n,t,inf);
        ta.add(t,-1);
        s=t;
    }
    cout<<ans<<endl;
    return 0;
}