ABC 330 E Mex and Update
题意
给出一个长度为N的序列A,有Q次询问,每次询问输入两个整数i,k,表示将A[i]赋值为x。
每次询问输出序列A的mex。mex是指序列中未出现的最小非负整数。
思路
由于N是小于等于2e5的,那么说明每次询问的mex结果是无论如何都不会超过2e5+1的。我们先用set将1~2e5+1都存起来。然后每当A数组里出现一个数的时候,我们再把这个数从set里去掉(set自带去重和排序)。然后再用一个map记录A数组中A[i]出现了多少次,一但A[i]的出现次数为0,我们就在set里插入A[i],相反,如果A[i]的出现次数从0变化到其它正整数,我们就在set里把A[i]给去掉。每次询问mex就直接输出set的begin即可。
代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=2e5+10;
set<int>ds;
map<int,int>vis;
int a[maxn];
signed main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
for(int i=0;i<=2e5+10;i++) ds.insert(i);
int n,q;
cin>>n>>q;
for(int i=1;i<=n;i++)
{
cin>>a[i];
vis[a[i]]++;
ds.erase(a[i]);
}
//cout<<*ds.begin()<<"qwq"<<endl;
while(q--)
{
int k,x;
cin>>k>>x;
vis[a[k]]--;
if(!vis[a[k]]) ds.insert(a[k]);
a[k]=x;
vis[x]++;
ds.erase(x);
cout<<*ds.begin()<<endl;
}
return 0;
}