题目大意:有q组询问,询问有两种,一种是以’+‘开头的,意思是将后面的数字加入到集合中,一种是以’?‘开头的,后面接一个数字k,表示询问在集合中未出现过的数中最小的能被k整除的数是多少?,对于每一个这种询问要输出一个答案。
解题思路:对于这个问题我们首先想到的是暴力做法,就是对于每一个?类型的询问,我们每次从k出发,每次加k,查询当前的数字是否在集合中出现过,如果没有出现过就直接输出,否则就继续加k,直到出现一个没有出现在集合中的数字为止。但显然这样会超时,接下来就是要对暴力的做法进行优化,优化方法如下:对于每一个重复查询的数字x,记录上一次查询的结果,然后当前再查询的时候,从上一次查询的结果开始进行新一轮的查询。这样看似好像也没有优化多少,但是如果仔细算一下,会惊奇地发现,这样的优化会使得原本O(N^2)的暴力变成O(NLOG(N))的做法。
估计复杂度的方法如下:对于优化后的做法,最坏的情况是这样的,当前插入了n个数字,分别是x,2x,3x,……,nx,现在要询问未出现在集合中最小的x,2x,3x……,nx的倍数,那么查询次数分别为n,n/2,n/3,……,n/n,也就是n+n/2+n/3+⋯+n/n≈nlogn。
上代码:
#include <iostream>
#include <cstring>
#include <bits/stdc++.h>
using namespace std;
map<long long ,long long>mp,np;
int main()
{
long long q;
cin>>q;
while(q--)
{
getchar();
char op;
cin>>op;
long long x;
cin>>x;
// mp.clear();
if(op=='+')
mp[x]=1;
else
{
if(!mp[x])
{
cout<<x<<endl;
}
else
{
if(np[x]==0)
{
long long temp=x;
while(1)
{
if(mp[temp]==0)
{
cout<<temp<<endl;
np[x]=temp;
break;
}
else
temp+=x;
}
}
else
{
long long temp=np[x];
while(1)
{
if(mp[temp]==0)
{
cout<<temp<<endl;
np[x]=temp;
break;
}
else
temp+=x;
}
}
}
}
}
return 0;
}