제출 #1033021

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1033021		anango	Worm Worries (BOI18_worm)	C++17	41 / 100	3077 ms	1104 KiB

worm

#include <bits/stdc++.h>
#define int long long
using namespace std;
int INF = 1LL<<30;
mt19937 rng;
vector<int> dx = {0,0,0,0,-1,1};
vector<int> dy = {0,0,-1,1,0,0};
vector<int> dz = {-1,1,0,0,0,0};

int n,m,k,q;

int curmax=-1; int xmax=-1; int ymax=-1;

int qcount = 0;
int debug = 0;
vector<vector<vector<int>>> humidity;

vector<vector<int>> neighbours(vector<int> point) {
    vector<vector<int>> ans;
    for (int dir=0; dir<6; dir++) {
        int nx,ny,nz; nx=point[0]+dx[dir]; ny=point[1]+dy[dir]; nz=point[2]+dz[dir];
        if (!(1<=nx && nx<=n && 1<=ny && ny<=m && 1<=nz && nz<=k)) {
            continue;
        }
        ans.push_back({nx,ny,nz});
    }
    return ans;
}

map<vector<int>,int> cache;
int queryw(int x, int y, int z) {
    if (!(1<=x && x<=n && 1<=y && y<=m && 1<=z && z<=k)) {
        return -1;
    }
    if (cache.count({x,y,z})) return cache[{x,y,z}];
    cout << "? " << x <<" " << y <<" " <<z << endl;
    qcount++;
    if (qcount>=q-1) {
        while (1==1) {
            int o=0;
        }
    }
    int an; 
    if (!debug) {
        cin >> an; 
    }
    else {
        cout << humidity[x-1][y-1][z-1] << endl;
        return cache[{x,y,z}] = humidity[x-1][y-1][z-1];
    }
    return cache[{x,y,z}] = an;
}

int query(int x, int y, int z) {
    int ans = queryw(x,y,z);
    if (ans>curmax) {
        curmax = ans; xmax = x; ymax = y;
    }
    return ans;
}

void answer(int x, int y, int z) {
    cout << "! " << x  << " " << y << " " << z << endl;
    if (debug) {
        cout << "in " << qcount <<" " << "queries" << endl;
        int gg=0;
        for (auto j:neighbours({x,y,z})) {
            if (humidity[j[0]-1][j[1]-1][j[2]-1]>humidity[x-1][y-1][z-1]) {
                gg=1;
            }
        }
        if (gg) {
            cout << "INCORRECT" << endl;
        }
        else {
            cout << "CORRECT" << endl;
        }
    }
}




signed main() {
    rng.discard((time(NULL))%100000);
    cin >> n >> m >> k >> q;
    if (debug) humidity = vector<vector<vector<int>>>(n,vector<vector<int>>(m,vector<int>(k,0)));
    if (debug) {
        int findx = 50;
        int findy = 7;
        for (int i=0; i<n; i++) {
            for (int j=0; j<m; j++) {
                for (int k1=0; k1<k; k1++) {
                    humidity[i][j][k1]=100000-(abs(i-findx)+abs(j-findy)+1);//rng()%10000;
                    cout << humidity[i][j][k1] <<" ";
                }
            }
            cout << endl;
        }
    }
    int lx,rx,ly,ry;
    int answered = 0;
    lx = 1; rx = n; ly = 1; ry = m;
    while (lx<rx || ly<ry) {
        if (debug) cout << "binsearch " << lx <<" " << rx <<" " << ly <<" " << ry << endl;
        if (rx-lx>ry-ly) {
            //split by x
            int mx = (lx+rx)/2;
            int ma = -1;
            int mi = -1;
            for (int y=ly; y<=ry; y++) {
                if (query(mx,y,1)>ma) {
                    ma = query(mx,y,1);
                    mi = y;
                }
            }
            assert(ma<=curmax);
            
            if ((query(mx+1,mi,1)>query(mx,mi,1) && ma==curmax) || (xmax>mx && ma<curmax)) {
                //go down
                lx=mx+1;
            
            }
            else if ((query(mx-1,mi,1)>query(mx,mi,1) && ma==curmax) || (xmax<mx && ma<curmax)) {
                rx = mx-1;
            }
            else {
                answer(mx,mi,1); answered = 1;
                break;
            }
        }
        else {
            //split by y
            int my = (ly+ry)/2;
            int ma = -1;
            int mi = -1;
            for (int x=lx; x<=rx; x++) {
                if (query(x,my,1)>ma) {
                    ma = query(x,my,1);
                    mi = x;
                }
            }
            assert(ma<=curmax);
            if ((query(mi,my+1,1)>query(mi,my,1)  && ma==curmax) || (ymax>my && ma<curmax)) {
                //go down
                ly=my+1;
            }
            else if ((query(mi,my-1,1)>query(mi,my,1) && ma==curmax) || (ymax<my && ma<curmax)) {
                ry=my-1;
            }
            else {
                answer(mi,my,1); answered = 1;
                break;
            }
        }
    }
    if (!answered) {
        answer(lx,ly,1);
    }
}

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

worm.cpp: In function 'long long int queryw(long long int, long long int, long long int)':
worm.cpp:40:17: warning: unused variable 'o' [-Wunused-variable]
   40 |             int o=0;
      |                 ^

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