Submission #201717

#	Time	Username	Problem	Language	Result	Execution time	Memory
201717		Atill83	Bitaro’s Party (JOI18_bitaro)	C++14	7 / 100	2093 ms	413160 KiB

bitaro

#include <bits/stdc++.h>
#define ff first
#define ss second
#define endl '\n'
using namespace std;
const long long INF = (long long) 1e18;
const int mod = (int) 1e9+7;
const int MAXN = (int) 3e5+5;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll n, m, q;
vector<int> adj[MAXN];
vector<int> adj2[MAXN];
vector<int> kisiler;
vector<pii> ans[MAXN];
bool visit[MAXN];
int d[MAXN];
const int sq = 400;



int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);

    #ifdef Local
        freopen("../IO/int.txt","r",stdin);
        freopen("../IO/out.txt","w",stdout);
    #endif

    cin>>n>>m>>q;
    for(int i = 0; i < m; i++){
        int a, b;
        cin>>a>>b;
        adj[a].push_back(b);
        adj2[b].push_back(a);
    }

    for(int i = 1; i <= n; i++){
        for(int j: adj2[i]){
            for(pii u: ans[j])
                ans[i].push_back({u.ff + 1, u.ss});
        }
        ans[i].push_back({0, i});
        sort(ans[i].begin(), ans[i].end(), greater<pii>());
        while(ans[i].size() > sq){
            ans[i].pop_back();
        }
    }


    for(int i = 0; i < q; i++){
        int x;
        int k;
        cin>>x>>k;
        if(k >= sq){
            for(int i = 1; i < x; i++) d[i] = -mod;
            d[x] = 0;
            vector<int> kisi;
            for(int j = 0; j < k; j++){
                int temp;
                cin>>temp;
                visit[temp] = 1;
                kisi.push_back(temp);
            }
            int res = -1;
            for(int i = x; i >= 1; i--){
                if(d[i] > res && !visit[i]) res = d[i];
                for(int j: adj2[i]) d[j] = max(d[j], d[i] + 1);
            }
            cout<<res<<endl;
            for(int j: kisi){
                visit[j] = 0;
            }
        }else{
            int res = -1;
            vector<int> kisi;
            for(int j = 0; j < k; j++){
                int temp;
                cin>>temp;
                visit[temp] = 1;
                kisi.push_back(temp);
            }
            for(int i = 0; i < ans[x].size(); i++){
                if(visit[ans[x][i].ss]) continue;
                res = ans[x][i].ff;
                break;
            }
            for(int j: kisi) visit[j] = 0;
            cout<<res<<endl;
        }
    }
    

    #ifdef Local
        cout<<endl<<fixed<<setprecision(2)<<1000.0 * clock() / CLOCKS_PER_SEC<< " milliseconds ";
    #endif
}

Compilation message (stderr)

bitaro.cpp: In function 'int main()':
bitaro.cpp:87:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for(int i = 0; i < ans[x].size(); i++){
                            ~~^~~~~~~~~~~~~~~

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