Submission #247027

#	Time	Username	Problem	Language	Result	Execution time	Memory
247027		dolphingarlic	Monthly railway pass (LMIO18_menesinis_bilietas)	C++14	100 / 100	664 ms	66912 KiB

menesinis_bilietas

#include <bits/stdc++.h>
#define FOR(i, x, y) for (int i = x; i < y; i++)
typedef long long ll;
using namespace std;

int cmp[500001], cmp_cnt;
set<int> graph[500001];

int find(int A) {
    while (A != cmp[A]) cmp[A] = cmp[cmp[A]], A = cmp[A];
    return A;
}

void onion(int A, int B) {
    if (find(A) == find(B)) return;
    cmp[find(A)] = find(B);
    cmp_cnt--;
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n >> m;
    iota(cmp, cmp + n + 1, 0);
    cmp_cnt = n;
    vector<pair<int, int>> bus;
    while (m--) {
        int u, v;
        char t;
        cin >> u >> v >> t;
        if (t == 'A') bus.push_back({u, v});
        else onion(u, v);
    }
    for (pair<int, int> i : bus) {
        if (find(i.first) == find(i.second)) continue;
        graph[find(i.first)].insert(find(i.second));
        graph[find(i.second)].insert(find(i.first));
    }

    int ans = 0;
    FOR(i, 1, n + 1) if (graph[find(i)].size() == cmp_cnt - 1) ans++;
    cout << ans;
    return 0;
}

Compilation message (stderr)

menesinis_bilietas.cpp: In function 'int main()':
menesinis_bilietas.cpp:42:48: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     FOR(i, 1, n + 1) if (graph[find(i)].size() == cmp_cnt - 1) ans++;
                          ~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~

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