Submission #667420

#	Time	Username	Problem	Language	Result	Execution time	Memory
667420		someone	A Difficult(y) Choice (BOI21_books)	C++14	100 / 100	2 ms	336 KiB

books

#include "books.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
 
const int MAXN = 2e5 + 42;
 
ll val[MAXN];
 
ll get(int i) {
    if(val[i] == 0)
        val[i] = skim(i);
    return val[i];
}
 
void solve(int N, int K, ll A, int S) {
    vector<int> ans;
    ll sum = 0;
    for(int i = 1; i <= K; i++)
        sum += get(i);
    int last = K;
    for(int j = 18; j >= 0; j--)
      	if(last + (1 << j) <= N && sum + get(last + (1 << j)) - get(K) <= 2*A)
            last += (1 << j);
    for(int i = K; i > 0; i--) {
        if(sum + get(last + i - K) - get(i) <= 2*A) {
            sum += get(last + i - K) - get(i);
            ans.push_back(last + i - K);
        } else {
            impossible();
        }
        if(sum >= A) {
            for(int j = 1; j < i; j++)
                ans.push_back(j);
            answer(ans);
            return;
        }
    }
    
    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