Submission #967371

#	Time	Username	Problem	Language	Result	Execution time	Memory
967371		owoovo	Fun Tour (APIO20_fun)	C++17	100 / 100	198 ms	21196 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=0;
    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&&last!=mxsz.S){
            mer=1;
            int lf=3-last-mxsz.S;
            if(((!sub[lf].empty())&&(!sub[last].empty()))&&sub[lf].top().F>sub[last].top().F){
                last=lf;
                auto [x,y]=sub[last].top();
                sub[last].pop();
                ans.push_back(y);
                left--;
            }
            lf=3-last-mxsz.S;
            //cout<<lf<<" "<<last<<" "<<mxsz.S<<"\n";
            assert(lf>=0&&lf<3);
            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();
        ans.push_back(y);
        left--;
        last=best.S;
    }
    ans.push_back(root);
    return ans;
}

Compilation message (stderr)

fun.cpp: In function 'std::vector<int> createFunTour(int, int)':
fun.cpp:43:29: 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: 10 / 10
• Subtask 2: 16 / 16
• Subtask 3: 21 / 21
• Subtask 4: 19 / 19
• Subtask 5: 34 / 34