Submission #412574

#	Time	Username	Problem	Language	Result	Execution time	Memory
412574		ParsaAlizadeh	Bitaro’s Party (JOI18_bitaro)	C++17	14 / 100	2077 ms	114440 KiB

bitaro

#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;

#define X first
#define Y second

constexpr int SQ = 120;

int main() {
    ios_base::sync_with_stdio(false); cin.tie(nullptr);
    int n, m, q;
    cin >> n >> m >> q;
    vector<vector<int>> adj(n);
    for (int i = 0; i < m; i++) {
        int u, v;
        cin >> u >> v;
        u--; v--;
        adj[v].push_back(u);
    }
    vector<vector<pii>> best(n);
    vector<int> mark(n);
    int timer = 1;
    for (int i = 0; i < n; i++) {
        vector<pii> now = {{0, i}};
        for (int j : adj[i]) {
            for (pii p : best[j]) {
                p.X++;
                now.push_back(p);
            }
        }
        sort(begin(now), end(now), greater<>());
        for (int j = 0; best[i].size() <= SQ && j < now.size(); j++) {
            if (mark[now[j].Y] == timer)
                continue;
            mark[now[j].Y] = timer;
            best[i].push_back(now[j]);
        }
        timer++;
    }
    while (q --> 0) {
        int t, cnt;
        cin >> t >> cnt;
        t--;
        for (int i = 0; i < cnt; i++) {
            int u; cin >> u; u--;
            mark[u] = timer;
        }
        if (cnt > SQ) {
            vector<int> dp(n, -1);
            for (int i = 0; i < n; i++) {
                if (mark[i] != timer)
                    dp[i] = 0;
                for (int j : adj[i]) if (dp[j] > -1) {
                    dp[i] = max(dp[i], dp[j] + 1);
                }
            }
            cout << dp[t] << '\n';
        } else {
            int ans = -1;
            for (pii const& p : best[t]) {
                if (mark[p.Y] != timer) {
                    ans = p.X;
                    break;
                }
            }
            cout << ans << '\n';
        }
        timer++;
    }
    return 0;
}

Compilation message (stderr)

bitaro.cpp: In function 'int main()':
bitaro.cpp:37:51: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |         for (int j = 0; best[i].size() <= SQ && j < now.size(); j++) {
      |                                                 ~~^~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 7 / 7
• Subtask 3: 0 / 86