結果

問題 No.833 かっこいい電車
ユーザー ei1333333
提出日時 2019-05-24 21:44:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 240 ms / 2,000 ms
コード長 4,753 bytes
コンパイル時間 3,416 ms
コンパイル使用メモリ 228,612 KB
最終ジャッジ日時 2025-01-07 03:57:38
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<bits/stdc++.h>
using namespace std;
using int64 = long long;
const int mod = 1e9 + 7;
const int64 infll = (1LL << 62) - 1;
const int inf = (1 << 30) - 1;
struct IoSetup {
  IoSetup() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(10);
    cerr << fixed << setprecision(10);
  }
} iosetup;
template< typename T1, typename T2 >
ostream &operator<<(ostream &os, const pair< T1, T2 > &p) {
  os << p.first << " " << p.second;
  return os;
}
template< typename T1, typename T2 >
istream &operator>>(istream &is, pair< T1, T2 > &p) {
  is >> p.first >> p.second;
  return is;
}
template< typename T >
ostream &operator<<(ostream &os, const vector< T > &v) {
  for(int i = 0; i < (int) v.size(); i++) {
    os << v[i] << (i + 1 != v.size() ? " " : "");
  }
  return os;
}
template< typename T >
istream &operator>>(istream &is, vector< T > &v) {
  for(T &in : v) is >> in;
  return is;
}
template< typename T1, typename T2 >
inline bool chmax(T1 &a, T2 b) { return a < b && (a = b, true); }
template< typename T1, typename T2 >
inline bool chmin(T1 &a, T2 b) { return a > b && (a = b, true); }
template< typename T = int64 >
vector< T > make_v(size_t a) {
  return vector< T >(a);
}
template< typename T, typename... Ts >
auto make_v(size_t a, Ts... ts) {
  return vector< decltype(make_v< T >(ts...)) >(a, make_v< T >(ts...));
}
template< typename T, typename V >
typename enable_if< is_class< T >::value == 0 >::type fill_v(T &t, const V &v) {
  t = v;
}
template< typename T, typename V >
typename enable_if< is_class< T >::value != 0 >::type fill_v(T &t, const V &v) {
  for(auto &e : t) fill_v(e, v);
}
using edge = pair< int, int >;
struct UnionFind {
  vector< int64 > data;
  stack< pair< int64, int64 > > st;
  UnionFind() {}
  UnionFind(int sz) {
    data.assign(sz, -1);
  }
  bool unite(int x, int y) {
    x = find(x), y = find(y);
    st.emplace(x, data[x]);
    st.emplace(y, data[y]);
    if(x == y) return (false);
    data[x] += data[y];
    data[y] = x;
    return (true);
  }
  int find(int k) {
    if(data[k] < 0) return (k);
    return (find(data[k]));
  }
  int64 size(int k) {
    return (-data[find(k)]);
  }
  void undo() {
    data[st.top().first] = st.top().second;
    st.pop();
    data[st.top().first] = st.top().second;
    st.pop();
  }
};
UnionFind uf;
struct Dynamic_connectivity {
  int n, sz;
  vector< vector< edge > > edges;
  vector< pair< pair< int, int >, edge > > pending;
  map< edge, int > cnt, appear;
  Dynamic_connectivity(int nn, int q) : n(q) {
    uf = UnionFind(nn);
    sz = 1;
    while(sz < n) sz <<= 1;
    edges.resize(2 * sz - 1);
  }
  void insert(int idx, edge e) {
    if(e.first > e.second) swap(e.first, e.second);
    if(cnt[e]++ == 0) appear[e] = idx;
  }
  void erase(int idx, edge e) {
    if(e.first > e.second) swap(e.first, e.second);
    if(--cnt[e] == 0) pending.emplace_back(make_pair(appear[e], idx), e);
  }
  void add(int a, int b, const edge &e, int k, int l, int r) {
    if(r <= a || b <= l) return;
    if(a <= l && r <= b) {
      edges[k].emplace_back(e);
      return;
    }
    add(a, b, e, 2 * k + 1, l, (l + r) >> 1);
    add(a, b, e, 2 * k + 2, (l + r) >> 1, r);
  }
  void add(int a, int b, const edge &e) {
    add(a, b, e, 0, 0, sz);
  }
  void build() {
    for(auto &p : cnt) {
      if(p.second > 0) pending.emplace_back(make_pair(appear[p.first], sz), p.first);
    }
    for(auto &s : pending) {
      add(s.first.first, s.first.second, s.second);
    }
  }
  void execute(const function< void(int) > &f, int k = 0) {
    for(auto &e : edges[k]) {
      int a, b;
      tie(a, b) = e;
      uf.unite(a, b);
    }
    if(k < sz - 1) {
      execute(f, 2 * k + 1);
      execute(f, 2 * k + 2);
    } else if(k - (sz - 1) < n) {
      int query_index = k - (sz - 1);
      f(query_index);
    }
    for(auto &e : edges[k]) {
      uf.undo();
    }
  }
};
int main() {
  int N, Q;
  cin >> N >> Q;
  vector< int > A(N);
  cin >> A;
  Dynamic_connectivity dc(202020, 202020);
  for(int i = 0; i < N; i++) {
    uf.data[i] = -A[i];
  }
  int ptr = N;
  set< pair< int, int > > st;
  vector< int > C(202020, -1);
  for(int i = 0; i < Q; i++) {
    int a, b;
    cin >> a >> b;
    --b;
    int c = b + 1;
    if(a == 1) {
      if(st.count({b, c})) continue;
      st.emplace(b, c);
      dc.insert(i, edge(b, c));
    } else if(a == 2) {
      if(!st.count({b, c})) continue;
      st.erase({b, c});
      dc.erase(i, edge(b, c));
    } else if(a == 3) {
      dc.insert(i, edge(b, ptr++));
    } else {
      C[i] = b;
    }
  }
  dc.build();
  auto f = [&](int k) {
    if(~C[k]) {
      cout << uf.size(C[k]) << endl;
    }
  };
  dc.execute(f);
  // いやいきりました
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0