[HDU 4747] Mex （线段树）

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4747

这道题是我去年刚入校队的时候参加网赛的题目。

一年过去了，我依然还是不会做。。

这是我难题计划的开始吧。。

竟然还敲挫了，自己真是弱。

题意：

给你一个数列a，定义Mex[L,R]为a[L,R]中没有出现过的最小的自然数。求1<=l<=r<=n的所有Mex[l,r]之和。

解法我也是看了解题报告才知道的。

请参看cxlove大神的博文：http://blog.csdn.net/acm_cxlove/article/details/11749383

代码：

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <set>
 using namespace std;

 typedef long long LL;
 typedef pair<int,int> PII;
 const int MAX_N = 2e5+;
 LL delta[MAX_N<<],sum[MAX_N<<],maxn[MAX_N<<];
 PII b[MAX_N];
 int a[MAX_N],next[MAX_N],n;
 set<int> Set;

 void push_down(int idx,int l,int r){
     if( delta[idx]!=- ){
         int m = l+r>>;
         sum[idx<<] = (m-l+)*delta[idx];
         sum[idx<<|] = (r-m)*delta[idx];
         delta[idx<<] = delta[idx<<|] = maxn[idx<<] = maxn[idx<<|] = delta[idx];
         delta[idx] = -;
     }
 }

 void push_up(int idx){
     sum[idx] = sum[idx<<]+sum[idx<<|];
     maxn[idx] = max(maxn[idx<<],maxn[idx<<|]);
 }

 void update(int L,int R,LL x,int idx,int l,int r){
     if( R<l||L>r ) return;
     if( L<=l&&R>=r ) {
         sum[idx]  = (r-l+)*x;
         maxn[idx] = delta[idx] = x;
         return;
     }
     push_down(idx,l,r);
     int m = l+r>>;
     if( L<=m ) update(L,R,x,idx<<,l,m);
     if( R>m ) update(L,R,x,idx<<|,m+,r);
     push_up(idx);
 }

 int query(LL x,int idx,int l,int r){
     if( maxn[idx]<=x ) return r+;
     if( l==r ) return l;
     push_down(idx,l,r);
     int m = l+r>>;
     if( maxn[idx<<]>x ) return query(x,idx<<,l,m);
     else return query(x,idx<<|,m+,r);
 }

 LL querySum(int L,int R,int idx,int l,int r){
     if( L<=l&&R>=r ) return sum[idx];
     push_down(idx,l,r);
     int m = l+r >>  ;
     LL res = ;
     if( L<=m ) res =  querySum(L,R,idx<<,l,m);
     if( R>m ) res += querySum(L,R,idx<<|,m+,r);
     return res;
 }

 int main(){
     while(scanf("%d",&n)!=EOF&&n){
         memset(sum,,sizeof(sum));
         memset(delta,-,sizeof(delta));
         Set.clear();
         for(int i=;i<=n;i++){
             scanf("%d",&a[i]);
             b[i] = PII(a[i],i);
             next[i] = n+;
         }
         sort(b+,b++n);
         b[n+].first = b[n].first; b[n+].second = n+;
         for(int i=;i<=n;i++){
             if( b[i].first==b[i+].first ) next[b[i].second] = b[i+].second;
         }
         int mmex = ;
         for(int i=;i<=n;i++){
             Set.insert(a[i]);
             while( Set.find(mmex)!=Set.end() ) mmex++;
             update(i,i,mmex,,,n);
         }
         LL ans = sum[];
         for (int l = ; l <= n; l++) {
             update(l,l,,,,n);
             int p = query(a[l],,,n);
             p = max(l+,p);
             int q = next[l] - ;
             update(p,q,a[l],,,n);
             ans += sum[];
         }
         printf("%I64d\n", ans);
     }
     return ;
 }