제출 #424004

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
424004		zoooma13	A Difficult(y) Choice (BOI21_books)	C++14	100 / 100	1 ms	328 KiB

books

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

map <int ,long long> mem;
long long ask(int i){
    if(mem.count(i))
        return mem[i];
    return mem[i] = skim(i);
}
vector <int> range(int l ,int r){
    vector <int> ret;
    for(int i = l; i <= r; i++)
        ret.push_back(i);
    return ret;
}

void solve(int N, int K, long long A, int S) {
    int l = 1 ,r = N ,m;
    while(l <= r){
        m = (l+r)>>1;
        if(ask(m) > A/K)
            r = m-1;
        else
            l = m+1;
    }

    r = max(K ,l);
    l = r-K+1;

    long long tot = 0;
    for(int i = l; i <= r; i++)
        tot += ask(i);
    while(tot <= 2*A){
        if(tot >= A)
            answer(range(l ,r));
        if(r == N)
            impossible();
        tot += ask(++r);
        tot -= ask(l++);
    }

    auto ans = range(1 ,K-1);
    ans.push_back(r);

    tot = 0;
    for(int&i : ans)
        tot += ask(i);
    if(A <= tot && tot <= 2*A)
        answer(ans);
    impossible();
}

• Subtask 1: 5 / 5
• Subtask 2: 15 / 15
• Subtask 3: 10 / 10
• Subtask 4: 15 / 15
• Subtask 5: 15 / 15
• Subtask 6: 20 / 20
• Subtask 7: 20 / 20