Submission #707407

#	Time	Username	Problem	Language	Result	Execution time	Memory
707407		EthanKim8683	Bitaro’s Party (JOI18_bitaro)	C++17	7 / 100	2056 ms	268656 KiB

bitaro

#include<bits/stdc++.h>
using namespace std;
using I=int;
const I N=100000;
const I MIN=-1e9;
vector<I>adjs[N];
set<I>viss;
vector<pair<I,I>>pres[N];
vector<pair<I,I>>tops[N];
set<I>curs;
I dp[N];
I main(){
  cin.tie(0)->sync_with_stdio(0);
  I n,m,q;cin>>n>>m>>q;
  for(I i=0;i<m;i++){
    I s,e;cin>>s>>e,s--,e--;
    adjs[s].push_back(e);
  }
  I siz=sqrt(n/log2(n))*1.5;
  for(I i=0;i<n;i++){
    pres[i].push_back({0,i});
    sort(pres[i].begin(),pres[i].end());
    viss.clear();
    for(I j=0;tops[i].size()<siz&&j<pres[i].size();j++){
      auto[dis,k]=pres[i][j];
      if(viss.count(k))continue;
      viss.insert(k),tops[i].push_back({dis,k});
    }
    pres[i].clear();
    for(auto j:adjs[i])for(auto[dis,k]:tops[i])pres[j].push_back({dis-1,k});
  }
  while(q--){
    I t,y;cin>>t>>y,t--;
    curs.clear();
    for(I i=0;i<y;i++){I c;cin>>c,curs.insert(c-1);}
    I res=MIN;
    for(auto[dis,i]:tops[t])if(!curs.count(i)){res=-dis;break;}
    if(res!=MIN){printf("%i\n",res);continue;}
    if(curs.size()<=siz){printf("-1\n");continue;}
    fill_n(dp,t+1,0);
    for(auto c:curs)dp[c]=MIN;
    for(I i=0;i<t;i++)if(dp[i]!=MIN)for(auto j:adjs[i])dp[j]=max(dp[j],dp[i]+1);
    printf("%i\n",dp[t]==MIN?-1:dp[t]);
  }
}

Compilation message (stderr)

bitaro.cpp: In function 'I main()':
bitaro.cpp:24:29: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'I' {aka 'int'} [-Wsign-compare]
   24 |     for(I j=0;tops[i].size()<siz&&j<pres[i].size();j++){
      |               ~~~~~~~~~~~~~~^~~~
bitaro.cpp:24:36: warning: comparison of integer expressions of different signedness: 'I' {aka 'int'} and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   24 |     for(I j=0;tops[i].size()<siz&&j<pres[i].size();j++){
      |                                   ~^~~~~~~~~~~~~~~
bitaro.cpp:39:19: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'I' {aka 'int'} [-Wsign-compare]
   39 |     if(curs.size()<=siz){printf("-1\n");continue;}
      |        ~~~~~~~~~~~^~~~~

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