oj.uz

Submission #1014371

# Submission time Handle Problem Language Result Execution time Memory
1014371 2024-07-04T19:16:51 Z JakobZorz Longest Trip (IOI23_longesttrip) C++17
5 / 100
 9 ms 764 KB 
#include"longesttrip.h"
#include<iostream>
#include<algorithm>
using namespace std;

int n;
int perm[1000];

bool connected(vector<int>a,vector<int>b){
    for(int&i:a)
        i=perm[i];
    for(int&i:b)
        i=perm[i];
    return are_connected(a,b);
}

pair<int,int>find_conn(vector<int>v1,vector<int>v2){
    for(int i:v1){
        if(connected({i},v2)){
            for(int j:v2){
                if(connected({i},{j}))
                    return{i,j};
            }
        }
    }
    // impossible
    exit(1);
}

vector<int>merge(vector<int>v1,vector<int>v2){
    while(v2.size()){
        v1.push_back(v2.back());
        v2.pop_back();
    }
    return v1;
}

pair<vector<int>,vector<int>>merge(vector<int>v1,vector<int>v2,vector<int>v3){
    if(connected({v1.back()},{v2.back()}))
        return{merge(v1,v2),v3};
    
    if(connected({v2.back()},{v3.back()}))
        return{merge(v2,v3),v1};
    
    return{merge(v1,v3),v2};
}

vector<int>cyclic_shift(vector<int>arr,int a){
    // it ends with a
    vector<int>res;
    bool b1=false;
    for(int i:arr){
        if(b1)
            res.push_back(i);
        if(i==a)
            b1=true;
    }
    
    bool b2=true;
    for(int i:arr){
        if(b2)
            res.push_back(i);
        if(i==a)
            b2=false;
    }
    return res;
}

vector<int>longest_trip(int N,int D){
    srand(54367582);
    n=N;
    for(int i=0;i<n;i++)
        perm[i]=i;
    for(int i=0;i<n;i++)
        swap(perm[i],perm[rand()%n]);
    
    vector<vector<int>>arr;
    for(int i=0;i<n;i++)
        arr.push_back({i});
    
    while(arr.size()>2){
        vector<int>v1,v2,v3;
        v1=arr.back();
        arr.pop_back();
        v2=arr.back();
        arr.pop_back();
        v3=arr.back();
        arr.pop_back();
        auto res=merge(v1,v2,v3);
        arr.push_back(res.first);
        arr.push_back(res.second);
    }
    
    vector<int>res1=arr[0];
    vector<int>res2=arr[1];
    
    if(!connected(res1,res2)){
        if(res1.size()>res2.size())
            return res1;
        else
            return res2;
    }
    
    if(connected({res1.back()},{res2.back()}))
        return merge(res1,res2);
    reverse(res1.begin(),res1.end());
    if(connected({res1.back()},{res2.back()}))
        return merge(res1,res2);
    reverse(res2.begin(),res2.end());
    if(connected({res1.back()},{res2.back()}))
        return merge(res1,res2);
    reverse(res1.begin(),res1.end());
    if(connected({res1.back()},{res2.back()}))
        return merge(res1,res2);
    
    // both must be cyclical then
    auto pr=find_conn(res1,res2);
    vector<int>r1=cyclic_shift(res1,pr.first);
    vector<int>r2=cyclic_shift(res2,pr.second);
    vector<int>r=merge(r1,r2);
    for(int&i:r)
        i=perm[i];
    return r;
}

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Incorrect
2 Halted 0 ms 0 KB -

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 6 ms 764 KB Output is correct

Subtask #2
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 8 ms 344 KB Output is correct
4 Correct 6 ms 344 KB Output is correct
5 Correct 7 ms 552 KB Output is correct
6 Incorrect 0 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -

Subtask #3
0.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 6 ms 344 KB Output is correct
5 Correct 6 ms 460 KB Output is correct
6 Incorrect 0 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -

Subtask #4
0.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 9 ms 344 KB Output is correct
3 Correct 8 ms 344 KB Output is correct
4 Correct 6 ms 448 KB Output is correct
5 Correct 6 ms 556 KB Output is correct
6 Incorrect 0 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -