题目大意
有一根长度为L的木头,这根木头上有L - 1个可以劈的点,对这根木头进行q次操作,操作有两种:操作一是往一个可劈点劈一刀,把这个点所在的木头段劈成两段;操作二是输出一个可劈点所在木头段的长度。
思路
直接用一个set存操作一所劈的点,因为set保证了所劈的点是递增的,所以对于操作二,直接用lower_bound找到第一个比要找的可劈点大的已劈点,然后lower_bound找到的位置后退一位就是第一个比要找的可劈点小的已劈点,两者相减就是该可劈点所在木头段的长度。
因为存在set中第一个已劈点比操作二的可劈点大或者set中最后一个已劈点比操作二的可劈点小的情况,所以一开始直接把0和木头长度L放到set中,这样就少了很多判断。
心得
比赛的时候一直TLE,但是方法就是set + lower_bound,最后比赛完看了一下tourist的代码,发现set本身就自带lower_bound函数,而我并不知道,直接用了algorithm库里的lower_bound,差不多一个半小时都是TLE…
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
set<int> s;
int main()
{
int len, q;
scanf("%d%d", &len, &q);
set<int>:: iterator it;
s.insert(0);
s.insert(len);
while(q--)
{
int op;
scanf("%d", &op);
if(op == 1)
{
int a;
scanf("%d", &a);
s.insert(a);
}
else
{
int a;
scanf("%d", &a);
it = s.lower_bound(a);
// 比赛时我的错代码...
// it = lower_bound(s.begin(), s.end(), a);
int r = *it;
it--;
int l = *it;
printf("%d\n", r - l);
}
}
return 0;
}