Submission #642787

#	Time	Username	Problem	Language	Result	Execution time	Memory
642787		sanstzu	Monthly railway pass (LMIO18_menesinis_bilietas)	C++14	6 / 100	924 ms	123748 KiB

menesinis_bilietas

#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];

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];
    }
}

void search(int x){
    if(visited[x]) return;
    visited[x] = true;
    for(int i: train[x]){
        if(visited[i]) continue;
        merge(x,i);
        search(x);
    }
}

int main(){
    init();
    int n,m; cin >> n >> m;
    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++){
        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;
    set<int> pars;
    for(int i=0; i<n; i++){
        pars.insert(fp(i));
    }
    for(int i=0; i<n; i++){
        //if(adj[i].size() != 0) cout << i << " " << adj[i].size() << "\n";
        if(adj[i].size() == pars.size()-1) ans += sz[fp(i)];
    }
    cout << ans << "\n";
}

• Subtask 1: 0 / 10
• Subtask 2: 6 / 6
• Subtask 3: 0 / 30
• Subtask 4: 0 / 35
• Subtask 5: 0 / 19