Codeforces Round #830 (Div. 2) D1. Balance (Easy version)(暴力+优化)

题目大意:有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;
 } 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值