hdu 4747 Mex( 线段树？ 不，区间处理就行（dp？））

Posted 2020-09-11 hk lin

Mex

Time Limit: 15000/5000 MS (Java/Others)    Memory Limit: 65535/65535 K (Java/Others)
Total Submission(s): 3056    Accepted Submission(s): 1006

Problem Description

Mex is a function on a set of integers, which is universally used for impartial game theorem. For a non-negative integer set S, mex(S) is defined as the least non-negative integer which is not appeared in S. Now our problem is about mex function on a sequence.

Consider a sequence of non-negative integers {ai}, we define mex(L,R) as the least non-negative integer which is not appeared in the continuous subsequence from aL to aR, inclusive. Now we want to calculate the sum of mex(L,R) for all 1 <= L <= R <= n.

 

 

Input

The input contains at most 20 test cases.
For each test case, the first line contains one integer n, denoting the length of sequence.
The next line contains n non-integers separated by space, denoting the sequence.
(1 <= n <= 200000, 0 <= ai <= 10^9)
The input ends with n = 0.

 

 

Output

For each test case, output one line containing a integer denoting the answer.

 

 

Sample Input

3 0 1 3 5 1 0 2 0 1 0

 

 

Sample Output

5 24

Hint

For the first test case: mex(1,1)=1, mex(1,2)=2, mex(1,3)=2, mex(2,2)=0, mex(2,3)=0,mex(3,3)=0. 1 + 2 + 2 + 0 +0 +0 = 5.

 

 

Source

2013 ACM/ICPC Asia Regional Hangzhou Online

 

 

Recommend

liuyiding   |   We have carefully selected several similar problems for you:  6022 6021 6020 6019 6018 

---

 

 

 

题意：Mex(l,r) 求的是[l,r]区间内所有数的集合里没有出现过的最小的数字，即博弈里的mex。现在给出一个序列[1,n]，求解所有$1 \le l \le r\le n$ 的[l,r]中的mex[l,r]的和。

 

 

 

  首先我们将区间右端点统一为k，用num[k]存k这个位置对应的数字。即我们处理到k这个端点时，我们处理的是所有$[l,k] （1 \le l \le k）$的mex。此时我们只需要for一遍然后对应处理k就行。

  那怎么加速k这个端点的处理呢？ 我们用t[l] 表示[l,k] 中所有数的集合set从零开始连续的最大的数，即mex-1，那么假如我们的序列为6 0 3 2 0 1 0 ，处理到最后一个数k=7，对应的t为 3 3 3 2 1 1 0,可以看出t为一个非递减序列,因此我们用aft[o]存t中数字o连续序列的最后一个位置，例如aft[1]=6。

  我们可以将这个问题化为区间右端点扩张的问题，每次更新对应一些区间的右端点扩张。我们还用last[o]存数字o最后出现的位置。可以看出我们每到一个新端点k，它对应的数字为knum，影响的t区间为大于等于knum的数字所在的区间，准确的说是s大于等于knum并且aft[s]==aft[knum]的数字所在的区间，因为一旦aft[s]<aft[knum]那么限制他的右端点扩张的数字就不是knum了，而是大于knum的数字。所以我们需要更新这些数字所在区间右端点，并且从小到大更新，设更新的区间右端点最大值为maxn。由于这些数字区间右端点还受aft[knum-1]限制，初始maxn=aft[knum-1]。然后每个数字s的右端点最大值为min（aft[knum-1],last[knum]，last[knum+1]......last[s]）（最左端的端点限制扩张）。至此我们快速的更新t区间。另外我们用all存现在l∈[1,k] 所有[l,k]mex的和，那么每次更新右端点就需要给all加上更新的区间的长度作为扩张对mex的贡献。最后我们把所有的all加起来就是答案。

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<algorithm>
 5 #define clr(x) memset(x,0,sizeof(x))
 6 #define LL long long
 7 using namespace std;
 8 int a[200010],aft[200010],last[200010];
 9 LL ans,all;
10 int min(int a,int b)
11 {
12     return a<b?a:b;
13 }
14 int main() 
15 {
16     int n,m,k,dk,maxn;
17     while(scanf("%d",&n)==1 && n!=0)
18     {
19         ans=0;
20         all=0;
21         clr(last);
22         clr(aft);
23         for(int i=1; i<=n; i++)
24         {
25             scanf("%d",&a[i]);
26             if(a[i]>=n)
27             {
28                 ans+=all;
29                 continue;
30             }
31             last[a[i]]=i;
32             if(a[i]>0)
33             {
34                 maxn=aft[a[i]-1];
35             }
36             else
37             {
38                 maxn=i;
39             }
40             k=a[i];
41             dk=aft[k];
42             while(aft[k]==aft[k+1])
43             {
44                 maxn=min(maxn,last[k]);
45                 if(maxn==dk)
46                     break;
47                 all+=(LL)(maxn-aft[k]);
48                 aft[k]=maxn;
49                 k++;
50             }
51             maxn=min(maxn,last[k]);
52             all+=(LL)(maxn-aft[k]);
53             aft[k]=maxn;
54             ans+=all;
55         }
56         printf("%lld\n",ans);
57     }
58     return 0;
59 }