Problem 1 bfs+set

$des$

小G有一个长度为 $n$ 的 01 串 T ，其中只有 $T_S = 1$，其余位置都是 $0$。现在小G可以进行若干
次以下操作：
选择一个长度为 $K$ 的连续子串（K是给定的常数），翻转这个子串。
对于每个 $i,i ∈ [1,n]$，小G想知道最少要进行多少次操作使得 $T_i = 1$. 特别的，有 $m$ 个 “禁
止位置”，你需要保证在操作过程中 $1$ 始终不在任何一个禁止位置上。

$sol$

考虑当前在每个点时可以通过一次翻转转移到哪些点，直接遂遆道即可算出每个点的所需步
数。然而边数会达到 $O(n ^ 2)$ 级别。
可以发现转移到的点一定是一段区间内的奇数或者偶数点，于是一种简单的优化
方法是在 BFS 时开两个 SET 维护当前有哪些奇数点和偶数点还未被 BFS 到，转移时直接
在 SET 上 lower_bound，就不会访问已经 BFS 到过的点了。$O(nlogn)$  

$code$

#include <bits/stdc++.h>

#define Rep(i, j, k) for (int i = j; i <= k; i++)

using namespace std;

const int N = 1e5 + 10;

int n, K, m, S;
int dis[N];
bool ban[N];
set<int> add1, even2;

void BFS() {
    memset(dis, -1, sizeof dis);
    queue<int> q;
    dis[S] = 0, q.push(S);
    while (!q.empty()) {
        int o = q.front(); q.pop();
        int L = max(1, o - K + 1), R = min(n, o + K - 1);
        L = L + (L + K - 1) - o, R = R + (R - K + 1) - o;
//        cout << o << " " << L << " " << R << "
"; 
        set<int> &p = L & 1 ? add1 : even2;
        for (set<int> :: iterator i = p.lower_bound(L); i != p.end() && *i <= R; p.erase(i++))
            dis[*i] = dis[o] + 1, q.push(*i);
    }
}

int main() {
    scanf("%d%d%d%d", &n, &K, &m, &S);
    Rep(i, 1, m) {
        int x;
        scanf("%d", &x);
        ban[x] = true;
    }
    Rep(i, 1, n) if (!ban[i] && i != S) i & 1 ? add1.insert(i) : even2.insert(i);
    BFS();
    Rep(i, 1, n) printf("%d ", dis[i]);
    return 0;
}