arujbansal

Joker

Apr 13th, 2021 (edited)
133
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.28 KB | None | 0 0
  1. #include <iostream>
  2. #include <algorithm>
  3. #include <vector>
  4. #include <map>
  5. #include <set>
  6. #include <array>
  7. #include <stack>
  8. #include <queue>
  9. #include <random>
  10. #include <numeric>
  11. #include <functional>
  12. #include <chrono>
  13. #include <utility>
  14. #include <iomanip>
  15. #include <assert.h>
  16.  
  17. using namespace std;
  18.  
  19. void dbg_out() { cerr << endl; }
  20. template<typename Head, typename... Tail>
  21. void dbg_out(Head H, Tail... T) { cerr << ' ' << H; dbg_out(T...); }
  22. #define dbg(...) cerr << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
  23.  
  24. #define rng_init mt19937 rng(chrono::steady_clock::now().time_since_epoch().count())
  25. #define rng_seed(x) mt19937 rng(x)
  26. #define all(x) (x).begin(), (x).end()
  27. #define sz(x) (int) (x).size()
  28. // #define int long long
  29.  
  30. const int MXN = 2e5 + 5, INF = 1e9 + 5;
  31. int N, M, Q;
  32. pair<int, int> edges[MXN];
  33. int info[MXN], parity[MXN];
  34. int is_bipartite;
  35. stack<array<int, 4>> changes; // {node, size of component, did this make the graph non bipartite}
  36. int valid[MXN];
  37.  
  38. pair<int, int> find_par(int x) {
  39.     if (info[x] < 0) return make_pair(x, parity[x]);
  40.  
  41.     pair<int, int> p = find_par(info[x]);
  42.     p.second ^= 1;
  43.    
  44.     return p;
  45. }
  46.  
  47. void unite(int x, int y) {
  48.     auto [px, parity_x] = find_par(x);
  49.     auto [py, parity_y] = find_par(y);
  50.  
  51.     if (px == py) {
  52.         if (parity_x == parity_y) {
  53.             changes.emplace(array<int, 4>{py, -info[py], is_bipartite, parity[py]});
  54.             is_bipartite = 0;
  55.         } else {
  56.             changes.emplace(array<int, 4>{py, -info[py], 0, parity[py]});
  57.         }
  58.  
  59.         return;
  60.     }
  61.    
  62.     if (info[px] > info[py]) {
  63.         swap(px, py);
  64.         swap(parity_x, parity_y);
  65.         swap(x, y);
  66.     }
  67.  
  68.     changes.emplace(array<int, 4>{py, -info[py], 0, parity[py]});
  69.  
  70.     parity[py] = parity_x ^ parity_y ^ 1;
  71.     info[px] += info[py];
  72.     info[py] = px;
  73. }  
  74.  
  75. void rollback() {
  76.     assert(changes.empty() == false);
  77.  
  78.     auto [y, comp_sz, non_bipartite, parity_y] = changes.top();
  79.  
  80.     int x = info[y];
  81.     if (x == y) return;
  82.  
  83.     parity[y] = parity_y;  
  84.  
  85.     info[x] += comp_sz;
  86.     info[y] = -comp_sz;
  87.  
  88.     if (non_bipartite)
  89.         is_bipartite = 1;
  90.  
  91.     changes.pop();
  92. }
  93.  
  94. void add_edge(int idx) { unite(edges[idx].first, edges[idx].second); }
  95.  
  96. void dnc(int l, int r, int optl, int optr) {
  97.     if (l > r) return;
  98.  
  99.     int mid = (l + r) / 2;
  100.  
  101.     dbg(l, r, mid);
  102.     dbg(optl, optr);
  103.  
  104.     // Add prefix of edges
  105.     for (int i = l; i < mid; i++)
  106.         add_edge(i);
  107.  
  108.     dbg(sz(changes));
  109.  
  110.     int ufds_r = optr; // Edge list pointer
  111.  
  112.     // Calculate optimal right for mid, store in best
  113.     int best = 0;
  114.  
  115.     while (ufds_r > mid + 1 && ufds_r >= optl && is_bipartite)
  116.         add_edge(--ufds_r);
  117.  
  118.     dbg(ufds_r, sz(changes));
  119.  
  120.     if (!is_bipartite)
  121.         best = ufds_r;
  122.  
  123.     valid[mid] = best;
  124.  
  125.     dbg(is_bipartite, best);
  126.  
  127.     // Right side will be dnc(mid + 1, r, best, optr)
  128.     // For right side, already have prefix, need to work on removing suffix
  129.     while (ufds_r < optr) {
  130.         rollback();
  131.         ufds_r++;
  132.     }
  133.  
  134.     dbg(ufds_r, sz(changes));
  135.     dbg(is_bipartite, best);
  136.  
  137.     dbg_out();
  138.  
  139.     dnc(mid + 1, r, best, optr);
  140.  
  141.     // Remove prefix
  142.     for (int i = mid - 1; i >= l; i--)
  143.         rollback();
  144.  
  145.     // Now add those edges again
  146.     while (ufds_r > best + 1)
  147.         add_edge(--ufds_r);
  148.  
  149.     unite(edges[mid].first, edges[mid].second);
  150.  
  151.     // Left side will be dnc(l, mid - 1, optl, best + 1)
  152.     dnc(l, mid - 1, optl, best + 1);
  153.  
  154.     rollback();
  155.  
  156.     // Rollback suffix add
  157.     while (ufds_r < optr) {
  158.         rollback();
  159.         ufds_r++;
  160.     }
  161. }
  162.  
  163. void solve() {
  164.     cin >> N >> M >> Q;
  165.  
  166.     for (int i = 0; i < M; i++) {
  167.         cin >> edges[i].first >> edges[i].second;
  168.         edges[i].first--, edges[i].second--;
  169.     }
  170.  
  171.     is_bipartite = 1;
  172.     for (int i = 0; i < MXN; i++)
  173.         info[i] = -1, parity[i] = 0;
  174.  
  175.     dnc(0, M - 1, 0, M);
  176.  
  177.     for (int i = 0; i < M; i++)
  178.         dbg(i, valid[i]);
  179.  
  180.     while (Q--) {
  181.         int l, r;
  182.         cin >> l >> r;
  183.         l--, r--;
  184.  
  185.         cout << (valid[l] > r ? "YES" : "NO") << "\n";
  186.     }
  187. }
  188.  
  189. signed main() {
  190.     ios_base::sync_with_stdio(false);
  191.     cin.tie(nullptr);
  192.  
  193.     int TC = 1;
  194.     // cin >> TC;
  195.     while (TC--) solve();
  196. }
Add Comment
Please, Sign In to add comment