oj.uz

Submission #841637

# Submission time Handle Problem Language Result Execution time Memory
841637 2023-09-01T18:56:48 Z Pajaraja Longest Trip (IOI23_longesttrip) C++17
100 / 100
 11 ms 760 KB 
#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
int broj;
pair<int,int> findedge(vector<int> a,vector<int> b)
{
    int l=0,r=b.size()-1;
    while(l!=r)
    {
        int s=(l+r)/2;
        vector<int> q;
        for(int i=l;i<=s;i++) q.push_back(b[i]);
        if(are_connected(a,q)) r=s;
        else l=s+1;
    }
    int ind=l;
    vector<int> d;
    d.push_back(b[l]);
    l=0; r=a.size()-1;
    while(l!=r)
    {
        int s=(l+r)/2;
        vector<int> q;
        for(int i=l;i<=s;i++) q.push_back(a[i]);
        if(are_connected(q,d)) r=s;
        else l=s+1;
    }
    return {l,ind};
}
mt19937 rng(time(0));
bool ask(int a,int b)
{
    broj++;
    vector<int> a1,b1;
    a1.push_back(a);
    b1.push_back(b);
    return are_connected(a1,b1);
}
std::vector<int> longest_trip(int N, int D)
{
    broj=0;
    vector<int> cur1,cur2;
    vector<int> all;
    for(int i=0;i<N;i++) all.push_back(i);
    shuffle(all.begin(),all.end(),rng);
    cur1.push_back(all[0]);
    bool nema=true;
    for(int i=1;i<N;i++)
    {
        int x=all[i];
        if(ask(cur1.back(),x))
        {
            cur1.push_back(x);
            nema=false;
            if(cur2.size()==0) nema=true;
        }
        else
        {
            if(nema) cur2.push_back(x);
            else
            {
                if(!ask(cur2.back(),x))
                {
                    if(cur2.size()==1) nema=true;
                    else if(!ask(cur2[0],x)) nema=true;
                    for(int i=cur2.size()-1;i>=0;i--) cur1.push_back(cur2[i]);
                    cur2.clear();
                    if(nema)  cur2.push_back(x);
                    else cur1.push_back(x);
                    nema=true;
                }
                else
                {
                    cur2.push_back(x);
                    nema=true;
                }
            }
        }
    }
    if(cur1.size()<cur2.size()) swap(cur1,cur2);
    if(cur2.size()==0) return cur1;
    if(!are_connected(cur1,cur2))
    {
        if(cur1.size()<cur2.size()) return cur2;
        return cur1;
    }
    if(cur2.size()==1)
    {
        if(ask(cur1.back(),cur2.back()))
        {
            cur1.push_back(cur2.back());
            return cur1;
        }
        pair<int,int> p=findedge(cur1,cur2);
        vector<int> ans;
        ans.push_back(cur2[0]);
        int c=p.first;
        do{
            ans.push_back(cur1[c]);
            c++;
            c%=cur1.size();
        } while(c!=p.first);
        return ans;
    }
    int al=cur1[0],ar=cur1.back(),bl=cur2[0],br=cur2.back();
    if(are_connected({al,ar},{bl,br}))
    {
        vector<int> ans;
        if(ask(al,bl))
        {
            reverse(cur1.begin(),cur1.end());
            for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
            for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
            return ans;
        }
        if(ask(al,br))
        {
            reverse(cur1.begin(),cur1.end());
            reverse(cur2.begin(),cur2.end());
            for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
            for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
            return ans;
        }
        if(ask(ar,bl))
        {
            for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
            for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
            return ans;
        }
        if(ask(ar,br))
        {
            reverse(cur2.begin(),cur2.end());
            for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
            for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
            return ans;
        }
    }
    pair<int,int> p=findedge(cur1,cur2);
    vector<int> ans;
    int c=p.second;
    do
    {
        c++;
        c%=cur2.size();
        ans.push_back(cur2[c]);
    } while(c!=p.second);
    c=p.first;
    do
    {
        ans.push_back(cur1[c]);
        c++;
        c%=cur1.size();
    } while(c!=p.first);
    return ans;
}

Compilation message

longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:112:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  112 |             for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:113:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  113 |             for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:120:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  120 |             for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:121:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  121 |             for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:126:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  126 |             for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:127:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  127 |             for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:133:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  133 |             for(int i=0;i<cur1.size();i++) ans.push_back(cur1[i]);
      |                         ~^~~~~~~~~~~~
longesttrip.cpp:134:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  134 |             for(int i=0;i<cur2.size();i++) ans.push_back(cur2[i]);
      |                         ~^~~~~~~~~~~~

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 596 KB Output is correct
2 Correct 1 ms 344 KB Output is correct

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 5 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 5 ms 344 KB Output is correct
5 Correct 5 ms 344 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 5 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 4 ms 344 KB Output is correct
6 Correct 6 ms 344 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 5 ms 344 KB Output is correct
9 Correct 4 ms 600 KB Output is correct
10 Correct 4 ms 344 KB Output is correct
11 Correct 4 ms 596 KB Output is correct
12 Correct 4 ms 344 KB Output is correct
13 Correct 4 ms 596 KB Output is correct

Subtask #3
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 344 KB Output is correct
2 Correct 4 ms 344 KB Output is correct
3 Correct 4 ms 344 KB Output is correct
4 Correct 4 ms 344 KB Output is correct
5 Correct 4 ms 344 KB Output is correct
6 Correct 8 ms 344 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 4 ms 344 KB Output is correct
9 Correct 5 ms 344 KB Output is correct
10 Correct 4 ms 344 KB Output is correct
11 Correct 5 ms 344 KB Output is correct
12 Correct 4 ms 344 KB Output is correct
13 Correct 5 ms 344 KB Output is correct
14 Correct 7 ms 344 KB Output is correct
15 Correct 6 ms 344 KB Output is correct
16 Correct 6 ms 344 KB Output is correct
17 Correct 6 ms 344 KB Output is correct
18 Correct 5 ms 600 KB Output is correct
19 Correct 5 ms 344 KB Output is correct
20 Correct 6 ms 600 KB Output is correct
21 Correct 4 ms 440 KB Output is correct
22 Correct 6 ms 440 KB Output is correct
23 Correct 5 ms 344 KB Output is correct
24 Correct 5 ms 440 KB Output is correct
25 Correct 6 ms 596 KB Output is correct
26 Correct 6 ms 344 KB Output is correct
27 Correct 5 ms 344 KB Output is correct
28 Correct 5 ms 344 KB Output is correct
29 Correct 5 ms 344 KB Output is correct
30 Correct 4 ms 344 KB Output is correct
31 Correct 5 ms 344 KB Output is correct
32 Correct 5 ms 344 KB Output is correct
33 Correct 4 ms 344 KB Output is correct
34 Correct 4 ms 344 KB Output is correct
35 Correct 5 ms 344 KB Output is correct
36 Correct 4 ms 344 KB Output is correct
37 Correct 4 ms 344 KB Output is correct
38 Correct 7 ms 344 KB Output is correct
39 Correct 6 ms 344 KB Output is correct
40 Correct 5 ms 344 KB Output is correct
41 Correct 5 ms 344 KB Output is correct
42 Correct 5 ms 344 KB Output is correct
43 Correct 6 ms 344 KB Output is correct
44 Correct 7 ms 344 KB Output is correct
45 Correct 9 ms 344 KB Output is correct
46 Correct 11 ms 344 KB Output is correct
47 Correct 6 ms 344 KB Output is correct
48 Correct 8 ms 344 KB Output is correct
49 Correct 6 ms 344 KB Output is correct
50 Correct 5 ms 344 KB Output is correct
51 Correct 6 ms 436 KB Output is correct
52 Correct 6 ms 344 KB Output is correct
53 Correct 5 ms 344 KB Output is correct
54 Correct 5 ms 344 KB Output is correct
55 Correct 5 ms 344 KB Output is correct
56 Correct 5 ms 344 KB Output is correct
57 Correct 7 ms 600 KB Output is correct
58 Correct 6 ms 448 KB Output is correct
59 Correct 6 ms 600 KB Output is correct
60 Correct 6 ms 448 KB Output is correct
61 Correct 6 ms 344 KB Output is correct
62 Correct 7 ms 344 KB Output is correct
63 Correct 7 ms 448 KB Output is correct
64 Correct 6 ms 600 KB Output is correct

Subtask #4
60.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 5 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 6 ms 344 KB Output is correct
6 Correct 9 ms 344 KB Output is correct
7 Correct 5 ms 344 KB Output is correct
8 Correct 4 ms 344 KB Output is correct
9 Correct 5 ms 344 KB Output is correct
10 Correct 4 ms 344 KB Output is correct
11 Correct 5 ms 344 KB Output is correct
12 Correct 4 ms 344 KB Output is correct
13 Correct 4 ms 344 KB Output is correct
14 Correct 7 ms 344 KB Output is correct
15 Correct 6 ms 596 KB Output is correct
16 Correct 9 ms 344 KB Output is correct
17 Correct 6 ms 344 KB Output is correct
18 Correct 6 ms 600 KB Output is correct
19 Correct 5 ms 344 KB Output is correct
20 Correct 5 ms 344 KB Output is correct
21 Correct 6 ms 344 KB Output is correct
22 Correct 6 ms 344 KB Output is correct
23 Correct 5 ms 344 KB Output is correct
24 Correct 5 ms 344 KB Output is correct
25 Correct 5 ms 344 KB Output is correct
26 Correct 4 ms 344 KB Output is correct
27 Correct 5 ms 344 KB Output is correct
28 Correct 6 ms 344 KB Output is correct
29 Correct 5 ms 344 KB Output is correct
30 Correct 5 ms 344 KB Output is correct
31 Correct 5 ms 344 KB Output is correct
32 Correct 9 ms 344 KB Output is correct
33 Correct 11 ms 344 KB Output is correct
34 Correct 8 ms 344 KB Output is correct
35 Correct 9 ms 344 KB Output is correct
36 Correct 10 ms 344 KB Output is correct
37 Correct 5 ms 344 KB Output is correct
38 Correct 7 ms 444 KB Output is correct
39 Correct 7 ms 344 KB Output is correct
40 Correct 5 ms 344 KB Output is correct
41 Correct 5 ms 344 KB Output is correct
42 Correct 6 ms 600 KB Output is correct
43 Correct 5 ms 444 KB Output is correct
44 Correct 5 ms 600 KB Output is correct
45 Correct 5 ms 448 KB Output is correct
46 Correct 5 ms 448 KB Output is correct
47 Correct 4 ms 344 KB Output is correct
48 Correct 5 ms 344 KB Output is correct
49 Correct 5 ms 344 KB Output is correct
50 Correct 6 ms 344 KB Output is correct
51 Correct 6 ms 344 KB Output is correct
52 Correct 5 ms 344 KB Output is correct
53 Correct 6 ms 344 KB Output is correct
54 Correct 5 ms 344 KB Output is correct
55 Correct 5 ms 344 KB Output is correct
56 Correct 5 ms 344 KB Output is correct
57 Correct 5 ms 344 KB Output is correct
58 Correct 6 ms 344 KB Output is correct
59 Correct 6 ms 540 KB Output is correct
60 Correct 5 ms 344 KB Output is correct
61 Correct 6 ms 344 KB Output is correct
62 Correct 4 ms 344 KB Output is correct
63 Correct 5 ms 440 KB Output is correct
64 Correct 5 ms 600 KB Output is correct
65 Correct 6 ms 760 KB Output is correct
66 Correct 7 ms 600 KB Output is correct
67 Correct 6 ms 344 KB Output is correct
68 Correct 7 ms 344 KB Output is correct
69 Correct 6 ms 444 KB Output is correct
70 Correct 6 ms 448 KB Output is correct
71 Correct 5 ms 448 KB Output is correct
72 Correct 6 ms 600 KB Output is correct
73 Correct 6 ms 600 KB Output is correct
74 Correct 6 ms 448 KB Output is correct
75 Correct 6 ms 344 KB Output is correct
76 Correct 9 ms 344 KB Output is correct
77 Correct 5 ms 444 KB Output is correct
78 Correct 5 ms 600 KB Output is correct
79 Correct 7 ms 444 KB Output is correct
80 Correct 6 ms 600 KB Output is correct
81 Correct 7 ms 344 KB Output is correct
82 Correct 6 ms 344 KB Output is correct