Submission #1277457

#	Time	Username	Problem	Language	Result	Execution time	Memory
1277457		hiepsimauhong	Bitaro’s Party (JOI18_bitaro)	C++20	7 / 100	2094 ms	10104 KiB

bitaro

#include <bits/stdc++.h>

using namespace std;

//#define int long long

#define FOR(I, L, R) for(int I(L) ; I <= (int)R ; ++I)
#define FOD(I, R, L) for(int I(R) ; I >= (int)L ; --I)
#define FOA(I, A) for(auto &I : A)

#define print(A,L,R) FOR(OK, L, R){if(A[OK]<=-oo / 10||A[OK]>=oo)cout<<"- ";else cout<<A[OK]<<' ';}cout<<'\n';
#define prints(A) FOA(OK, A){cout<<OK<<' ';}cout << '\n';
#define printz(A,L,R) FOR(OK, 1, L){FOR(KO, 1, R){if(A[OK][KO]>-oo&&A[OK][KO]<oo)cout<<A[OK][KO]<<' ';else cout << "- ";} cout << '\n';}cout << '\n';

#define fs first
#define sd second
#define ii pair<int,int>
#define iii pair<int, ii>
#define all(A) A.begin(), A.end()
#define quickly cin.tie(0) -> ios_base::sync_with_stdio(0);

const int N = 2e5 + 5;
const int SQRT = 320;

const int mod = 1e9 + 7;
const int oo = 1e9;

int n, m, q;
bool dead[N], vis[N];
int f[N], cnt[N];
ii dp[N][SQRT + 5];
vector<int> g[N];

signed main(){ quickly
        cin >> n >> m >> q;

        FOR(i, 1, m){
                int u, v;
                cin >> u >> v;
                g[v].push_back(u);
        }

        FOR(u, 1, n){
                dp[u][++cnt[u]] = {0, u};
                vector<ii> maxi = {{0, u}};

                FOA(v, g[u]){
                        FOR(i, 1, cnt[v]){
                                maxi.push_back(ii(dp[v][i].fs + 1, dp[v][i].sd));
                        }
                }
                sort(all(maxi));
                maxi.erase(unique(all(maxi)), maxi.end());
                reverse(all(maxi));

                cnt[u] += maxi.size() - 1;

                FOR(i, 1, cnt[u]){
                        dp[u][i] = maxi[i - 1];
                }

        }

        while(q--){
                int t, s;
                cin >> t >> s;

                if(s >= SQRT){
                        bool is = 0;

                        memset(dead, 0, sizeof dead);
                        memset(f, 0, sizeof f);

                        FOR(i, 1, s){
                                int u; cin >> u;
                                dead[u] = 1;

                                if(u == t) is = 1;
                        }

                        FOR(u, 1, t){
                                FOA(v, g[u]){
                                        if(!dead[v]){
                                                f[u] = max(f[v] + 1, f[u]);
                                        }
                                        else{
                                                if(f[v] != 0){
                                                        f[u] = max(f[u], f[v] + 1);
                                                }
                                        }
                                }
                        }

                        if(f[t] == 0 && is == 1){
                                cout << -1 << '\n';
                                continue;
                        }
                        cout << f[t] << '\n';
                }
                else{
                        vector<int> save;
                        FOR(i, 1, s){
                                int u; cin >> u;
                                save.push_back(u);
                                vis[u] = 1;
                        }

                        vis[0] = 1;

                        int ans = -1;
                        FOR(i, 1, SQRT){
                                if(!vis[dp[t][i].sd]){
                                        ans = dp[t][i].fs;
                                        break;
                                }
                        }

                        FOR(i, 0, s - 1){
                                vis[save[i]] = 0;
                        }

                        cout << ans << '\n';
                }
        }
}

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