Submission #906037

#TimeUsernameProblemLanguageResultExecution timeMemory
906037nightfalThe Big Prize (IOI17_prize)C++14
0 / 100
14 ms11416 KiB
//#include "prize.h"
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

static const int max_q = 10000;
static int n;
static int query_count = 0;
static vector<int> g;
static vector<vector<int> > rank_count;

int find_best(int n);
std::vector<int> ask(int i);

int subtask0(int n) {
    std::vector<int> ans;
    for(int i=0; i<n; i++) {
        ans = ask(i); 
        if(ans[0]+ans[1]==0) return i;
    }
}

int subtask1(int n){
    int start=0, end=n-1;
    while (start<end) {
        int mid = (start+end)/2;
        vector<int> ans = ask(mid);
        if (!ans[0] && !ans[1]) return mid;
        else if (!ans[0]) start=mid+1;
        else end = mid-1;
    }
    return start;
}

int bSearch(int start, int end, std::vector<int>& sAns, std::vector<int>& eAns, int maxVal) {
    // printf("start=%d, end=%d, ",start,end);
    int result = -1;
    if (eAns[0]==sAns[0]) return -1;
    int mid = (start+end+1)/2;
    // printf("mid=%d ",mid);
    vector<int> mAns = ask(mid);
    if (mAns[0]+mAns[1]==0) return mid;
    int midL=mid, midR=mid;

    if(mAns[0]+mAns[1]<maxVal) {
        midL = start;
        for(int i=mid-1; i>start; i--) {
            mAns = ask(i);
            if(mAns[0]+mAns[1]==0) return i;
            if(mAns[0]+mAns[1]==maxVal) {midL=i; break;}
        }
        midR = end;
        for(int i=mid+1; i<end; i++) {
            mAns = ask(i);
            if(mAns[0]+mAns[1]==0) return i;
            if(mAns[0]+mAns[1]==maxVal) {midR=i; break;}
        }
    }
    // printf("midL=%d, midR=%d\n",midL,midR);
    if (start<midL) result = bSearch(start,midL,sAns,mAns,maxVal);
    if (result != -1) return result;
    if(midR<end) return bSearch(midR,end,mAns,eAns,maxVal);
    return -1;
}

int find_best(int n) {
    // if (n<=5000) {return subtask0(n);}

    vector<vector<int>> ans(n,vector<int>(2,-1));

    int m = min(500,n), maxVal=0, maxIdx=-1;
	for(int i=0; i<m; i++) {
        ans[i] = ask(i);
        int sum = ans[i][0]+ans[i][1];
        if(sum==0) return i;
        if (maxVal<sum) {maxVal=sum; maxIdx=i;}
    }
    // printf("maxVal:%d, maxIdx:%d\n",maxVal,maxIdx);
        ans[n-1] = ask(n-1);
	return bSearch(0,n-1,ans[0],ans[n-1],maxVal);
}

Compilation message (stderr)

prize.cpp: In function 'int find_best(int)':
prize.cpp:73:35: warning: variable 'maxIdx' set but not used [-Wunused-but-set-variable]
   73 |     int m = min(500,n), maxVal=0, maxIdx=-1;
      |                                   ^~~~~~
prize.cpp: In function 'int subtask0(int)':
prize.cpp:18:22: warning: control reaches end of non-void function [-Wreturn-type]
   18 |     std::vector<int> ans;
      |                      ^~~
prize.cpp: At global scope:
prize.cpp:10:12: warning: 'query_count' defined but not used [-Wunused-variable]
   10 | static int query_count = 0;
      |            ^~~~~~~~~~~
prize.cpp:9:12: warning: 'n' defined but not used [-Wunused-variable]
    9 | static int n;
      |            ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...