Submission #642995

#	Time	Username	Problem	Language	Result	Execution time	Memory
642995		CyberSleeper	Monthly railway pass (LMIO18_menesinis_bilietas)	C++17	100 / 100	666 ms	102912 KiB

menesinis_bilietas

#include <bits/stdc++.h>
using namespace std;

const int MX=500007;

int N, M, group[MX], sz[MX], cnt_group, ans;
vector<int> bus[MX], tr[MX];
set<int> adj[MX];

void DFS(int x){
    group[x]=cnt_group;
    sz[cnt_group]++;
    for(int i:tr[x]){
        if(group[i])
            continue;
        DFS(i);
    }
}

int main(){
    cin >> N >> M;
    for(int i=0; i<M; i++){
        int u, v;
        char w;
        cin >> u >> v >> w;
        if(u == v)
            continue;
        if(w == 'A'){
            bus[u].push_back(v);
            bus[v].push_back(u);
        }else{
            tr[u].push_back(v);
            tr[v].push_back(u);
        }
    }
    for(int i=1; i<=N; i++){
        if(!group[i]){
            cnt_group++;
            DFS(i);
        }
    }
    for(int i=1; i<=N; i++){
        for(int j:bus[i]){
            if(group[i] != group[j])
                adj[group[i]].insert(group[j]);
        }
    }
    for(int i=1; i<=N; i++){
        if(adj[i].size()==cnt_group-1)
            ans+=sz[i];
    }
    cout << ans << endl;
}

Compilation message (stderr)

menesinis_bilietas.cpp: In function 'int main()':
menesinis_bilietas.cpp:49:25: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   49 |         if(adj[i].size()==cnt_group-1)
      |            ~~~~~~~~~~~~~^~~~~~~~~~~~~

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