답안 #846637

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
846637 2023-09-08T08:02:36 Z ttamx 가장 긴 여행 (IOI23_longesttrip) C++17
5 / 100
9 ms 512 KB
#include "longesttrip.h"
#include<bits/stdc++.h>

using namespace std;

vector<int> perm;

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

int bsearch(vector<int> vec,vector<int> fixed){
    int l=0,r=vec.size()-1;
    while(l<r){
        int m=(l+r)/2;
        vector<int> res;
        for(int i=l;i<=m;i++)res.emplace_back(vec[i]);
        if(ask(res,fixed))r=m;
        else l=m+1;
    }
    return l;
}

vector<int> longest_trip(int N, int D){
    perm.resize(N);
    iota(perm.begin(),perm.end(),0);
    shuffle(perm.begin(),perm.end(),mt19937_64());
    if(D==3)return perm;
    if(D==2){
        deque<int> path;
        for(int i=0;i<3;i++)path.emplace_back(i);
        for(int i=1;i<3;i++){
            if(!ask({i-1},{i})){
                swap(path[0],path[i-1]);
                swap(path[2],path[i]);
                break;
            }
        }
        for(int i=3;i<N;i++){
            if(ask({i},{path[0]})){
                path.emplace_front(i);
            }else{
                path.emplace_back(i);
            }
        }
        vector<int> ans;
        for(auto x:path)ans.emplace_back(x);
        return ans;
    }
    vector<int> path[2];
    for(int i=0;i<2;i++)path[i].emplace_back(i);
    for(int i=2;i<N;i++){
        if(ask({i},{path[0].back()})){
            path[0].emplace_back(i);
            continue;
        }
        if(ask({i},{path[1].back()})){
            path[1].emplace_back(i);
            continue;
        }
        while(!path[1].empty()){
            path[0].emplace_back(path[1].back());
            path[1].pop_back();
        }
        path[1].emplace_back(i);
    }
    if(path[0].size()<path[1].size())swap(path[0],path[1]);
    if(!ask(path[0],path[1]))return path[0];
    if(ask({path[0][0]},{path[1][0]})){
        reverse(path[1].begin(),path[1].end());
        auto ans=path[1];
        for(auto x:path[0])ans.emplace_back(x);
        return ans;
    }
    if(ask({path[0][0]},{path[1].back()})){
        auto ans=path[1];
        for(auto x:path[0])ans.emplace_back(x);
        return ans;
    }
    if(ask({path[0].back()},{path[1][0]})){
        auto ans=path[0];
        for(auto x:path[1])ans.emplace_back(x);
        return ans;
    }
    if(ask({path[0].back()},{path[1].back()})){
        auto ans=path[0];
        reverse(path[1].begin(),path[1].end());
        for(auto x:path[1])ans.emplace_back(x);
        return ans;
    }
    int id0=bsearch(path[0],path[1]);
    int id1=bsearch(path[1],{path[0][id0]});
    vector<int> ans;
    for(int i=0;i<path[0].size();i++)ans.emplace_back(path[0][(id0+i+1)%path[0].size()]);
    for(int i=0;i<path[1].size();i++)ans.emplace_back(path[1][(id1+i)%path[1].size()]);
    for(auto &x:ans)x=perm[x];
    return ans;
}

Compilation message

longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:96:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   96 |     for(int i=0;i<path[0].size();i++)ans.emplace_back(path[0][(id0+i+1)%path[0].size()]);
      |                 ~^~~~~~~~~~~~~~~
longesttrip.cpp:97:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   97 |     for(int i=0;i<path[1].size();i++)ans.emplace_back(path[1][(id1+i)%path[1].size()]);
      |                 ~^~~~~~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 344 KB Incorrect
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 512 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 4 ms 344 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Incorrect 0 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 9 ms 344 KB Output is correct
2 Correct 5 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 4 ms 344 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Incorrect 0 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 7 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 4 ms 344 KB Output is correct
6 Incorrect 1 ms 344 KB Incorrect
7 Halted 0 ms 0 KB -