Submission #1030255

#	Time	Username	Problem	Language	Result	Execution time	Memory
1030255		Agreb	Longest Trip (IOI23_longesttrip)	C++17	70 / 100	34 ms	1108 KiB

longesttrip

#include <bits/stdc++.h>
#include "longesttrip.h"
using namespace std;

vector <int> longest_trip(int N, int D){
    vector <int> path, clique, used(N);
    for (int i = 0; i <= 2; i++){
        for (int j = 0; j < i; j++){
            if (are_connected({i}, {j})){
                path = {i, j};
                used[i] = used[j] = 1;
                break;
            }
        }
        if (!path.empty()) break;
    }

    mt19937 rnd(239);
    vector <int> perm(N);
    for (int i = 0; i < N; i++) perm[i] = i;
    shuffle(perm.begin(), perm.end(), rnd);
    
    vector <int> non_edge;
    for (int vv = 0; vv < N; vv++){
        int v = perm[vv];
        if (used[v]) continue;
        int P = path.size();
        int C = clique.size();
        int st = path[0], fin = path[P-1];

        // cout << v << endl;
        // for (int x : path) cout << x << " ";
        // cout << endl;
        // for (int y : clique) cout << y << endl;
        // cout << endl;

        if (!clique.empty()){
            int cp = are_connected(path, {v});
            int cc = are_connected(clique, {v});
            if (cp && cc){
                non_edge = {clique[0], path[0]};

                int ip = -1;
                for (int i = 0; i < P; i++){
                    if (are_connected({path[i]}, {v})){
                        ip = i;
                        break;
                    }
                }

                int ic = -1;
                for (int i = 0; i < C; i++){
                    if (are_connected({clique[i]}, {v})){
                        ic = i;
                        break;
                    }
                }

                vector <int> npath;
                for (int j = 0; j < P; j++) npath.push_back(path[(j+ip+1)%P]);
                npath.push_back(v);
                for (int j = 0; j < C; j++) npath.push_back(clique[(j+ic)%C]);
                path = npath;
                clique.clear();
                continue;
            }
            else if (cc){
                clique.push_back(v);
                continue;
            }
            else if (cp){
                path.push_back(v);
                continue;
            }
        }

        if (!non_edge.empty() && are_connected({st}, {fin})){
            shuffle(non_edge.begin(), non_edge.end(), rnd);
            for (int u : non_edge){
                if (are_connected({u}, {v})){   
                    int i = -1;
                    for (int ii = 0; ii < P; ii++)
                        if (path[ii] == u) i = ii;
                             
                    vector <int> npath;
                    for (int j = 0; j < P; j++) 
                        npath.push_back(path[(j+i+1)%P]);
                    npath.push_back(v);
                    path = npath;
                    break;
                }
            }
            continue;
        }

        if (!are_connected(path, {v})){
            clique.push_back(v);
            continue;
        }

        if (are_connected({st}, {v})){
            reverse(path.begin(), path.end());
            path.push_back(v);
            continue;
        }
        else if (non_edge.empty()) non_edge = {st, v};        

        if (are_connected({fin}, {v})){
            path.push_back(v);
            continue;
        }

        for (int i = 1; i < P-1; i++){
            if (are_connected({path[i]}, {v})){   
                vector <int> npath;
                for (int j = 0; j < P; j++) 
                    npath.push_back(path[(j+i+1)%P]);
                npath.push_back(v);
                path = npath;
                break;
            }
        }        
    }
    assert(path.size() + clique.size() == N);
    if (path.size() >= clique.size()) return path;
    else return clique;
}

Compilation message (stderr)

In file included from /usr/include/c++/10/cassert:44,
                 from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
                 from longesttrip.cpp:1:
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:124:40: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  124 |     assert(path.size() + clique.size() == N);
      |            ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~

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