oj.uz

Submission #416787

# Submission time Handle Problem Language Result Execution time Memory
416787 2021-06-02T23:40:31 Z duality Counting Mushrooms (IOI20_mushrooms) C++14
100 / 100
 9 ms 456 KB 
#define DEBUG 0

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

#if DEBUG
// basic debugging macros
int __i__,__j__;
#define printLine(l) for(__i__=0;__i__<l;__i__++){cout<<"-";}cout<<endl
#define printLine2(l,c) for(__i__=0;__i__<l;__i__++){cout<<c;}cout<<endl
#define printVar(n) cout<<#n<<": "<<n<<endl
#define printArr(a,l) cout<<#a<<": ";for(__i__=0;__i__<l;__i__++){cout<<a[__i__]<<" ";}cout<<endl
#define print2dArr(a,r,c) cout<<#a<<":\n";for(__i__=0;__i__<r;__i__++){for(__j__=0;__j__<c;__j__++){cout<<a[__i__][__j__]<<" ";}cout<<endl;}
#define print2dArr2(a,r,c,l) cout<<#a<<":\n";for(__i__=0;__i__<r;__i__++){for(__j__=0;__j__<c;__j__++){cout<<setw(l)<<setfill(' ')<<a[__i__][__j__]<<" ";}cout<<endl;}

// advanced debugging class
// debug 1,2,'A',"test";
class _Debug {
    public:
        template<typename T>
        _Debug& operator,(T val) {
            cout << val << endl;
            return *this;
        }
};
#define debug _Debug(),
#else
#define printLine(l)
#define printLine2(l,c)
#define printVar(n)
#define printArr(a,l)
#define print2dArr(a,r,c)
#define print2dArr2(a,r,c,l)
#define debug
#endif

// define
#define MAX_VAL 999999999
#define MAX_VAL_2 999999999999999999LL
#define EPS 1e-6
#define mp make_pair
#define pb push_back

// typedef
typedef unsigned int UI;
typedef long long int LLI;
typedef unsigned long long int ULLI;
typedef unsigned short int US;
typedef pair<int,int> pii;
typedef pair<LLI,LLI> plli;
typedef vector<int> vi;
typedef vector<LLI> vlli;
typedef vector<pii> vpii;
typedef vector<plli> vplli;

// ---------- END OF TEMPLATE ----------
#include "mushrooms.h"

int query(vi v) {
    return use_machine(v);
}
int count_mushrooms(int n) {
    int i,j,ans = 0;
    if (n <= 220) {
        for (i = 1; i < n; i++) ans += !use_machine({0,i});
        return ans+1;
    }
    vi v,a,b;
    a.pb(0);
    for (i = 1; i <= 2; i++) {
        if (query({0,i})) b.pb(i);
        else a.pb(i);
    }
    if (a.size() >= 2) {
        int r = query({3,a[0],4,a[1]});
        if (r & 1) b.pb(3);
        else a.pb(3);
        if (r & 2) b.pb(4);
        else a.pb(4);
    }
    else {
        int r = query({3,b[0],4,b[1]});
        if (r & 1) a.pb(3);
        else b.pb(3);
        if (r & 2) a.pb(4);
        else b.pb(4);
    }
    int s = 0,x,y,z;
    for (i = 5; i < n; i++) {
        v.pb(i);
        if ((max(a.size(),b.size()) < 105) || (s != 0)) {
            if ((s == 0) && (v.size() == 3)) {
                if (a.size() >= 3) {
                    int r = query({v[0],a[0],v[1],a[1],v[2],a[2]});
                    if (r & 1) b.pb(v[0]);
                    else a.pb(v[0]);
                    if (r/2 == 0) a.pb(v[1]),a.pb(v[2]);
                    else if (r/2 == 2) b.pb(v[1]),b.pb(v[2]);
                    else s = 1,x = v[1],y = v[2];
                }
                else {
                    int r = query({v[0],b[0],v[1],b[1],v[2],b[2]});
                    if (r & 1) a.pb(v[0]);
                    else b.pb(v[0]);
                    if (r/2 == 0) b.pb(v[1]),b.pb(v[2]);
                    else if (r/2 == 2) a.pb(v[1]),a.pb(v[2]);
                    else s = 1,x = v[1],y = v[2];
                }
                v.clear();
            }
            else if ((s == 1) && (v.size() == 2)) {
                if ((a.size() >= 3) && (b.size() >= 2)) {
                    int r = query({v[0],a[0],v[1],a[1],x,a[2],b[0],y,b[1]})-1;
                    if (r & 1) b.pb(v[0]);
                    else a.pb(v[0]);
                    if (r & 2) b.pb(v[1]);
                    else a.pb(v[1]);
                    if (r & 4) b.pb(x),a.pb(y),s = 0;
                    else a.pb(x),b.pb(y),s = 0;
                }
                else if ((a.size() >= 2) && (b.size() >= 3)) {
                    int r = query({v[0],b[0],v[1],b[1],x,b[2],a[0],y,a[1]})-1;
                    if (r & 1) a.pb(v[0]);
                    else b.pb(v[0]);
                    if (r & 2) a.pb(v[1]);
                    else b.pb(v[1]);
                    if (r & 4) a.pb(x),b.pb(y),s = 0;
                    else b.pb(x),a.pb(y),s = 0;
                }
                else if (a.size() >= 3) {
                    int r = query({v[0],a[0],v[1],a[1],x,a[2]});
                    if (r & 1) b.pb(v[0]);
                    else a.pb(v[0]);
                    if (r/2 == 0) a.pb(v[1]),a.pb(x),b.pb(y),s = 0;
                    else if (r/2 == 2) b.pb(v[1]),b.pb(x),a.pb(y),s = 0;
                    else s = 2,z = v[1];
                }
                else {
                    int r = query({v[0],b[0],v[1],b[1],x,b[2]});
                    if (r & 1) a.pb(v[0]);
                    else b.pb(v[0]);
                    if (r/2 == 0) b.pb(v[1]),b.pb(x),a.pb(y),s = 0;
                    else if (r/2 == 2) a.pb(v[1]),a.pb(x),b.pb(y),s = 0;
                    else s = 2,z = v[1];
                }
                v.clear();
            }
            else if ((s == 2) && (v.size() == 2)) {
                if (a.size() >= 4) {
                    int r = query({v[0],a[0],v[1],a[1],y,a[2],z,a[3]});
                    if (r & 1) b.pb(v[0]);
                    else a.pb(v[0]);
                    if (r & 2) b.pb(v[1]);
                    else a.pb(v[1]);
                    if (r & 4) b.pb(y),b.pb(z),a.pb(x),s = 0;
                    else a.pb(y),a.pb(z),b.pb(x),s = 0;
                }
                else {
                    int r = query({v[0],b[0],v[1],b[1],y,b[2],z,b[3]});
                    if (r & 1) a.pb(v[0]);
                    else b.pb(v[0]);
                    if (r & 2) a.pb(v[1]);
                    else b.pb(v[1]);
                    if (r & 4) a.pb(y),a.pb(z),b.pb(x),s = 0;
                    else b.pb(y),b.pb(z),a.pb(x),s = 0;
                }
                v.clear();
            }
        }
        else if (v.size() == max(a.size(),b.size())) {
            if (a.size() > b.size()) {
                vi q;
                for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(a[j]);
                int r = query(q);
                ans += v.size()-r/2-1;
                if (r & 1) b.pb(v[0]);
                else a.pb(v[0]);
            }
            else {
                vi q;
                for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(b[j]);
                int r = query(q);
                ans += r/2;
                if (r & 1) a.pb(v[0]);
                else b.pb(v[0]);
            }
            v.clear();
        }
    }
    if (!v.empty()) {
        if (a.size() > b.size()) {
            vi q;
            for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(a[j]);
            int r = query(q);
            ans += v.size()-r/2-1;
            if (r & 1) b.pb(v[0]);
            else a.pb(v[0]);
        }
        else {
            vi q;
            for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(b[j]);
            int r = query(q);
            ans += r/2;
            if (r & 1) a.pb(v[0]);
            else b.pb(v[0]);
        }
        v.clear();
    }
    return a.size()+ans;
}

Compilation message

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:173:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  173 |                 for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(a[j]);
      |                             ~~^~~~~~~~~~
mushrooms.cpp:181:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  181 |                 for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(b[j]);
      |                             ~~^~~~~~~~~~
mushrooms.cpp:193:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  193 |             for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(a[j]);
      |                         ~~^~~~~~~~~~
mushrooms.cpp:201:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  201 |             for (j = 0; j < v.size(); j++) q.pb(v[j]),q.pb(b[j]);
      |                         ~~^~~~~~~~~~

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 200 KB Output is correct
3 Correct 1 ms 200 KB Output is correct
4 Correct 1 ms 200 KB Output is correct
5 Correct 2 ms 200 KB Output is correct
6 Correct 2 ms 200 KB Output is correct
7 Correct 6 ms 200 KB Output is correct
8 Correct 7 ms 200 KB Output is correct
9 Correct 6 ms 328 KB Output is correct
10 Correct 7 ms 328 KB Output is correct
11 Correct 6 ms 328 KB Output is correct
12 Correct 8 ms 324 KB Output is correct
13 Correct 6 ms 328 KB Output is correct
14 Correct 4 ms 328 KB Output is correct
15 Correct 9 ms 200 KB Output is correct
16 Correct 7 ms 200 KB Output is correct
17 Correct 4 ms 200 KB Output is correct
18 Correct 6 ms 320 KB Output is correct
19 Correct 7 ms 312 KB Output is correct
20 Correct 6 ms 328 KB Output is correct
21 Correct 7 ms 200 KB Output is correct
22 Correct 7 ms 328 KB Output is correct
23 Correct 9 ms 296 KB Output is correct
24 Correct 6 ms 328 KB Output is correct
25 Correct 7 ms 320 KB Output is correct
26 Correct 6 ms 328 KB Output is correct
27 Correct 7 ms 328 KB Output is correct
28 Correct 7 ms 312 KB Output is correct
29 Correct 7 ms 324 KB Output is correct
30 Correct 7 ms 328 KB Output is correct
31 Correct 7 ms 328 KB Output is correct
32 Correct 9 ms 200 KB Output is correct
33 Correct 7 ms 312 KB Output is correct
34 Correct 6 ms 328 KB Output is correct
35 Correct 7 ms 200 KB Output is correct
36 Correct 7 ms 312 KB Output is correct
37 Correct 7 ms 328 KB Output is correct
38 Correct 7 ms 328 KB Output is correct
39 Correct 9 ms 304 KB Output is correct
40 Correct 7 ms 328 KB Output is correct
41 Correct 7 ms 328 KB Output is correct
42 Correct 9 ms 328 KB Output is correct
43 Correct 7 ms 328 KB Output is correct
44 Correct 7 ms 328 KB Output is correct
45 Correct 7 ms 456 KB Output is correct
46 Correct 9 ms 328 KB Output is correct
47 Correct 7 ms 328 KB Output is correct
48 Correct 7 ms 352 KB Output is correct
49 Correct 7 ms 328 KB Output is correct
50 Correct 7 ms 328 KB Output is correct
51 Correct 7 ms 332 KB Output is correct
52 Correct 7 ms 328 KB Output is correct
53 Correct 7 ms 328 KB Output is correct
54 Correct 7 ms 328 KB Output is correct
55 Correct 7 ms 328 KB Output is correct
56 Correct 9 ms 328 KB Output is correct
57 Correct 7 ms 328 KB Output is correct
58 Correct 7 ms 328 KB Output is correct
59 Correct 8 ms 328 KB Output is correct
60 Correct 8 ms 328 KB Output is correct
61 Correct 7 ms 300 KB Output is correct
62 Correct 1 ms 200 KB Output is correct
63 Correct 1 ms 200 KB Output is correct
64 Correct 1 ms 200 KB Output is correct
65 Correct 1 ms 200 KB Output is correct
66 Correct 1 ms 200 KB Output is correct
67 Correct 1 ms 200 KB Output is correct
68 Correct 1 ms 200 KB Output is correct
69 Correct 1 ms 200 KB Output is correct
70 Correct 1 ms 200 KB Output is correct
71 Correct 1 ms 200 KB Output is correct
72 Correct 1 ms 200 KB Output is correct
73 Correct 0 ms 200 KB Output is correct
74 Correct 1 ms 200 KB Output is correct
75 Correct 1 ms 200 KB Output is correct
76 Correct 1 ms 200 KB Output is correct
77 Correct 1 ms 200 KB Output is correct
78 Correct 1 ms 200 KB Output is correct
79 Correct 1 ms 200 KB Output is correct
80 Correct 1 ms 200 KB Output is correct
81 Correct 1 ms 200 KB Output is correct
82 Correct 1 ms 200 KB Output is correct
83 Correct 1 ms 200 KB Output is correct
84 Correct 1 ms 200 KB Output is correct
85 Correct 0 ms 200 KB Output is correct
86 Correct 1 ms 200 KB Output is correct
87 Correct 1 ms 200 KB Output is correct
88 Correct 1 ms 200 KB Output is correct
89 Correct 1 ms 200 KB Output is correct
90 Correct 0 ms 200 KB Output is correct
91 Correct 1 ms 200 KB Output is correct