Submission #416334

#	Time	Username	Problem	Language	Result	Execution time	Memory
416334		pauloamed	Bitaro’s Party (JOI18_bitaro)	C++14	14 / 100	2074 ms	216516 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(x+1, -1);

  int ans = -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);
      }
    }
    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 = max(1, ((int)sqrt(n)) / 2);
  // 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++;
    if(bestDists[i].size() < block) 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)

bitaro.cpp: In function 'int main()':
bitaro.cpp:50:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |     for(int j = 0; j < rv[i].size(); ++j){
      |                    ~~^~~~~~~~~~~~~~
bitaro.cpp:54: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]
   54 |     while(pq.size() && bestDists[i].size() < block){
      |                        ~~~~~~~~~~~~~~~~~~~~^~~~~~~
bitaro.cpp:60:28: 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]
   60 |       if(++currPointers[j] < bestDists[rv[i][j]].size()){
      |          ~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
bitaro.cpp:71:28: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   71 |     if(bestDists[i].size() < block) bestDists[i].push_back({0, i});
      |        ~~~~~~~~~~~~~~~~~~~~^~~~~~~

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