제출 #529833

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

books

#include <map>
#include "books.h"
using namespace std;
 
map<int, long long> m;
int s;
 
long long get(int x, int K) {
    long long ret = 0;
    for (int i = 0; i < K; i++) {
        if (m.find(x + i + 1) == m.end()) m[x + i + 1] = skim(x + i + 1);
        ret += m[x + i + 1];
        if (ret > (long long)4e18) ret = (long long)4e18;
    }
    return ret;
}
 
void solve(int N, int K, long long A, int S) {
    if (get(0, K) >= A && get(0, K) <= 2 * A) {
        vector<int> v;
        for (int i = 0; i < K; i++) v.push_back(i + 1);
        answer(v);
        return;
    }
    int l = 0, r = N;
    while (l < r) {
        int m = (l + r) / 2;
        long long t = (m == N ? (long long)4e18 : get(m, 1));
        if (t < A) l = m + 1;
        else r = m;
    }
    if (l < N && l >= K - 1 && get(l, 1) + get(0, K - 1) >= A && get(l, 1) + get(0, K - 1) <= 2 * A) {
        vector<int> v;
        for (int i = 0; i < K - 1; i++) v.push_back(i + 1);
        v.push_back(l + 1);
        answer(v);
        return;
    }
    if (l >= K) for (int i = 0; i <= K; i++) if (get(0, i) + get(l - K + i, K - i) >= A && get(0, i) + get(l - K + i, K - i) <= 2 * A) {
        vector<int> v;
        for (int j = 0; j < i; j++) v.push_back(j + 1);
        for (int j = 0; j < K - i; j++) v.push_back(l - K + i + j + 1);
        answer(v);
        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