HDU 4747 线段树+思维

点击打开链接

题意：给出一个数字序列，定义mex为一段序列中没有出现的最小的自然数，问所有的mex的和

思路：网络赛题目，为何感觉如此的难，对于200000的数列并且有多次的区间和，想是想到要用线段树处理了，但是根本没有接下来的思路了，借鉴神犇的思路神犇的思路来写把，真是不知道人家怎么想的那么一步到位，慢慢学习把~~~，我们首先预处理出来对于第一个数的mex[i],1<=i<=n；然后再将每一个数x的下一个位置Next[i]预处理出来，如果没有下一个位置了，就将下一个位置记为n+1来处理，然后用线段树将mex记录下来，父亲记录区间和，这样num[1]就为每一次的mex的总和，全部加起来就是结果，对于每一次的更新，看例子把，第二组样例，1
0 2 0 1，mex第一次为0 2 3 3 3，然后第一个点去掉后，更新为0 1 1 1 3，然后说一说Next数组的作用，还是这个样例，删掉1后，对于下一个1和下一个1以后的所有mex都不会变，因为没有第一个1，我下一个1也可以代替第一个1的作用，这样后面的所有值都不变，记pos为下一个1之前的位置，变得部分是第二个位置到pos的mex，怎么变呢，找到第一个大于1的mex的位置，也就是第二个位置，到pos的mex更新为1，为什么呢，假如一个位置的mex为2意味着什么，它之前肯定是出现了1，而我们将1删除，他们的值自然都变为1，因为序列的mex肯定是递增的，不用考虑其它情况
   PS：第一次写了这长的思路，因为这想法真的太难想了，真心佩服神犇们现场就可以A出来#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=200010;
long long num[maxn*4];
int num1[maxn*4],A[maxn],max1[maxn*4];//num1为懒惰标记，num为和，max1记录区间最大值
int pre[maxn],Next[maxn],mex[maxn],vis[maxn];//Next为下一个数的位置
void pushup(int node){
num[node]=num[node<<1]+num[node<<1|1];
max1[node]=max(max1[node<<1],max1[node<<1|1]);
}
void pushdown(int node,int m){
if(num1[node]){
num1[node<<1]=num1[node<<1|1]=1;
num[node<<1]=(ll)max1[node]*(m-(m>>1));
num[node<<1|1]=(ll)max1[node]*(m>>1);
max1[node<<1]=max1[node<<1|1]=max1[node];
num1[node]=0;
}
}
void buildtree(int le,int ri,int node){
if(le==ri){
num[node]=mex[le];
max1[node]=mex[le];
return ;
}
int t=(le+ri)>>1;
buildtree(le,t,node<<1);
buildtree(t+1,ri,node<<1|1);
pushup(node);
}
void update(int l,int r,int val,int le,int ri,int node){
if(l<=le&&ri<=r){
num1[node]=1;
num[node]=(long long)val*(ri-le+1);
max1[node]=val;
return ;
}
pushdown(node,ri-le+1);
int t=(le+ri)>>1;
if(l<=t) update(l,r,val,le,t,node<<1);
if(r>t) update(l,r,val,t+1,ri,node<<1|1);
pushup(node);
}//到这里都是成段更新的模版
int querymax(int val,int le,int ri,int node){//找到区间大于val的第一个位置
if(le==ri) return le;
pushdown(node,ri-le+1);
int ans,t=(le+ri)>>1;
if(max1[node<<1]>val) ans=querymax(val,le,t,node<<1);
else ans=querymax(val,t+1,ri,node<<1|1);
pushup(node);
return ans;
}
int main(){
int n;
while(scanf("%d",&n)!=-1){
if(n==0) break;
for(int i=1;i<=n;i++) scanf("%d",&A[i]);
memset(vis,0,sizeof(vis));
memset(num1,0,sizeof(num1));
memset(pre,0,sizeof(pre));
int k=0;
for(int i=1;i<=n;i++){
for(int j=k;j<=n;j++){
if(vis[j]==0){
k=j;break;
}
}
if(A[i]==k){
for(int j=k+1;j<=n;j++){
if(vis[j]==0){
k=j;break;
}
}
mex[i]=k;
}else mex[i]=k;
if(A[i]<=n) vis[A[i]]=1;
}
for(int i=n;i>=1;i--){
if(A[i]>n) Next[i]=n+1;
else if(pre[A[i]]==0){
pre[A[i]]=i;
Next[i]=n+1;
}else{
Next[i]=pre[A[i]];
pre[A[i]]=i;
}
}
buildtree(1,n,1);
long long ans=num[1];
for(int i=1;i<=n;i++){
update(i,i,0,1,n,1);
if(max1[1]>A[i]){
int pos=querymax(A[i],1,n,1);
if(pos<Next[i]) update(pos,Next[i]-1,A[i],1,n,1);
}
ans+=num[1];
}
printf("%I64d\n",ans);
}
return 0;
}