codeforces830B Cards Sorting -- 线段树

来源:互联网 发布:sha1 js 时间 编辑:程序博客网 时间:2024/06/01 09:04

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

代码:

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;#define N 100010#define ll long longstruct 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;}
原创粉丝点击