AtCoder Beginner Contest 217 D - Cutting Woods (set+lower_bound)

Posted 2021-10-17 karshey

D

set+lower_bound;
注意：
s.lower_bound(x)返回的是迭代器，想要输出迭代器指向的值和它的前一位指向的值应该是：- -t而不是t- -；

auto t=s.lower_bound(x);
cout<<*t-*(--t);

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 

//======================
const int N=2e5+10;
int a[N];
int main()
{
	set<int>s;
	int n,q;
	cin>>n>>q;
	s.insert(0),s.insert(n);
	for(int i=0;i<q;i++)
	{
		int a,b;cin>>a>>b;
		if(a==1)
		{
			s.insert(b);
		}
		else
		{
			auto t=s.lower_bound(b);			
			int ans=*t-*(--t);
			cout<<ans<<endl;
		}
	}
	return 0; 
}