oj.uz

Submission #642792

# Submission time Handle Problem Language Result Execution time Memory
642792 2022-09-20T13:40:57 Z sanstzu Monthly railway pass (LMIO18_menesinis_bilietas) C++14
6 / 100
 999 ms 123264 KB 
#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];
set<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);
            pars.insert(px); pars.insert(py);
            adj[px].insert(py);
            adj[py].insert(px);
        }
    }
    int ans = 0;
    if((int)pars.size() == 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(adj[i].size() == (int)pars.size()-1) ans += sz[fp(i)];
    }
    cout << ans << "\n";
}

Compilation message

menesinis_bilietas.cpp: In function 'int main()':
menesinis_bilietas.cpp:83:26: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   83 |         if(adj[i].size() == (int)pars.size()-1) ans += sz[fp(i)];
      |            ~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 296 ms 62260 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 51672 KB Output is correct
2 Correct 27 ms 51636 KB Output is correct
3 Correct 27 ms 51784 KB Output is correct
4 Correct 29 ms 52432 KB Output is correct
5 Correct 25 ms 51692 KB Output is correct
6 Correct 649 ms 88844 KB Output is correct
7 Correct 999 ms 123264 KB Output is correct
8 Correct 52 ms 54220 KB Output is correct
9 Correct 56 ms 55528 KB Output is correct

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 27 ms 51636 KB Output is correct
2 Correct 27 ms 51784 KB Output is correct
3 Correct 29 ms 52432 KB Output is correct
4 Correct 26 ms 51644 KB Output is correct
5 Correct 24 ms 51668 KB Output is correct
6 Correct 25 ms 51668 KB Output is correct
7 Correct 29 ms 51740 KB Output is correct
8 Incorrect 28 ms 51844 KB Output isn't correct
9 Halted 0 ms 0 KB -

Subtask #4
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 26 ms 51644 KB Output is correct
2 Correct 24 ms 51668 KB Output is correct
3 Correct 25 ms 51668 KB Output is correct
4 Correct 29 ms 51740 KB Output is correct
5 Incorrect 28 ms 51844 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #5
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 26 ms 51644 KB Output is correct
2 Correct 24 ms 51668 KB Output is correct
3 Correct 25 ms 51668 KB Output is correct
4 Correct 29 ms 51740 KB Output is correct
5 Incorrect 28 ms 51844 KB Output isn't correct
6 Halted 0 ms 0 KB -