제출 #807164

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

books

// In the name of Allah

#include <bits/stdc++.h>
#include "books.h"
using namespace std;

typedef long long int	ll;
typedef long double	ld;
typedef pair<int, int>	pii;
typedef pair<ll, ll>	pll;

#define all(x)		(x).begin(),(x).end()
#define len(x)		((ll) (x).size())
#define F		first
#define S		second
#define pb		push_back
#define sep             ' '
#define endl            '\n'
#define Mp		make_pair
#define debug(x)	cerr << #x << ": " <<  x << endl;
#define kill(x)		cout << x << '\n', exit(0)
#define set_dec(x)	cout << fixed << setprecision(x);
#define file_io(x,y)	freopen(x, "r", stdin); freopen(y, "w", stdout);

void answerx(set<int> res) {
	vector<int> vc;
	for (int i : res) vc.pb(i);
	answer(vc);
}

void solve(int N, int K, ll A, int S) {
	ll arr[N + 1]; set<int> res;
	
	ll sm = 0;
	for (int i = 1; i <= K; i++) {
		arr[i] = skim(i); sm += arr[i];
		res.insert(i);
	}
	
	if (sm >= A && sm <= 2 * A) {
		answerx(res); return ;
	}
	else if (sm > 2 * A) {
		impossible(); return ;
	}
	
	sm -= arr[K]; res.erase(K);
	int l = K, r = N + 1;
	while (r - l > 1) {
		int mid = (l + r) / 2;
		arr[mid] = skim(mid);
		
		if (sm + arr[mid] <= 2 * A) l = mid;
		else r = mid;
	}
	sm += arr[l]; res.insert(l);
	
	int i = K - 1, j = l - 1;
	while (i >= 1 && (sm < A || sm > 2 * A)) {
		sm -= arr[i]; res.erase(i); i--;
		sm += arr[j]; res.insert(j); j--;
	}
	
	if (sm >= A && sm <= 2 * A) {
		answerx(res); return ;
	}
	else {
		impossible(); return ;
	}
}

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