CF576E 筆記

前置知識：線段樹分治。

題意

給定一個 \(n\) 個點 \(m\) 個點的圖，有 \(k\) 種顏色，初始時每條邊都沒有顏色。

有 \(q\) 組詢問，每組詢問 \((i, c)\) 表示把第 \(i\) 條邊顏色改成 \(c\)，然後判斷：所有顏色為 \(c\) 的邊拉出來是否是一個二分圖。

如果是就輸出 YES；否則輸出 NO，並撤銷該操作。

\(n, m \le 5 \times 10^5, k \le 50\)

題解

考慮線段樹分治。但是線上段樹分治時，會有一個問題：無法確定一條邊的出現時間。

思考一下線段樹分治的性質，可以發現：當線上段樹上遞迴到葉子結點，並處理當前葉子的詢問時，前面的詢問都被處理過了。就是說，遍歷線段樹，葉節點的詢問一定是 \(1, 2, \cdots, q\) 的順序被遍歷。

於是，考慮同一條邊被改的相鄰的時間節點 \(x, y\)，這條邊會影響到的，只有 \([x + 1, y - 1]\) 中的詢問。而遞迴到結點 \([l, r]\) ( \(x < l \le r < y\) ) 時，因為這個時候拿到的邊的顏色是處理了 \([1, l - 1]\) 過後的，所以我們直接連邊即可。總複雜度 \(O(n \log^2 n)\)。

程式碼

code:

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
const int N = 1E6 + 5, K = 51;
int n, m, q, k, a[N], c[N], pre[N], nc[N];
pair <int, int> e[N];
stack <pair <pair <int, int>, int>> st;
struct bcj {
  int fa[N], sz[N];
  void init() {
    for (int i = 1; i <= 2 * n; ++i)
      fa[i] = i, sz[i] = 1;
  }
  int find(int x) {return x == fa[x] ? x : find(fa[x]);}
  void merge(int x, int y, int col) {
    x = find(x), y = find(y);
    if (x == y) return ;
    if (sz[x] > sz[y]) swap(x, y);
    st.emplace(make_pair(make_pair(x, y), col));
    fa[x] = y; sz[y] += sz[x];
  }
  void mer(int id, int col) {
    auto [x, y] = e[id];
    merge(x + n, y, col);
    merge(x, y + n, col);
  }
  void cut(int x, int y) {fa[x] = x; sz[y] -= sz[x];}
  bool same(int x, int y) {return find(x) == find(y);}
} f[K];
void cl(int to) {
  while ((int)st.size() > to) {
    auto [x, y] = st.top().first; int c = st.top().second; st.pop();
    f[c].cut(x, y);
  }
}
vector <int> v[N << 1];
void ins(int x, int l, int r, int L, int R, int p) {
  if (l >= L && r <= R) {
    v[x].emplace_back(p);
    return ;
  } int mid = (l + r) >> 1;
  if (L <= mid) ins(x << 1, l, mid, L, R, p);
  if (R > mid) ins(x << 1 | 1, mid + 1, r, L, R, p);
}
int wtf = 0;
void solve(int x, int l, int r) {
  int to = st.size();
  for (auto id : v[x]) {
    if (nc[id]) f[nc[id]].mer(id, nc[id]);
  }
  int mid = (l + r) >> 1;
  if (l == r) {
    int id = a[l]; auto [u, v] = e[id];
    if (f[c[l]].same(u, v)) cout << "NO\n";
    else cout << "YES\n", nc[id] = c[l];
  } else solve(x << 1, l, mid), solve(x << 1 | 1, mid + 1, r);
  cl(to);
}
signed main(void) {
  ios :: sync_with_stdio(false);
  cin.tie(nullptr); cout.tie(nullptr);
  cin >> n >> m >> k >> q;
  for (int i = 0; i <= k; ++i)
    f[i].init();
  for (int i = 1; i <= m; ++i)
    cin >> e[i].first >> e[i].second;
  for (int i = 1; i <= q; ++i) 
    cin >> a[i] >> c[i];
  for (int i = 1; i <= q; ++i) {
    if (pre[a[i]] + 1 <= i - 1)
      ins(1, 1, q, pre[a[i]] + 1, i - 1, a[i]);
    pre[a[i]] = i;
  } 
  for (int i = 1; i <= m; ++i)
    if (pre[i] < q) ins(1, 1, q, pre[i] + 1, q, i);
  solve(1, 1, q);
}