제출 #1370833

#제출 시각아이디문제언어결과실행 시간메모리
1370833huangallenLongest Trip (IOI23_longesttrip)C++20
85 / 100
5 ms452 KiB
#include "longesttrip.h"
#ifdef LOCAL
#include "grader.cpp"
#endif
#include<bits/stdc++.h>
using namespace std;
// #pragma GCC optimize("O3,unroll-loops")
// #pragma GCC target("avx2,sse4,bmi2,popcnt")
#define int long long
#define iint signed
#define REP(i,n) for(int i=0;i<(n);i++)
#define REP1(i,n) for(int i=1;i<=(n);i++)
#define RREP(i,n) for(int i=(n)-1;i>=0;i--)
#define RREP1(i,n) for(int i=(n);i>=1;i--)
#define f first
#define s second
#define pb push_back
#define ALL(x) (x).begin(),(x).end()
#define SZ(x) ((int)((x).size()))
#define SQ(x) ((x)*(x))
#define pii pair<int,int>
#define pipii pair<int,pii>
#define ppi pair<pii,int>
#define Graph vector<vector<int>>
#define Graphw vector<vector<pii>>
#define IOS() ios::sync_with_stdio(0),cin.tie(0)
#define md(x) (((x)%(mod)+(mod))%(mod))
#define MD(x,M) (((x)%(M)+(M))%(M))
#define ld long double
#define pdd pair<ld,ld>
template<typename S> void chmin(S &x,S y) { x=min(x,y); }
template<typename S> void chmax(S &x,S y) { x=max(x,y); }
#define addmod(x,y) x=((x+(y))%mod)
#define Vi vector<int>
#define Vpii vector<pii>
#ifdef LOCAL
#define op(x) cout<<(#x)<<"="<<(x)<<", ";
#define ope(x) cout<<(#x)<<"="<<(x)<<endl;
#define oparr(x) {cout<<(#x)<<":";for(auto allen:(x)) cout<<allen<<" ";cout<<" size="<<(x).size()<<endl;}
#define entr cout<<endl;
#else
#define op(x) ;
#define ope(x) ;
#define oparr(x) ;
#define entr ;
#endif
template<typename T1,typename T2>
ostream& operator<<(ostream& os,pair<T1,T2> p) { return os<<'{'<<p.f<<','<<p.s<<'}'; }
template<typename T1,typename T2>
istream& operator>>(istream& os,pair<T1,T2> &p) { return os>>p.f>>p.s; }
template<typename S>
ostream& operator<<(ostream& os,vector<S> p) { for(auto allen:p) os<<allen<<' ';return os<<'\n'; }
template<typename S>
istream& operator>>(istream& os,vector<S> &p) { for(auto &allen:p) os>>allen;return os; }
template<typename T1,typename T2>
pair<T1,T2> operator+(pair<T1,T2> p1,pair<T1,T2> p2) { return pair<T1,T2>(p1.f+p2.f,p1.s+p2.s); }
const int mod=998244353;
const int maxn=1e5+5;
const int maxv=1e6+5;
const int inf=1ll<<60;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int rd(int l,int r) {
    return uniform_int_distribution<int>(l,r)(rng);
}
#define Vii vector<iint>
Vi perm;
bool ac(Vii a,Vii b) {
    for(auto &x:a) x=perm[x];
    for(auto &x:b) x=perm[x];
    return are_connected(a,b);
}
std::vector<iint> longest_trip(iint N, iint D)
{
    int n=N,d=D;
    perm=Vi(n);
    REP(i,n) perm[i]=i;
    shuffle(ALL(perm),rng);
    if(N==2) {
        if(are_connected({0},{1})) return {0,1};
        else return {0};
    }
    Vi an1={0};
    Vi an2={1};
    int l1=0,r1=0,l2=1,r2=1;
    for(int i=2;i<n;i++) {
        if(ac({r1},{r2})) {
            reverse(ALL(an2));
            for(int x:an2) an1.pb(x);
            an2={i};
            r1=an1.back();
            l2=r2=i;
        }else if(ac({r1},{i})){
            an1.pb(i);
            r1=i;
        }else {
            an2.pb(i);
            r2=i;
        }
    }
    if(ac({an1.back()},{an2[0]})) {
        Vii an;
        for(int x:an1) an.pb(x);
        for(int x:an2) an.pb(x);
        for(auto &x:an) x=perm[x];
        return an;
    }
    if(ac({an1[0]},{an2[0]})) {
        Vii an;
        reverse(ALL(an1));
        for(int x:an1) an.pb(x);
        for(int x:an2) an.pb(x);
        for(auto &x:an) x=perm[x];
        return an;
    }
    if(ac({an1.back()},{an2.back()})) {
        reverse(ALL(an2));
        Vii an;
        for(int x:an1) an.pb(x);
        for(int x:an2) an.pb(x);
        for(auto &x:an) x=perm[x];
        return an;
    }
    Vii s1(SZ(an1)),s2(SZ(an2));
    REP(i,SZ(an1)) s1[i]=an1[i];
    REP(i,SZ(an2)) s2[i]=an2[i];
    // return SZ(s1)>SZ(s2)?s1:s2;
    if(!ac(s1,s2)) {
        Vii an=SZ(s1)>SZ(s2)?s1:s2;
        for(auto &x:an) x=perm[x];
        return an;
    }
    int id1,id2;
    {
        int l=0,r=SZ(s1)-1,m;
        while(l<r) {
            m=l+r>>1;
            Vii t;
            for(int i=0;i<=m;i++) t.pb(s1[i]);
            if(ac(t,s2)) r=m;
            else l=m+1;
        }
        id1=l;
    }
    // Vii t;REP(i,id1+1) t.pb(s1[i]);
    // assert(are_connected(t,s2)); 
    // t.pop_back();
    // assert(!are_connected(t,s2)); 
    {
        // assert(are_connected(s2,{id1}));
        int l=0,r=SZ(s2)-1,m;
        while(l<r) {
            m=l+r>>1;
            Vii t;
            for(int i=0;i<=m;i++) t.pb(s2[i]);
            if(ac(t,{s1[id1]})) r=m;
            else l=m+1;
        }
        id2=l;
    }
    // op(id1)ope(s1[id1])
    // op(id2)ope(s2[id2])
    Vii an;
    REP(i,SZ(s1)) an.pb(s1[(id1+i+1)%SZ(s1)]);
    REP(i,SZ(s2)) an.pb(s2[(id2+i)%SZ(s2)]);
    // Vii _an
    for(auto &x:an) x=perm[x];
    return an;
}

컴파일 시 표준 에러 (stderr) 메시지

longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:86:16: warning: narrowing conversion of 'r1' from 'long long int' to 'int' [-Wnarrowing]
   86 |         if(ac({r1},{r2})) {
      |                ^~
longesttrip.cpp:86:16: warning: narrowing conversion of 'r1' from 'long long int' to 'int' [-Wnarrowing]
longesttrip.cpp:86:21: warning: narrowing conversion of 'r2' from 'long long int' to 'int' [-Wnarrowing]
   86 |         if(ac({r1},{r2})) {
      |                     ^~
longesttrip.cpp:86:21: warning: narrowing conversion of 'r2' from 'long long int' to 'int' [-Wnarrowing]
longesttrip.cpp:92:22: warning: narrowing conversion of 'r1' from 'long long int' to 'int' [-Wnarrowing]
   92 |         }else if(ac({r1},{i})){
      |                      ^~
longesttrip.cpp:92:22: warning: narrowing conversion of 'r1' from 'long long int' to 'int' [-Wnarrowing]
longesttrip.cpp:92:27: warning: narrowing conversion of 'i' from 'long long int' to 'int' [-Wnarrowing]
   92 |         }else if(ac({r1},{i})){
      |                           ^
longesttrip.cpp:92:27: warning: narrowing conversion of 'i' from 'long long int' to 'int' [-Wnarrowing]
longesttrip.cpp:100:20: warning: narrowing conversion of 'an1.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
  100 |     if(ac({an1.back()},{an2[0]})) {
      |            ~~~~~~~~^~
longesttrip.cpp:100:20: warning: narrowing conversion of 'an1.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:100:10: warning: narrowing conversion of 'an2.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
  100 |     if(ac({an1.back()},{an2[0]})) {
      |        ~~^~~~~~~~~~~~~~~~~~~~~~~
longesttrip.cpp:100:10: warning: narrowing conversion of 'an2.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:107:10: warning: narrowing conversion of 'an1.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
  107 |     if(ac({an1[0]},{an2[0]})) {
      |        ~~^~~~~~~~~~~~~~~~~~~
longesttrip.cpp:107:10: warning: narrowing conversion of 'an1.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:107:10: warning: narrowing conversion of 'an2.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:107:10: warning: narrowing conversion of 'an2.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:115:20: warning: narrowing conversion of 'an1.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
  115 |     if(ac({an1.back()},{an2.back()})) {
      |            ~~~~~~~~^~
longesttrip.cpp:115:20: warning: narrowing conversion of 'an1.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
longesttrip.cpp:115:33: warning: narrowing conversion of 'an2.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
  115 |     if(ac({an1.back()},{an2.back()})) {
      |                         ~~~~~~~~^~
longesttrip.cpp:115:33: warning: narrowing conversion of 'an2.std::vector<long long int>::back()' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…