Submission #967364

#	Time	Username	Problem	Language	Result	Execution time	Memory
967364		owoovo	Fun Tour (APIO20_fun)	C++17	0 / 100	2 ms	1116 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 j=0;j<n;j++){
    if(j==root)continue;
    int to=chsz-1;
    for(int i=0;i<chsz-1;i++){
      tc[i][j]=hoursRequired(ch[i],j);
      if(tc[i][j]<tr[j])to=i;
    }
    sub[to].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(sub[j].size()>mxsz.F){
        mxsz={sub[j].size(),j};
      }
    }
    if(mer==0&&mxsz.F*2>left){
      mer=1;
      if(last==mxsz.S)last=(last+1)%3;
      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);
      }
      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();
      }
      best.S=mxsz.S;
    }
    if(best.S==-1){
      for(int j=0;j<chsz;j++){
        if(!sub[j].empty())best.S=j;
      }
    }
    assert(!sub[best.S].empty());
    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:43:23: 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]
   43 |       if(sub[j].size()>mxsz.F){
      |                       ^

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