Submission #967358

#	Time	Username	Problem	Language	Result	Execution time	Memory
967358		owoovo	Fun Tour (APIO20_fun)	C++17	10 / 100	77 ms	15680 KiB

fun

#include "fun.h"
#include<bits/stdc++.h>
#define ll long long 
#define F first 
#define S second 
using namespace std;
vector<int> createFunTour(int n, int q) {
  int root=0,hf=(n+1)/2,mn=n;
  for(int i=0;i<n;i++){
    int A=attractionsBehind(0,i);
    if(A>hf&&A<mn){
      mn=A,root=i;
    }
  }
  int tr[100010],tc[3][100010];
  vector<int> ch;
  priority_queue<pair<int,int>> sub[3];
  for(int i=0;i<n;i++){
    tr[i]=hoursRequired(root,i);
    if(tr[i]==1)ch.push_back(i);
  }
  int chsz=ch.size();
  for(int i=0;i<chsz;i++){
    for(int j=0;j<n;j++){
      tc[i][j]=hoursRequired(ch[i],j);
      if(tc[i][j]<tr[j]){
        sub[i].push({tr[j],j});
      }
    }
  }
  int left=(n-1),mer=0,last=-1;
  vector<int> ans;
  while(left>0){
    pair<int,int> best={0,-1};
    for(int j=0;j<chsz;j++){
      if(j!=last&&(!sub[j].empty())&&sub[j].top().F>best.F){
        best={sub[j].top().F,j};
      }
    }
    pair<int,int> mxsz={0,0};
    for(int j=0;j<chsz;j++){
      if(j!=last&&(!sub[j].empty())&&sub[j].size()>mxsz.F){
        mxsz={sub[j].size(),j};
      }
    }

    // if(mer==0&&mxsz.F*2>left){
    //   mer=1;
    //   int lf=3-last-mxsz.S;
    //   if(((!sub[lf].empty())&&(!sub[last].empty()))&&sub[lf].top().F>sub[last].top().F){
    //     left--;
    //     last=lf;
    //     auto [x,y]=sub[last].top();
    //     sub[last].pop();
    //     ans.push_back(y);
    //   }
    //   best.S=mxsz.S;
    //   if(last==mxsz.S)last=(last+1)%3;
    //   lf=3-last-mxsz.S;
    //   while(!sub[lf].empty()){
    //     sub[last].push(sub[lf].top());
    //     sub[lf].pop();
    //   }
    // }
    if(best.S==-1){
      for(int j=0;j<chsz;j++){
        if(!sub[j].empty())best.S=j;
      }
    }
    auto [x,y]=sub[best.S].top();
    sub[best.S].pop();
    last=best.S;
    ans.push_back(y);
    left--;
  }
  ans.push_back(root);
  return ans;
}

Compilation message (stderr)

fun.cpp: In function 'std::vector<int> createFunTour(int, int)':
fun.cpp:42:51: warning: comparison of integer expressions of different signedness: 'std::priority_queue<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   42 |       if(j!=last&&(!sub[j].empty())&&sub[j].size()>mxsz.F){
      |                                                   ^
fun.cpp:31:18: warning: unused variable 'mer' [-Wunused-variable]
   31 |   int left=(n-1),mer=0,last=-1;
      |                  ^~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 16
• Subtask 3: 0 / 21
• Subtask 4: 0 / 19
• Subtask 5: 0 / 34