Bzoj1018/洛谷P4246 [SHOI2008]堵塞的交通（線段樹分治+並查集）

題面

Bzoj

洛谷

題解

考慮用並查集維護圖的連通性，接着用線段樹分治對每一個修改進行分治。

具體來講，就是用一個時間軸表示圖的狀態，用線段樹維護，對於一條邊，咱們判斷若是他的存在時間正好在這個區間內，那就把它用並查集並起來。最後對於一個詢問，直接用並查集找就行了。

可是由於有撤銷操做，因此在並查集合並的時候，咱們將須要合併的兩個點放進棧中，最後棧序撤銷，因此只能考慮按秩合併而不能路徑壓縮。

#include <map>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
using std::min; using std::max;
using std::sort; using std::swap;
using std::unique; using std::lower_bound;
using std::map; using std::vector;
typedef long long ll;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar(); x *= flag;
}

const int N = 2e5 + 10;
int n, id[3][N], q_num, top, fa[N], siz[N], cnt;
map<int, int> G[N];
struct Edge { int u, v, beg, end; };
struct Node { int x, y; } q[N], stk[N << 2];
vector<Edge> e;

int find(int x) { while(fa[x] != x) x = fa[x]; return x; }

inline void merge(int x, int y) {
    int fx = find(x), fy = find(y);
    if(siz[fx] > siz[fy]) swap(fx, fy);
    fa[fx] = fy, siz[fy] += siz[fx], stk[++top] = (Node){fx, fy}; 
}

void doit (int l, int r, vector<Edge> E) {
    vector<Edge> L, R;
    int mid = (l + r) >> 1, tmp = top;
    for(vector<Edge>::iterator it = E.begin(); it != E.end(); ++it)
        if(it->beg <= l && it->end >= r) merge(it->u, it->v);
        else {
            if(it->beg <= mid) L.push_back(*it);
            if(it->end > mid) R.push_back(*it);
        }
    if(l == r) puts(find(q[l].x) == find(q[l].y) ? "Y" : "N");
    else doit(l, mid, L), doit(mid + 1, r, R);
    while(top > tmp) {
        int x = stk[top].x, y = stk[top--].y;
        fa[x] = x, siz[y] -= siz[x];
    }
}

int main () {
    read(n);
    for(int i = 1; i <= n; ++i)
        id[1][i] = ++cnt, id[2][i] = ++cnt;
    while(true) {
        char s[10]; int r1, c1, r2, c2;
        scanf("%s", s); if(s[0] == 'E') break;
        read(r1), read(c1), read(r2), read(c2);
        if(s[0] == 'O') {
            G[id[r1][c1]][id[r2][c2]] = G[id[r2][c2]][id[r1][c1]] = e.size();
            e.push_back((Edge){id[r1][c1], id[r2][c2], q_num + 1, -1});
        } else if(s[0] == 'C') e[G[id[r1][c1]][id[r2][c2]]].end = q_num;
        else if(s[0] == 'A') q[++q_num] = (Node){id[r1][c1], id[r2][c2]};
    }
    for(vector<Edge>::iterator it = e.begin(); it != e.end(); ++it)
        if(it->end == -1) it->end = q_num;
    for(int i = 1; i <= n + n; ++i) fa[i] = i, siz[i] = 1;
    doit(1, q_num, e);
    return 0;
}