E - Mex

3986 ワード

http://acm.sdut.edu.cn:8080/vjudge/contest/view.action?cid=216#problem/E
また線分樹神題です.
非負のシーケンスについてmex(i,j)を区間[i,j]内に最小の正の整数が現れないように定義した.(1<=i<=j<=n)を満たすすべてのmex(i,j)の和を求める.
シーケンス1 2 0 4 6 3 5 7 1 4 8を例にとると,まずi=1のmex(1,j),i=2のmex(2,j)を
i = 1     0 0 3 3 3 5 7 8 8 8 8 8 9
i = 2        0 1 1 1 1 1 1 8 8 8 8 9
mex(2,j)とmex(1,j)との関係は,1番目の1(a[1])と2番目の1(a[9])との間のmex(1,j)>1(a[1])の数がmex(2,j)ではいずれも1となり,残りのmex(2,j)は変わらないことがわかる.したがって,1(a[1])になる区間lとrを見つけ,この区間の値和を更新すればよい.
また、すべての固定iに対してmex(i,j)がインクリメントされていることが判明し、この性質に基づいて区間の左端点l、すなわち、最初のmex値がa[1]より大きい位置を見つけることができ、rは、各位置の数を記録するnext配列を設定することができ、r=next[i]−1である.
したがって,mex(1,j)を初期化してから,以降のmex(i,j)を順次求め,区間の和最大値をセグメントツリーで維持する.
#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
//#define LL long long
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 200010;

struct node
{
    int l,r;
    int lazy; //lazy             		
    LL sum; //    
    LL mx;//      
}tree[maxn*4];

int a[maxn];
int next[maxn];
int mex[maxn];
map<int,int>M;

void push_up(int v)
{
    if(tree[v].l == tree[v].r) return;
    tree[v].mx = max(tree[v*2].mx,tree[v*2+1].mx);
    tree[v].sum = tree[v*2].sum + tree[v*2+1].sum;
}

void push_down(int v)
{
    if(tree[v].l == tree[v].r || tree[v].lazy == 0)
        return;
    tree[v*2].lazy = tree[v*2+1].lazy = 1;
    tree[v*2].mx = tree[v*2+1].mx = tree[v].mx;
    tree[v*2].sum = tree[v].mx * (tree[v*2].r - tree[v*2].l + 1);
    tree[v*2+1].sum = tree[v].mx * (tree[v*2+1].r - tree[v*2+1].l + 1);
    tree[v].lazy = 0;
}

void build(int v, int l, int r)
{
    tree[v].l = l;
    tree[v].r = r;
    tree[v].lazy = 0;
    if(l == r)
    {
        tree[v].sum = mex[tree[v].l];
        tree[v].mx = mex[tree[v].l];
        return;
    }
    int mid = (l+r)>>1;
    build(v*2,l,mid);
    build(v*2+1,mid+1,r);
    push_up(v);
}

//  mex      a[i]   
int get(int v, int key)
{
    if(tree[v].l == tree[v].r)
        return tree[v].l;
    push_down(v);
    if(key < tree[v*2].mx)
        return get(v*2,key);
    else return get(v*2+1,key);
}

void update(int v, int l, int r, int key)
{
    if(tree[v].l == l && tree[v].r == r)
    {
        tree[v].mx = key;
        tree[v].lazy = 1;
        tree[v].sum = key * (tree[v].r - tree[v].l + 1);
        return;
    }
    push_down(v);
    int mid = (tree[v].l + tree[v].r) >> 1;
    if(r <= mid)
        update(v*2,l,r,key);
    else if(l > mid)
        update(v*2+1,l,r,key);
    else
    {
        update(v*2,l,mid,key);
        update(v*2+1,mid+1,r,key);
    }
    push_up(v);
}

int main()
{
    int n,Min;
    while(~scanf("%d",&n)&&n)
    {
        for(int i = 1; i <= n; i++)
            scanf("%d",&a[i]);
        M.clear();
        Min = 0;
        for(int i = 1; i <= n; i++)
        {
            M[a[i]] = 1;
            while(M.find(Min) != M.end())
                Min++;
            mex[i] = Min;
        }

        M.clear();
        for(int i = n; i >= 1; i--)
        {
            if(M.find(a[i]) == M.end())
                next[i] = n+1; //next[i]   next[a[i]]
            else next[i] = M[a[i]];
            M[a[i]] = i;
        }

        build(1,1,n);
        LL sum = 0;
        for(int i = 1; i <= n; i++)
        {
            sum += tree[1].sum;
            if(tree[1].mx > a[i])
            {
                int l = get(1,a[i]);
                int r = next[i];
                if(l < r)
                    update(1,l,r-1,a[i]);
            }
            update(1,i,i,0);
        }
        printf("%I64d
",sum); } return 0; }