#include <bits/extc++.h>
#include <bits/stdc++.h>
#define int long long
using namespace std;
string to_string(string s) { return s; }
template <typename T> string to_string(T v) {
bool first = true;
string res = "[";
for (const auto &x : v) {
if (!first)
res += ", ";
first = false;
res += to_string(x);
}
res += "]";
return res;
}
void dbg_out() { cout << endl; }
template <typename Head, typename... Tail> void dbg_out(Head H, Tail... T) {
cout << ' ' << to_string(H);
dbg_out(T...);
}
#ifdef DEBUG
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif
struct chash { // large odd number for C
const uint64_t C = (int)(4e18 * acos(0)) | 71;
int operator()(int x) const { return __builtin_bswap64(x * C); }
};
using hash_set = __gnu_pbds::gp_hash_table<int, __gnu_pbds::null_type, chash>;
signed main(void) {
ios_base::sync_with_stdio(false);
cin.tie(0);
int nbSommet, nbAretes;
cin >> nbSommet >> nbAretes;
vector<map<int, hash_set>> outEdges(nbSommet), inEdges(nbSommet);
vector<int> inDeg(nbSommet);
vector<int> idCC(nbSommet), szCC(nbSommet, 1);
vector<vector<int>> inCC(nbSommet);
for (int i = 0; i < nbSommet; ++i)
inCC[i].push_back(i);
iota(idCC.begin(), idCC.end(), 0LL);
int nbFollows = 0;
for (int iArete = 0; iArete < nbAretes; ++iArete) {
int u, v;
cin >> u >> v;
--u, --v;
v = idCC[v];
if (inEdges[v].count(idCC[u]) and
inEdges[v][idCC[u]].find(u) != inEdges[v][idCC[u]].end()) {
cout << nbFollows << '\n';
continue;
}
nbFollows += szCC[v];
inEdges[v][idCC[u]].insert(u);
outEdges[idCC[u]][v].insert(u);
inDeg[v]++;
u = idCC[u];
queue<pair<int, int>> toMerge;
if (outEdges[v].count(u)) {
toMerge.emplace(u, v);
}
while (!toMerge.empty()) {
auto [A, B] = toMerge.front();
toMerge.pop();
if (idCC[A] == idCC[B])
continue;
dbg(A, B);
if (szCC[A] < szCC[B])
swap(A, B);
if (inEdges[A].count(B)) {
nbFollows -= szCC[A] * inEdges[A][B].size();
inDeg[A] -= inEdges[A][B].size();
inEdges[A].erase(B);
outEdges[B].erase(A);
}
if (outEdges[A].count(B)) {
nbFollows -= szCC[B] * outEdges[A][B].size();
inDeg[B] -= outEdges[A][B].size();
outEdges[A].erase(B);
inEdges[B].erase(A);
}
int newSz = szCC[A] + szCC[B];
nbFollows += inDeg[A] * szCC[B] + inDeg[B] * szCC[A];
nbFollows += newSz * (newSz - 1) - szCC[A] * (szCC[A] - 1) -
szCC[B] * (szCC[B] - 1);
for (auto &[cc, s] : inEdges[B]) {
for (int y : s) {
outEdges[cc][B].erase(y);
if (!inEdges[A].count(cc) or
inEdges[A][cc].find(y) == inEdges[A][cc].end()) {
inEdges[A][cc].insert(y), inDeg[A]++;
outEdges[cc][A].insert(y);
}
}
}
for (auto &[cc, s] : outEdges[B]) {
for (int y : s) {
inEdges[cc][B].erase(y);
inDeg[cc]--;
if (!outEdges[A].count(cc) or
outEdges[A][cc].find(y) == outEdges[A][cc].end()) {
outEdges[A][cc].insert(y), inDeg[cc]++;
inEdges[cc][A].insert(y);
++inDeg[cc];
}
}
}
szCC[A] += szCC[B];
for (int x : inCC[B])
inCC[A].push_back(x), idCC[x] = A;
}
cout << nbFollows << '\n';
}
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
