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;
}
上一篇:AtCoder Beginner Contest 216


下一篇:AtCoder Beginner Contest 216 个人题解