AtCoder Beginner Contest 217 D - Cutting Woods（set + 二分查找）

题目大意

有一根长度为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;
}