#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T l, T r) { while (l != r) cerr << *l << " \n"[next(l)==r], ++l; }
#else
#define DE(...) 0
#define debug(...) 0
#endif
// My bug list :
// integer overflow
// 0based, 1based forgotten
// index out of bound
// n, m, i, j typo
// some cases are missing
const int MAX_N = 100010;
ll ans;
int n, m, g[MAX_N], sz[MAX_N], cost[MAX_N];
ll cntin[MAX_N];
map<int, set<pair<int,int>>> in[MAX_N], out[MAX_N];
void init() {
iota(g, g+n+1, 0);
fill(sz, sz + n + 1, 1);
}
int F(int i) { return i == g[i] ? i : g[i] = F(g[i]); }
void rm(int x, int u) {
--cost[x], --cost[u];
ans -= sz[ F(u) ];
--cntin[ F(u) ];
in[F(u)][F(x)].erase({x, u});
out[F(x)][F(u)].erase({x, u});
}
void ad(int x, int u) {
++cost[x], ++cost[u];
ans += sz[ F(u) ];
++cntin[ F(u) ];
in[F(u)][F(x)].insert({x, u});
out[F(x)][F(u)].insert({x, u});
}
bool mcnt(set<pair<int,int>> &st, int k) {
return st.lower_bound({k, -1}) != st.upper_bound({k, -1});
}
void kick(int a, int b) {
auto vec = out[F(a)][F(b)];
for (auto [x, y] : vec) rm(x, y);
}
void add_edge(int a, int b) {
if (F(a) == F(b)) return;
if (mcnt(in[F(b)][F(a)], a)) return;
ad(a, b);
if (out[F(b)][F(a)].empty()) return;
a = F(a), b = F(b);
if (cost[a] > cost[b]) swap(a, b);
kick(a, b), kick(b, a);
vector<pair<int,int>> kl;
for (auto [_, vec] : in[a])
kl.insert(end(kl), AI(vec));
for (auto [_, vec] : out[a])
kl.insert(end(kl), AI(vec));
in[a].clear(), out[a].clear();
for (auto [x, y] : kl) rm(x, y);
ans += 2ll * sz[a] * sz[b] + 1ll * cntin[b] * sz[a];
sz[b] += sz[a], g[a] = b;
for (auto [x, y] : kl) add_edge(x, y);
}
int32_t main() {
ios_base::sync_with_stdio(0), cin.tie(0);
cin >> n >> m;
init();
while (m--) {
int a, b;
cin >> a >> b;
add_edge(a, b);
cout << ans << '\n';
}
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
6 ms |
9676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
6 ms |
9676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
6 ms |
9676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
