hdu 4747 Mex

题意:给你N个数,枚举所有的L和R,问由区间[L,R]之间的数构成的集合里,哪一个数字没有出现,如果有多个,找出最早的一个,设其为x,求所以x的和。

如果固定R,设R这个位置的值为Key[R],可以发现整个x数列是递减的,随着R的左移,可以从当前的R到前一个Key[R]出现位置之间的X,都不应该大于R,如果发现了这个性质,就是一个简单的成段更新了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <set>
#include <map>
using namespace std;

typedef long long LL;

#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define MID(a,b) (a+((b-a)>>1))

const int N=2e5+5;

set<int> order;
map<int,int> H;
int a[N],pre[N],pos[N],valu[N];

struct Segtree
{
    LL sum[N*4];
    int lKey[N*4],rKey[N*4],delay[N*4];
    void PushUp(int ind)
    {
        lKey[ind]=lKey[LL(ind)];
        rKey[ind]=rKey[RR(ind)];
        sum[ind]=sum[LL(ind)]+sum[RR(ind)];
    }
    void fun(int valu,int lft,int rht,int ind)
    {
        lKey[ind]=rKey[ind]=delay[ind]=valu;
        sum[ind]=valu*(rht-lft+1);
    }
    void PushDown(int ind,int lft,int rht)
    {
        if(delay[ind]!=-1)
        {
            int mid=MID(lft,rht);
            fun(delay[ind],lft,mid,LL(ind));
            fun(delay[ind],mid+1,rht,RR(ind));
            delay[ind]=-1;
        }
    }
    void build(int lft,int rht,int ind)
    {
        sum[ind]=lKey[ind]=rKey[ind]=0; delay[ind]=-1;

        if(lft==rht) lKey[ind]=rKey[ind]=sum[ind]=valu[lft];
        else
        {
            int mid=MID(lft,rht);
            build(lft,mid,LL(ind));
            build(mid+1,rht,RR(ind));
            PushUp(ind);
        }
    }
    void updata(int st,int ed,int valu,int lft,int rht,int ind)
    {
        if(st<=lft&&rht<=ed) fun(valu,lft,rht,ind);
        else
        {
            PushDown(ind,lft,rht);
            int mid=MID(lft,rht);
            if(st<=mid) updata(st,ed,valu,lft,mid,LL(ind));
            if(ed> mid) updata(st,ed,valu,mid+1,rht,RR(ind));
            PushUp(ind);
        }
    }
    int query(int st,int ed,int valu,int lft,int rht,int ind)
    {
        if(lft==rht)
        {
            if(lKey[ind]>=valu) return lft;
            return -1;
        }
        else
        {
            int mid=MID(lft,rht),pos=-1;
            PushDown(ind,lft,rht);

            if(ed<=mid) pos=query(st,ed,valu,lft,mid,LL(ind));
            else if(st>mid) pos=query(st,ed,valu,mid+1,rht,RR(ind));
            else
            {
                if(lKey[RR(ind)]>=valu) pos=query(st,ed,valu,mid+1,rht,RR(ind));
                else pos=query(st,ed,valu,lft,mid,LL(ind));
            }
            PushUp(ind);
            return pos;
        }
    }
    int getValu(int pos,int lft,int rht,int ind)
    {
        if(lft==rht) return lKey[ind];
        else
        {
            int mid=MID(lft,rht),tmp=0;
            PushDown(ind,lft,rht);
            if(pos<=mid) tmp=getValu(pos,lft,mid,LL(ind));
            else tmp=getValu(pos,mid+1,rht,RR(ind));
            PushUp(ind);
            return tmp;
        }
    }
}seg;
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0) break;

        int sc=0;
        order.clear(); H.clear();
        memset(pre,-1,sizeof(pre));

        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            if(H.find(a[i])==H.end()) { H.insert(make_pair(a[i],sc++)); }
            else { pre[i]=pos[ H[a[i]] ]; }
            pos[H[a[i]]]=i;
        }

        int tmp=0;
        for(int i=n-1;i>=0;i--)
        {
            order.insert(a[i]);
            while(order.find(tmp)!=order.end()) tmp++;
            valu[i]=tmp;
        }

        seg.build(0,n-1,1);

        LL ans=0;

        for(int i=n-1;i>=0;i--)
        {
            ans+=seg.sum[1];
            int st=pre[i]+1,ed=i-1,pos=-1;

            if(st<=ed) pos=seg.query(st,ed,a[i],0,n-1,1);
            if(pos!=-1) seg.updata(st,pos,a[i],0,n-1,1);

            seg.updata(i,n-1,0,0,n-1,1);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值