codeforces830B Cards Sorting -- 线段树

最新推荐文章于 2021-10-21 20:32:52 发布

gjghfd

最新推荐文章于 2021-10-21 20:32:52 发布

阅读量498

点赞数

分类专栏：线段树文章标签： codeforces

本文链接：https://blog.csdn.net/gjghfd/article/details/75314922

版权

线段树专栏收录该内容

33 篇文章 0 订阅

订阅专栏

用线段树记录区间有几个数还没被取走，将所有数从小到大排序后依次更新答案就可以了。注意相等的时候取的数是在上次取的数后面的第一个数。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
#define ll long long
struct Node{
    int w,f;
}a[N];
int i,j,k,p,n,m,c[N<<2],Last;
ll Ans;
inline bool Cmp(Node a,Node b){
    if(a.w==b.w)return a.f<b.f;
    return a.w<b.w;
}
inline void Build(int x,int l,int r){
    c[x]=r-l+1;
    if(l==r)return;
    int Mid=l+r>>1;
    Build(x<<1,l,Mid);
    Build(x<<1|1,Mid+1,r);
}
inline int Query(int x,int l,int r,int L,int R){
    if(l>R||r<L)return 0;
    if(l>=L&&r<=R)return c[x];
    int Mid=l+r>>1;
    return Query(x<<1,l,Mid,L,R)+Query(x<<1|1,Mid+1,r,L,R);
}
inline void Update(int x,int l,int r,int y){
    c[x]--;
    if(l==r)return;
    int Mid=l+r>>1;
    if(Mid>=y)Update(x<<1,l,Mid,y);else Update(x<<1|1,Mid+1,r,y);
}
inline int Get(int x,int y){
    if(x<=y)return Query(1,1,n,x,y);
    return Query(1,1,n,1,y)+Query(1,1,n,x,n);
}
inline void Calc(int y){
    Ans+=Get(Last,y);
    Update(1,1,n,y);
    Last=y;
}
int main(){
    scanf("%d",&n);
    for(i=1;i<=n;i++)scanf("%d",&a[i].w),a[i].f=i;
    sort(a+1,a+n+1,Cmp);
    Build(1,1,n);
    for(i=1;i<=n;i=j+1){
        k=0;
        for(j=i;j<n&&a[j+1].w==a[j].w;j++)
        if(a[j].f>Last&&!k)k=j;
        if(a[j].f>Last&&!k)k=j;
        if(k){
            for(p=k;p<=j;p++)Calc(a[p].f);
            for(p=i;p<k;p++)Calc(a[p].f);
        }else for(p=i;p<=j;p++)Calc(a[p].f);
    }
    printf("%I64d\n",Ans);
    return 0;
}