Submission #386108

#TimeUsernameProblemLanguageResultExecution timeMemory
386108phathnvFun Tour (APIO20_fun)C++11
0 / 100
2 ms492 KiB
#include <bits/stdc++.h>
#include "fun.h"

using namespace std;

vector<int> createFunTour(int N, int Q) {
    pair<int, int> best = {N, 0};
    for(int i = 0; i < N; i++){
        int x = attractionsBehind(0, i);
        if (x >= (N + 1) / 2)
            best = min(best, {x, i});
    }
    int centroid = best.second;

    vector<int> subtreeRoots, distToCentroid(N);
    for(int i = 0; i < N; i++){
        distToCentroid[i] = hoursRequired(centroid, i);
        if (distToCentroid[i] == 1)
            subtreeRoots.push_back(i);
    }
    vector<vector<pair<int, int>>> subtree(3);
    for(int i = 0; i < N; i++){
        if (i == centroid)
            continue;
        for(int j = 0; j < (int) subtreeRoots.size(); j++)
            if (j + 1 == (int) subtreeRoots.size())
                subtree[j].push_back({distToCentroid[i], i});
            else if (hoursRequired(subtreeRoots[j], i) == distToCentroid[i] - 1){
                subtree[j].push_back({distToCentroid[i], i});
                break;
            }
    }
    for(auto &v : subtree)
        sort(v.begin(), v.end());

    vector<int> answer;
    int lastSubtree = -1;
    while (true){
        int sum = 0, maxSz = -1, big = -1;
        pair<int, int> nxt = {-1, -1};
        for(int i = 0; i < 3; i++){
            sum += subtree[i].size();
            if (maxSz < (int) subtree[i].size()){
                big = i;
                maxSz = subtree[i].size();
            }
            if (i != lastSubtree)
                nxt = max(nxt, {subtree[i].back().first, i});
        }
        int nxtSubtree = nxt.second;
        if (sum == 2 * maxSz){
            cerr << "merge " << endl;
            vector<int> small;
            for(int i = 0; i < 3; i++)
                if (i != big)
                    small.push_back(i);
            for(auto p : subtree[small[1]])
                subtree[small[0]].push_back(p);
            sort(subtree[small[0]].begin(), subtree[small[0]].end());
            if (lastSubtree != big && nxtSubtree != big)
                swap(subtree[small[0]], subtree[big]);
            assert(subtree[small[0]].size() == subtree[big].size());
            while (!subtree[small[0]].empty()){
                answer.push_back(subtree[big].back().second);
                answer.push_back(subtree[small[0]].back().second);
                subtree[big].pop_back();
                subtree[small[0]].pop_back();
            }
            break;
        } else {
            lastSubtree = nxtSubtree;
            answer.push_back(subtree[lastSubtree].back().second);
            subtree[lastSubtree].pop_back();
        }
    }
    answer.push_back(centroid);
    return answer;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...