Submission #416107

#	Time	Username	Problem	Language	Result	Execution time	Memory
416107		pauloamed	Bitaro’s Party (JOI18_bitaro)	C++14	Compilation error	0 ms	0 KiB

bitaro

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

#define MAXN 100010
#define ppiii pair<pair<int,int>,int>

int n;
vector<int> v[MAXN];
vector<int> rv[MAXN];

int fullSolution(int x, vector<bool> &on){
  vector<int> pd(n, -1);

  pd[x] = 0;
  for(int i = x; i >= 0; --i){
    if(pd[i] != -1){
      for(auto y : rv[i]){
          pd[y] = max(pd[y], pd[i] + 1);
      }
    }
  }

  int ans = -1;
  for(int i = 0; i <= x; ++i){
    // cout << pd[i] << endl;
    if(on[i]) ans = max(ans, pd[i]);
  }
  return ans;
}


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

  int m, q; cin >> n >> m >> q;
  for(int i = 0; i < m; ++i){
    int a, b; cin >> a >> b; a--, b--;
    if(a > b) swap(a, b);
    v[a].push_back(b);
    rv[b].push_back(a);
  }

  int block = sqrt(n) + 3;
  // cout << block << endl;

  vector<vector<pair<int,int>>> bestDists(n);
  for(int i = 0; i < n; ++i){
    vector<int> currPointers(rv[i].size());
    vector<bool> occour(n);

    priority_queue<ppiii, vector<ppiii>, less<ppiii>> pq;
    for(int j = 0; j < rv[i].size(); ++j){
      pq.push(make_pair(bestDists[rv[i][j]].front(), j));
    }

    while(pq.size() && bestDists[i].size() < block){
      auto x = pq.top();
      int j = x.second;
      int id = x.first.second;
      pq.pop();

      if(++currPointers[j] < bestDists[rv[i][j]].size()){
        pq.push(make_pair(bestDists[rv[i][j]][currPointers[j]], j));
      }

      if(!occour[id]){
        occour[id] = true;
        bestDists[i].push_back(x.first);
      }
    }

    for(auto &x : bestDists[i]) x.first++;
    bestDists[i].push_back({0, i});
  }

  vector<bool> on(n, true);
  while(q--){
    int x, y; cin >> x >> y; x--;
    vector<int> currOff;
    while(y--){
      int z; cin >> z; z--;
      on[z] = false;
      currOff.push_back(z);
    }

    int ans = -1;
    for(auto z : bestDists[x]){
      if(on[z.second]){
        ans = z.first; break;
      }
    }

    if(ans == -1) ans = fullSolution(x, on);
    cout << ans << "\n";
    for(auto z : currOff) on[z] = true;
  }
}

Compilation message (stderr)

Compilation timeout while compiling bitaro