Submission #846959

#TimeUsernameProblemLanguageResultExecution timeMemory
846959Ahmed57A Difficult(y) Choice (BOI21_books)C++17
Compilation error
0 ms0 KiB
#include <bits/stdc++.h>
using namespace std;
//#include "books.h"

long long val[100001];
int xd= 0;
int s;
long long qu(int i){
    if(val[i]!=-1)return val[i];
    xd++;
    if(xd>s)assert(0);
    return val[i] = skim(i);
}
void solve(int N,int K,long long A,int S){
    if(N<=S){
    long long arr[N];
    for(int i = 0;i<N;i++){
        arr[i] = skim(i+1);
    }
    for(int i = 0;i<N;i++){
        long long sum = 0;
        for(int j = i;j<min(N,i+K);j++){
            sum+=arr[j];
            if(sum>2*A)break;
            if(j==i+K-1){
                if(sum>=A){
                    vector<int> v;
                    for(int e = i;e<=j;e++){
                        v.push_back(e+1);
                    }
                    answer(v);
                }
                break;
            }
            int rem = K-(j-i+1);
            int l = j+1 , r = N-rem+1;
            while(l<=r){
                int mid = (l+r)/2;
                long long val = 0;
                for(int e = mid;e<mid+rem;e++)val+=arr[e];
                if(sum+val<A)l = mid+1;
                else if(sum+val>2*A)r = mid-1;
                else {
                    vector<int> v;
                    for(int e = i;e<=j;e++){
                        v.push_back(e+1);
                    }
                    for(int e = mid;e<mid+rem;e++)v.push_back(e+1);
                    answer(v);
                }
            }
        }
    }
    impossible();
    }else{
    memset(val,-1,sizeof val);
    s = S;
    long long l = 1 , r = N-K+1 , ans = N-K+2 , sum = 0;
    while(l<=r){
        int mid = (l+r)/2;
        long long val = 0;
        for(int i = mid;i<mid+K;i++){
            val+=qu(i);
        }
        if(val>=A){
            r = mid-1;
            sum = val;
            ans = mid;
        }else l = mid+1;
    }
    if(ans==N-K+2){
        for(int i = N-K+2;i<=N;i++)sum+=qu(i);
    }else sum-=qu(ans+K-1);
    vector<int> xd;
    for(int i = 1;i<=N;i++)if(i<ans||i>=ans+K-1)xd.push_back(i);
    l = 0 , r = xd.size()-1;
    while(l<=r){
        int mid = (l+r)/2;
        if(sum+qu(xd[mid])<A){
            l= mid+1;
        }else if(sum+qu(xd[mid])>2*A){
            r = mid-1;
        }else{
            vector<int> v;
            for(int i = ans;i<ans+K-1;i++)v.push_back(i);
            v.push_back(xd[mid]);
            answer(v);
            return ;
        }
    }
    if(ans==1)impossible();
    ans--;sum = 0;
    for(int i = ans;i<ans+K-1;i++)sum+=qu(i);
    l = 0 , r = xd.size()-1;
    while(l<=r){
        int mid = (l+r)/2;
        if(sum+qu(xd[mid])<A){
            l= mid+1;
        }else if(sum+qu(xd[mid])>2*A){
            r = mid-1;
        }else{
            vector<int> v;
            for(int i = ans;i<ans+K-1;i++)v.push_back(i);
            v.push_back(xd[mid]);
            answer(v);
            return ;
        }
    }
    if(ans==1)impossible();
    ans--;sum = 0;
    for(int i = ans;i<ans+K-1;i++)sum+=qu(i);
    l = 0 , r = xd.size()-1;
    while(l<=r){
        int mid = (l+r)/2;
        if(sum+qu(xd[mid])<A){
            l= mid+1;
        }else if(sum+qu(xd[mid])>2*A){
            r = mid-1;
        }else{
            vector<int> v;
            for(int i = ans;i<ans+K-1;i++)v.push_back(i);
            v.push_back(xd[mid]);
            answer(v);
            return ;
        }
    }
    if(ans==1)impossible();
    ans--;sum = 0;
    for(int i = ans;i<ans+K-1;i++)sum+=qu(i);
    l = 0 , r = xd.size()-1;
    while(l<=r){
        int mid = (l+r)/2;
        if(sum+qu(xd[mid])<A){
            l= mid+1;
        }else if(sum+qu(xd[mid])>2*A){
            r = mid-1;
        }else{
            vector<int> v;
            for(int i = ans;i<ans+K-1;i++)v.push_back(i);
            v.push_back(xd[mid]);
            answer(v);
            return ;
        }
    }
    }
}

Compilation message (stderr)

books.cpp: In function 'long long int qu(int)':
books.cpp:12:21: error: 'skim' was not declared in this scope
   12 |     return val[i] = skim(i);
      |                     ^~~~
books.cpp: In function 'void solve(int, int, long long int, int)':
books.cpp:18:18: error: 'skim' was not declared in this scope
   18 |         arr[i] = skim(i+1);
      |                  ^~~~
books.cpp:31:21: error: 'answer' was not declared in this scope
   31 |                     answer(v);
      |                     ^~~~~~
books.cpp:49:21: error: 'answer' was not declared in this scope
   49 |                     answer(v);
      |                     ^~~~~~
books.cpp:54:5: error: 'impossible' was not declared in this scope
   54 |     impossible();
      |     ^~~~~~~~~~
books.cpp:87:13: error: 'answer' was not declared in this scope
   87 |             answer(v);
      |             ^~~~~~
books.cpp:91:15: error: 'impossible' was not declared in this scope
   91 |     if(ans==1)impossible();
      |               ^~~~~~~~~~
books.cpp:105:13: error: 'answer' was not declared in this scope
  105 |             answer(v);
      |             ^~~~~~
books.cpp:109:15: error: 'impossible' was not declared in this scope
  109 |     if(ans==1)impossible();
      |               ^~~~~~~~~~
books.cpp:123:13: error: 'answer' was not declared in this scope
  123 |             answer(v);
      |             ^~~~~~
books.cpp:127:15: error: 'impossible' was not declared in this scope
  127 |     if(ans==1)impossible();
      |               ^~~~~~~~~~
books.cpp:141:13: error: 'answer' was not declared in this scope
  141 |             answer(v);
      |             ^~~~~~