#include<bits/stdc++.h>
#define int long long
using namespace std;
multiset<pair<int, bool>> ad[100005];
unordered_set<int> sus[100005];
int ans = 0, root[100005], sz[100005], deg[100005];
int getroot(int u)
{
if(root[u] == u) return u;
else return getroot(root[u]);
}
void unite(int u, int v)
{
u = getroot(u); v = getroot(v);
if(u == v) return;
if(sz[u] < sz[v]) swap(u, v);
root[v] = u; sz[u] += sz[v];
}
set<int> mergeComponent(int u, int v)
{
u = getroot(u); v = getroot(v);
if(u == v) return {};
int num = ad[v].count({u, 0}), num2 = ad[u].count({v, 0});
set<int> add;
ans += -num*sz[u] - num2*sz[v]+ 2*sz[u]*sz[v];
//cout<<"A"<<ans<<" "<<2*sz[u]*sz[v]<<endl;
deg[u] -= num; deg[v] -= num2; ad[v].erase({u, 0}); ad[v].erase({u, 1});
ad[u].erase({v, 0}); ad[u].erase({v, 1});
if(sz[u] < sz[v]) swap(u, v);
for(pair<int, bool> i : ad[v]){ //Merge to ad[u]
if(ad[u].count(make_pair(i.first, i.second^1)) > 0) add.insert(i.first);/**/
ad[u].insert(i);
pair<int, bool> j = make_pair(v, i.second^1);
ad[i.first].erase(ad[i.first].lower_bound(j));
j.first = u;
ad[i.first].insert(j);
}
for(int i : sus[v]) sus[u].insert(i);
for(int i : add){
ad[u].erase({i, 0});
ad[u].erase({i, 1});
}
ans += deg[u]*sz[v] + deg[v]*sz[u];
deg[u] += deg[v]; deg[v] = 0;
ad[v].clear();
unite(u, v);
//cout<<"A"<<add.size()<<endl;
return add;
}
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
int n, m;
cin>>n>>m;
for(int i = 1; i <= n; i++){
root[i] = i; sz[i] = 1;
}
for(int test = 1; test <= m; test++){
int u, v;
cin>>u>>v;
int reu = u;
u = getroot(u); v = getroot(v);
if(u == v || sus[v].count(reu) == 1){cout<<ans<<'\n'; continue;}
int num = ad[v].count({u, 0});
if(num > 0){
//Merge two components
//mergeComponent(1, 2, 0);
vector<int> order; order.push_back(v);
for(int i = 0; i < order.size(); i++){
set<int> x = mergeComponent(u, order[i]);
for(int j : x) order.push_back(j);
//cout<<"A"<<order[i]<<" "<<order.size()<<endl;
}
}
else{
ans += sz[v];
ad[u].insert({v, 0});
ad[v].insert({u, 1});
sus[v].insert(reu);
deg[v]++;
}
cout<<ans<<'\n';
}
}
Compilation message
joitter2.cpp: In function 'int main()':
joitter2.cpp:70:30: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
70 | for(int i = 0; i < order.size(); i++){
| ~~^~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
3 ms |
12636 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
3 ms |
12636 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
3 ms |
12636 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
