제출 #987244

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
987244		VMaksimoski008	A Difficult(y) Choice (BOI21_books)	C++17	45 / 100	2 ms	924 KiB

books

#include <bits/stdc++.h>
#include "books.h"

using namespace std;
using ll = long long;
using ull = unsigned long long;

map<int, ll> memo;

ll ask(int p) {
    if(memo.count(p)) return memo[p];
    return memo[p] = skim(p);
}

void solve(int N, int K, ll A, int S) {
    // if(N == S) {
    //     vector<ull> v = { 0 };
    //     for(int i=1; i<=N; i++) v.push_back(skim(i));

    //     ull sum1 = 0;
    //     for(int i=1; i<=K; i++) sum1 += v[i];

    //     if(sum1 >= A && sum1 <= 2 * A) {
    //         vector<int> ans;
    //         for(int i=1; i<=K; i++) ans.push_back(i);
    //         answer(ans);
    //     }

    //     for(int i=2; i+K-1<=N; i++) {
    //         sum1 -= v[i-1];
    //         sum1 += v[i+K-1];
    //         if(sum1 > 2 * A) break;

    //         if(sum1 >= A && sum1 <= 2 * A) {
    //             vector<int> ans;
    //             for(int j=i; j<=i+K-1; j++) ans.push_back(j);
    //             answer(ans);
    //         }
    //     }

    //     for(int i=K; i<=N; i++) {
    //         if(v[i] >= A) {
    //             ll sum = 0;
    //             for(int j=1; j<=K-1; j++) sum += v[j];
    //             if(sum + v[i] >= A && sum + v[i] <= 2 * A) {
    //                 vector<int> ans;
    //                 for(int j=1; j<=K-1; j++) ans.push_back(j);
    //                 ans.push_back(i);
    //                 answer(ans);
    //             }
    //             break;
    //         }
    //     }

    //     impossible();
    //     return ;
    // }

    //sub 3
    int l=1, r=N-K+1;
    while(l <= r) {
        int mid = (l + r) / 2;
        ll sum = 0;
        for(int i=mid; i<=mid+K-1; i++) sum += ask(i);

        if(sum >= A && sum <= 2 * A) {
            vector<int> ans;
            for(int i=mid; i<=mid+K-1; i++) ans.push_back(i);
            answer(ans);
        }

        if(sum < A) l = mid + 1;
        if(sum > 2 * A) r = mid - 1;
    }

    l=K, r=N-K+1;
    int p = -1;

    while(l <= r) {
        int mid = (l + r) / 2;
        if(ask(mid) >= A) p = mid, r = mid - 1;
        else l = mid + 1;
    }

    if(p != -1) {
        ll sum = ask(p);
        for(int i=1; i<=K-1; i++) sum += ask(i);
        if(sum >= A && sum <= 2 * A) {
            vector<int> ans;
            for(int i=1; i<=K-1; i++) ans.push_back(i);
            ans.push_back(p);
            answer(ans);
        }
    }
    
    impossible();
    return ;
}

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