Submission #1212582

#	Time	Username	Problem	Language	Result	Execution time	Memory
1212582		Aviansh	Longest Trip (IOI23_longesttrip)	C++20	15 / 100	337 ms	472 KiB

longesttrip

#include "longesttrip.h"

#include <bits/stdc++.h>

using namespace std;

vector<int> longest_trip(int n, int D) {
    if(D==3){
        vector<int>ans(n);
        iota(ans.begin(),ans.end(),0);
        return ans;
    }
    if(D==2){
        set<int>lef;
        for(int i = 0;i<n;i++){
            lef.insert(i);
        }
        vector<int>ans;
        ans.push_back(0);
        lef.erase(0);
        while(lef.size()>1){
            if(are_connected({ans.back()},{*lef.begin()})){
                ans.push_back(*lef.begin());
                lef.erase(lef.begin());
            }
            else{
                ans.push_back(*(++lef.begin()));
                lef.erase(++lef.begin());
            }
        }
        if(are_connected({*lef.begin()},{ans.back()})){
            ans.push_back(*lef.begin());
            return ans;
        }
        ans.insert(ans.begin(),*lef.begin());
        return ans;
    }
    assert(D==1);
    bool adj[n][n];
    for(int i = 0;i<n;i++){
        fill(adj[i],adj[i]+n,0);
    }
    for(int i = 0;i<n;i++){
        for(int j = i+1;j<n;j++){
            if(are_connected({i},{j})){
                adj[i][j]=adj[j][i]=1;
            }
        }
    }
    vector<int>pt1,pt2;
    pt1.push_back(0);
    for(int i = 1;i<n;i++){
        int prev1 = pt1.back();
        if(adj[prev1][i]){
            pt1.push_back(i);
            continue;
        }
        //not adjacent.
        if(pt2.size()){
            assert(adj[pt2.back()][i]);
        }
        pt2.push_back(i);
    }
    if(pt1.size()>pt2.size()){
        return pt1;
    }
    return pt2;
}

• Subtask 0: 0 / 0
• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 0 / 25
• Subtask 4: 0 / 60