Submission #1115475

#TimeUsernameProblemLanguageResultExecution timeMemory
1115475Dan4LifeA Difficult(y) Choice (BOI21_books)C++17
0 / 100
2 ms1104 KiB
#include <bits/stdc++.h>
#include "books.h"
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)
using ll = long long;
using vi = vector<int>;
const int N = (int)1e5+10;
int n, K;
ll A, a[N];

ll Skim(int i){
	if(a[i]!=-1) return a[i];
	return a[i]=skim(i);
}
 
ll getSum(int mid, int num){
	ll s = 0;
	for(int j = mid; j < mid+num; j++) s+=Skim(j);
	return s;
}

void solve(int _N, int _K, ll _A, int S) {
	n = _N; A=_A,K=_K; 
	vi ans(K,0); iota(all(ans),1);
	memset(a,-1,sizeof(a));
	ll sum = getSum(1,K);
	if(sum>2*A) impossible();
	if(sum>=A) answer(ans);
	int l = 1, r = n;
	while(l<r){
		int mid = (l+r)/2;
		if(Skim(mid)>A) r=mid;
		else l=mid+1;
	}
	if(sum+a[l]-a[K]<=2*A) ans[K-1] = l, answer(ans);
	deque<int> ans2(K,0); iota(all(ans2),1);
	int j = max(K+1,l-K-1)-1;
	for(int i = 1; i <= K; i++){
		sum-=a[i]; ans2.pop_front();
		if(j+i>=l) break; 
		sum+=a[j+i]; ans2.push_back(j+i);
		if(sum<=2*A) answer(vi(all(ans2)));
	}
	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...