oj.uz

답안 #766808

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
766808 2023-06-26T07:43:14 Z onjo0127 조이터에서 친구를 만드는건 재밌어 (JOI20_joitter2) C++17
0 / 100
 7 ms 14292 KB 
#include <bits/stdc++.h>
#define sz(V) ((int)V.size())
using namespace std;
using pii = pair<int, int>;
using ll = long long;

set<pii> E;
int P[100009];
set<int> I[100009], O[100009], A[100009];
ll ans;

int root(int x) {
	if(P[x] == x) return x;
	return P[x] = root(P[x]);
}

ll f(int x) {
	return 1LL * sz(A[x]) * (sz(I[x]) + sz(A[x]) - 1);
}

void merg(int u, int v) {
	u = root(u); v = root(v);
	if(sz(A[u]) + sz(I[u]) + sz(I[v]) > sz(A[v]) + sz(I[v]) + sz(O[v])) swap(u, v);
	ans -= f(u) + f(v);
	for(auto& it: A[u]) if(I[v].find(it) != I[v].end()) I[v].erase(it);
	for(auto& it: I[u]) if(A[v].find(it) == A[v].end()) I[v].insert(it);
	for(auto& it: O[u]) O[v].insert(it);
	for(auto& it: A[u]) A[v].insert(it);
	ans += f(v);
	P[u] = v;
}

int main() {
	int N, M; scanf("%d%d", &N, &M);
	for(int i=1; i<=N; i++) {
		P[i] = i;
		A[i] = {i};
	}
	while(M--) {
		int u, v; scanf("%d%d", &u, &v);
		E.insert({u, v});
		int U = root(u), V = root(v);
		if(U != V) {
			ans -= f(V);
			I[V].insert(u);
			O[U].insert(v);
			ans += f(V);
			if(E.find({v, u}) != E.end()) merg(u, v);
		}
		printf("%lld\n", ans);
	}
	return 0;
}

Compilation message

joitter2.cpp: In function 'int main()':
joitter2.cpp:34:17: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   34 |  int N, M; scanf("%d%d", &N, &M);
      |            ~~~~~^~~~~~~~~~~~~~~~
joitter2.cpp:40:18: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   40 |   int u, v; scanf("%d%d", &u, &v);
      |             ~~~~~^~~~~~~~~~~~~~~~

Subtask #1
0 / 1.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14292 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14292 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 83.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14292 KB Output isn't correct
2 Halted 0 ms 0 KB -