#include <bits/stdc++.h>
using namespace std;
vector<int> train[500069], bus[500069];
set<int> adj[500069];
int par[500069], sz[500069];
bool visited[500069];
int pars;
void init(){
for(int i=0; i<500069; i++){
par[i] = i;
sz[i] = 1;
visited[i] = 0;
}
}
int fp(int x){
if(par[x]==x) return x;
else return par[x]=fp(par[x]);
}
void merge(int x, int y){
int px, py;
px = fp(x), py = fp(y);
if(sz[px] > sz[py]){
par[py] = px;
sz[px] += sz[py];
} else {
par[px] = py;
sz[py] += sz[px];
}
pars--;
}
void search(int x){
if(visited[x]) return;
visited[x] = true;
for(int i: train[x]){
if(visited[i]) continue;
if(fp(x) != fp(i)) merge(x,i);
search(x);
}
}
int main(){
ios::sync_with_stdio(0);
cout.tie(0); cin.tie(0);
init();
int n,m; cin >> n >> m;
pars = n;
for(int i=0; i<m; i++){
int a,b; char c;
cin >> a >> b >> c;
a--; b--;
if(c=='A'){
bus[a].push_back(b);
bus[b].push_back(a);
} else {
train[a].push_back(b);
train[b].push_back(a);
}
}
for(int i=0; i<n; i++){
if(!visited[i])search(i);
}
for(int i=0; i<n; i++){
for(int j: bus[i]){
int px = fp(i), py = fp(j);
adj[px].insert(py);
adj[py].insert(px);
}
}
int ans = 0;
if(pars == 1){
cout << n << "\n";
return 0;
}
for(int i=0; i<n; i++){
if(fp(i)!=i) continue;
//if(adj[i].size() != 0) cout << i << " " << adj[i].size() << "\n";
if(((int)adj[i].size()) == pars-1) ans += sz[fp(i)];
}
cout << ans << "\n";
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
295 ms |
62144 KB |
Output is correct |
2 |
Correct |
26 ms |
51632 KB |
Output is correct |
3 |
Correct |
27 ms |
51608 KB |
Output is correct |
4 |
Correct |
28 ms |
51668 KB |
Output is correct |
5 |
Correct |
27 ms |
51608 KB |
Output is correct |
6 |
Correct |
70 ms |
54096 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
28 ms |
51668 KB |
Output is correct |
2 |
Correct |
27 ms |
51608 KB |
Output is correct |
3 |
Correct |
23 ms |
51652 KB |
Output is correct |
4 |
Correct |
25 ms |
51796 KB |
Output is correct |
5 |
Correct |
28 ms |
52392 KB |
Output is correct |
6 |
Correct |
447 ms |
86484 KB |
Output is correct |
7 |
Correct |
681 ms |
104048 KB |
Output is correct |
8 |
Correct |
34 ms |
53588 KB |
Output is correct |
9 |
Correct |
43 ms |
54824 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
26 ms |
51632 KB |
Output is correct |
2 |
Correct |
27 ms |
51608 KB |
Output is correct |
3 |
Correct |
23 ms |
51652 KB |
Output is correct |
4 |
Correct |
25 ms |
51796 KB |
Output is correct |
5 |
Correct |
28 ms |
52392 KB |
Output is correct |
6 |
Correct |
23 ms |
51668 KB |
Output is correct |
7 |
Correct |
24 ms |
51680 KB |
Output is correct |
8 |
Correct |
25 ms |
51668 KB |
Output is correct |
9 |
Correct |
30 ms |
51680 KB |
Output is correct |
10 |
Incorrect |
33 ms |
51852 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
23 ms |
51668 KB |
Output is correct |
2 |
Correct |
24 ms |
51680 KB |
Output is correct |
3 |
Correct |
25 ms |
51668 KB |
Output is correct |
4 |
Correct |
30 ms |
51680 KB |
Output is correct |
5 |
Incorrect |
33 ms |
51852 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
23 ms |
51668 KB |
Output is correct |
2 |
Correct |
24 ms |
51680 KB |
Output is correct |
3 |
Correct |
25 ms |
51668 KB |
Output is correct |
4 |
Correct |
30 ms |
51680 KB |
Output is correct |
5 |
Incorrect |
33 ms |
51852 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |