Submission #416100

#	Time	Username	Problem	Language	Result	Execution time	Memory
416100		pauloamed	Bitaro’s Party (JOI18_bitaro)	C++14	0 / 100	10 ms	5448 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){
    for(auto y : rv[i]){
      if(pd[y] != -1){
        pd[i] = max(pd[i], pd[y] + 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<vector<pair<int,int>>> tmp;

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

    // cout << i << endl;
    // for(auto x : tmp){
    //   for(auto y:x){
    //     cout << y.first << " " << y.second << ", ";
    //   }cout << endl;
    // }


    vector<int> currPointers(tmp.size());
    vector<bool> occour(n);
    priority_queue<ppiii, vector<ppiii>, less<ppiii>> pq;

    for(int j = 0; j < tmp.size(); ++j){
      pq.push(make_pair(tmp[j].front(), j));
    }

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

      if(++currPointers[x.second] < tmp[x.second].size()){
        pq.push(make_pair(tmp[x.second][currPointers[x.second]], x.second));
      }

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

  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)

bitaro.cpp: In function 'int main()':
bitaro.cpp:68:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   68 |     for(int j = 0; j < tmp.size(); ++j){
      |                    ~~^~~~~~~~~~~~
bitaro.cpp:72:44: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   72 |     while(pq.size() && bestDists[i].size() < block){
      |                        ~~~~~~~~~~~~~~~~~~~~^~~~~~~
bitaro.cpp:76:35: warning: comparison of integer expressions of different signedness: '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'} and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |       if(++currPointers[x.second] < tmp[x.second].size()){
      |          ~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

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