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