Submission #1261489

#	Time	Username	Problem	Language	Result	Execution time	Memory
1261489		Zbyszek99	Worm Worries (BOI18_worm)	C++20	100 / 100	268 ms	5340 KiB

worm

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long
#define ld long double
#define ull unsigned long long
#define ff first
#define ss second
#define pii pair<int,int>
#define pll pair<long long, long long>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define rep(i, b) for(int i = 0; i < (b); ++i)
#define rep2(i,a,b) for(int i = a; i <= (b); ++i)
#define rep3(i,a,b,c) for(int i = a; i <= (b); i+=c)
#define count_bits(x) __builtin_popcountll((x))
#define all(x) (x).begin(),(x).end()
#define siz(x) (int)(x).size()
#define forall(it,x) for(auto& it:(x))
using namespace __gnu_pbds;
using namespace std;
typedef tree<int, null_type, less<int>, rb_tree_tag,tree_order_statistics_node_update> ordered_set;
mt19937 mt;void random_start(){mt.seed(chrono::time_point_cast<chrono::milliseconds>(chrono::high_resolution_clock::now()).time_since_epoch().count());}
ll los(ll a, ll b) {return a + (mt() % (b-a+1));}
const int INF = 1e9+50;
const ll INF_L = 1e18+40;
const ll MOD = 1e9+7;

ld c1 = ((ld)sqrt(5)-1)/2;
ld c2 = 1-c1;

struct point
{
    int x,y,z;
    bool operator<(const point& other) const
    {
        return x < other.x;
    }
};

ll n,m,k,q;
map<ll,int> query_map;

int ask(int x, int y = 0, int z = 0)
{
    x++;
    y++;
    z++;
    if(x < 1 || x > n || y < 1 || y > m || z < 1 || z > k) return 0;
    int h = x-1 + (y-1)*n + (z-1)*n*m;
    if(query_map.find(h) != query_map.end()) return query_map[h];
    cout << "? " << x << " " << y << " " << z << endl;
    cin >> query_map[h];
    return query_map[h];
}

bool check(point p)
{
    int x = p.x;
    int y = p.y;
    int z = p.z;
    return ask(x,y,z) >= max({ask(x-1,y,z),ask(x+1,y,z),ask(x,y-1,z),ask(x,y+1,z),ask(x,y,z-1),ask(x,y,z+1)});
}

point solve(int nl, int nr, int ml, int mr, int kl, int kr)
{
    if((nr-nl+1)*(mr-ml+1)*(kr-kl+1) <= 8)
    {
        rep2(x,nl,nr)
        {
            rep2(y,ml,mr)
            {
                rep2(z,kl,kr)
                {
                    if(check({x,y,z})) return {x,y,z};
                }
            }
        }
        return {-1,-1,-1};
    }
    int n_cost = (mr-ml+1)*(kr-kl+1);
    int m_cost = (nr-nl+1)*(kr-kl+1);
    int k_cost = (mr-ml+1)*(nr-nl+1);
    pair<int,point> best = {-1,{0,0,0}};
    rep2(y,ml,mr)
    {
        rep2(z,kl,kr)
        {
            best = max(best,{ask(nl-1,y,z),{nl-1,y,z}});
            best = max(best,{ask(nr+1,y,z),{nr+1,y,z}});
        }
    }
    rep2(x,nl,nr)
    {
        rep2(z,kl,kr)
        {
            best = max(best,{ask(x,ml-1,z),{x,ml-1,z}});
            best = max(best,{ask(x,mr+1,z),{x,mr+1,z}});
        }
    }
    rep2(y,ml,mr)
    {
        rep2(x,nl,nr)
        {
            best = max(best,{ask(x,y,kl-1),{x,y,kl-1}});
            best = max(best,{ask(x,y,kr+1),{x,y,kr+1}});
        }
    }
    //cout << n_cost << " " << m_cost << " " << k_cost << " xd\n";
    if(n_cost == min({n_cost,m_cost,k_cost}))
    {
        int mid = (nl+nr)/2;
        rep2(y,ml,mr)
        {
            rep2(z,kl,kr)
            {
                best = max(best,{ask(mid,y,z),{mid,y,z}});
            }
        }
        if(best.ss.x == mid)
        {
            best = max(best,{ask(mid-1,best.ss.y,best.ss.z),{mid-1,best.ss.y,best.ss.z}});
            best = max(best,{ask(mid+1,best.ss.y,best.ss.z),{mid+1,best.ss.y,best.ss.z}});
        }
        if(check(best.ss)) return best.ss;
        if(best.ss.x < mid)
        {
            return solve(nl,mid-1,ml,mr,kl,kr);
        }
        else
        {
            return solve(mid+1,nr,ml,mr,kl,kr);
        }
    }
    else if(m_cost == min({m_cost,k_cost}))
    {
        int mid = (ml+mr)/2;
   //     cerr << mid << " mid\n";
        rep2(x,nl,nr)
        {
            rep2(z,kl,kr)
            {
                best = max(best,{ask(x,mid,z),{x,mid,z}});
            }
        }
        if(best.ss.y == mid)
        {
            best = max(best,{ask(best.ss.x,mid-1,best.ss.z),{best.ss.x,mid-1,best.ss.z}});
            best = max(best,{ask(best.ss.x,mid+1,best.ss.z),{best.ss.x,mid+1,best.ss.z}});
        }
        if(check(best.ss)) return best.ss;
        if(best.ss.y < mid)
        {
            return solve(nl,nr,ml,mid-1,kl,kr);
        }
        else
        {
            return solve(nl,nr,mid+1,mr,kl,kr);
        }
    }
    else
    {
        int mid = (kl+kr)/2;
        rep2(x,nl,nr)
        {
            rep2(y,ml,mr)
            {
                best = max(best,{ask(x,y,mid),{x,y,mid}});
            }
        }
        if(best.ss.z == mid)
        {
            best = max(best,{ask(best.ss.x,best.ss.y,mid-1),{best.ss.x,best.ss.y,mid-1}});
            best = max(best,{ask(best.ss.x,best.ss.y,mid+1),{best.ss.x,best.ss.y,mid+1}});
        }
        if(check(best.ss)) return best.ss;
        if(best.ss.z < mid)
        {
            return solve(nl,nr,ml,mr,kl,mid-1);
        }
        else
        {
            return solve(nl,nr,ml,mr,mid+1,kr);
        }
    }
}

pair<int,pii> find_sons(int x, int y, int z)
{
    vector<pair<int,pii>> dir = {{0,{0,1}},{0,{0,-1}},{0,{1,0}},{0,{-1,0}},{1,{0,0}},{-1,{0,0}}};
    shuffle(all(dir),mt);
    int val = ask(x,y,z);
    forall(it,dir)
    {
        int x2 = x+it.ff;
        int y2 = y+it.ss.ff;
        int z2 = z+it.ss.ss;
        if(ask(x2,y2,z2) > val) return {x2,{y2,z2}};
    }
    return {-1,{-1,-1}};
}

void solve2(int l, int r, int q1, int q2)
{
    q1 = min(r,q1);
    q1 = max(l,q1);
    q2 = min(r,q2);
    q2 = max(l,q2);
    if(abs(q1-q2) <= 2)
    {
        ll d1 = l*c1+r*c2;
        ll d2 = l*c2+r*c1;
        if(abs(d1-q1) > abs(d2-q1)) q2 = d1;
        else q2 = d2;
    }
    if(q1 > q2) swap(q1,q2);
    if(r-l+1 <= 2)
    {
        rep2(i,l,r)
        {
            if(ask(i) >= ask(i-1) && ask(i) >= ask(i+1))
            {
                cout << "! " << i+1 << " 1 1\n";
                return;
            }
        }
    } 
    int v1 = ask(q1);
    int v2 = ask(q2);
    if(v1 > v2)
    {
        solve2(l,q2-1,q1,c2*(ld)l+(ld)(q2-1)*c1);
    }
    else
    {
        solve2(q1+1,r,q2,(ld)(q1+1)*c2+(ld)r*c1);
    }
}


int main()
{
    cin >> n >> m >> k >> q;
    if(n == 1000 && m == 1000)
    {
        point ans = solve(0,n-1,0,m-1,0,k-1);
        cout << "! " << ans.x+1 << " " << ans.y+1 << " " << ans.z+1 << "\n";
    }
    else if(n == 1000000)
    {
        ll l = (ld)n*c2-1;
        ll r = (ld)n*c1-1;
        solve2(0,n-1,l,r);
    }
    else 
    {
        random_start();
        pair<int,pair<int,pii>> best = {-1,{-1,{-1,-1}}};
        rep(i,(ll)((ld)q*0.5))
        {
            int x = los(0,n-1);
            int y = los(0,m-1);
            int z = los(0,k-1);
            best = max(best,{ask(x,y,z),{x,{y,z}}});
        }    
        int x = best.ss.ff;
        int y = best.ss.ss.ff;
        int z = best.ss.ss.ss;
        while(true)
        {
            pair<int,pii> nxt = find_sons(x,y,z);
            if(nxt.ff == -1) break;
            x = nxt.ff;
            y = nxt.ss.ff;
            z = nxt.ss.ss;
        }
        cout << "! " << x+1 << " " << y+1 << " " << z+1 << "\n";
    }   
}

• Subtask 1: 10 / 10
• Subtask 2: 22 / 22
• Subtask 3: 12 / 12
• Subtask 4: 19 / 19
• Subtask 5: 14 / 14
• Subtask 6: 23 / 23