Submission #1322463

#TimeUsernameProblemLanguageResultExecution timeMemory
1322463adiyerA Difficult(y) Choice (BOI21_books)C++20
20 / 100
76 ms1188 KiB
#include <bits/stdc++.h>

#include "books.h"

using namespace std;

typedef long long ll;

void solve(int n, int k, ll s, int c) {
    vector < int > ans, pos;
    ll a[n + 1], cur = 0;
    for(int i = 1; i <= n; i++) a[i] = skim(i);
    for(int i = k; i <= n; i++){
        if(a[i] >= s){
            cur = a[i];
            for(int j = 1; j < k; j++) cur += a[j];
            if(cur <= 2 * s){
                ans.push_back(i);
                for(int j = 1; j < k; j++) ans.push_back(j);
                answer(ans);
                return;
            }
        }
    }
    cur = 0;
    for(int i = n; i >= 1; i--)
        if(a[i] < s && pos.size() < k)
            pos.push_back(i);
    if(pos.size() < k) { impossible(); return; }
    for(int i = 1; i < pos[k - 1] && i <= k; i++)
        pos.push_back(i);
    for(int msk = 0; msk < (1 << (int)pos.size()); msk++){
        if(__builtin_popcount(msk) != k) continue;
        for(int i = 0; i < pos.size(); i++)
            if((msk >> i & 1))
                cur += a[pos[i]], ans.push_back(pos[i]);
        if(s <= cur && cur <= 2 * s){ answer(ans); return; }
        cur = 0, ans.clear();
    }
    impossible();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...