#include <bits/stdc++.h>
//#include "books.h"
#include <random>
#include <chrono>
using namespace std;
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize ("unroll-loops")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef pair<double,double> pdd;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
template<typename T> void _do(T x){cerr<<x<<"\n";}
template<typename T,typename ...U> void _do(T x,U ...y){cerr<<x<<", ";_do(y...);}
#define dbg(...) cerr<<#__VA_ARGS__<<" = ";_do(__VA_ARGS__);
const int MOD1=1e9+7;
const int MOD2=998244353;
const ll INF=3e18;
const ld PI=3.14159265358979323846;
ll gcd(ll a,ll b){if(b==0) return a; return gcd(b,a%b);}
ll fpow(ll a,ll b,ll m)
{
if(!b) return 1;
ll ans=fpow(a*a%m,b/2,m);
return (b%2?ans*a%m:ans);
}
ll inv(ll a,ll m) {return fpow(a,m-2,m);}
#define MottoHayaku ios::sync_with_stdio(false);cin.tie(0);
//#define int ll
#define rep(i,n) for(ll i=0;i<n;i++)
#define rep1(i,n) for(ll i=1;i<=n;i++)
#define repk(i,m,n) for(int i=m;i<n;i++)
#define F first
#define S second
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define uni(c) c.resize(distance(c.begin(),unique(c.begin(),c.end())))
#define unisort(c) sort(c.begin(),c.end()),uni(c)
ll a[100005];
ll skim(int n);
void answer(vector<int> v);
void impossible();
void solve(int n,int k,ll A,int S)
{
if(n==k)
{
ll tot=0;
vector<int> ans;
rep1(i,n)
{
a[i]=skim(i);
ans.pb(i);
tot+=a[i];
}
if(tot>=A&&tot<=2*A) answer(ans);
else impossible();
return;
}
ll tot=0;
rep1(i,k)
{
a[i]=skim(i);
tot+=a[i];
}
if(a[k]>=A)
{
if(tot>=A&&tot<=2*A)
{
vector<int> ans;
rep1(i,k) ans.pb(i);
answer(ans);
}
else impossible();
return;
}
ll l=k+1,r=n+1;
while(l<r)
{
ll mid=l+r>>1;
a[mid]=skim(mid);
if(a[mid]<A) l=mid+1;
else r=mid;
}
if(l!=n+1)
{
if(a[l]==0) a[l]=skim(l);
vector<int> ans={l};
tot=a[l];
rep1(i,k-1)
{
tot+=a[i];
ans.pb(i);
}
if(tot>=A&&tot<=2*A)
{
answer(ans);
return;
}
}
l--;
ll cnt=0;
for(int i=l;i>=1;i--)
{
if(cnt==k) break;
if(a[i]==0)
{
cnt++;
a[i]=skim(i);
}
}
if(l==k)
{
ll tot=0;
vector<int> ans;
rep1(i,k)
{
ans.pb(i);
tot+=a[i];
}
if(tot>=A&&tot<=2*A) answer(ans);
else impossible();
return;
}
rep(i,k+1)
{
vector<int> ans;
ll tot=0;
rep1(j,k-i)
{
tot+=a[j];
ans.pb(j);
}
rep(j,i)
{
tot+=a[l-j];
ans.pb(l-j);
}
if(tot>=A&&tot<=2*A)
{
answer(ans);
return;
}
}
impossible();
}
Compilation message
books.cpp: In function 'void solve(int, int, ll, int)':
books.cpp:86:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
86 | ll mid=l+r>>1;
| ~^~
books.cpp:94:26: warning: narrowing conversion of 'l' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
94 | vector<int> ans={l};
| ^
books.cpp:94:26: warning: narrowing conversion of 'l' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
200 KB |
Output is correct |
2 |
Correct |
1 ms |
200 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
292 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
200 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
316 KB |
Output is correct |
2 |
Correct |
1 ms |
200 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
1 ms |
200 KB |
Output is correct |
6 |
Correct |
2 ms |
200 KB |
Output is correct |
7 |
Correct |
2 ms |
308 KB |
Output is correct |
8 |
Correct |
1 ms |
200 KB |
Output is correct |
9 |
Correct |
0 ms |
328 KB |
Output is correct |
10 |
Correct |
1 ms |
308 KB |
Output is correct |
11 |
Correct |
1 ms |
200 KB |
Output is correct |
12 |
Correct |
1 ms |
200 KB |
Output is correct |
13 |
Correct |
1 ms |
200 KB |
Output is correct |
14 |
Correct |
1 ms |
304 KB |
Output is correct |
15 |
Correct |
1 ms |
200 KB |
Output is correct |
16 |
Correct |
1 ms |
200 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
304 KB |
Output is correct |
2 |
Correct |
1 ms |
284 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
200 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
328 KB |
Output is correct |
9 |
Correct |
1 ms |
296 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
304 KB |
Output is correct |
2 |
Correct |
1 ms |
284 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
200 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
328 KB |
Output is correct |
9 |
Correct |
1 ms |
296 KB |
Output is correct |
10 |
Correct |
1 ms |
296 KB |
Output is correct |
11 |
Correct |
1 ms |
280 KB |
Output is correct |
12 |
Correct |
1 ms |
200 KB |
Output is correct |
13 |
Correct |
1 ms |
200 KB |
Output is correct |
14 |
Correct |
0 ms |
200 KB |
Output is correct |
15 |
Correct |
1 ms |
200 KB |
Output is correct |
16 |
Correct |
0 ms |
200 KB |
Output is correct |
17 |
Correct |
1 ms |
288 KB |
Output is correct |
18 |
Correct |
1 ms |
280 KB |
Output is correct |
19 |
Correct |
1 ms |
292 KB |
Output is correct |
20 |
Correct |
1 ms |
304 KB |
Output is correct |
21 |
Correct |
1 ms |
200 KB |
Output is correct |
22 |
Correct |
1 ms |
200 KB |
Output is correct |
23 |
Correct |
1 ms |
304 KB |
Output is correct |
24 |
Correct |
1 ms |
300 KB |
Output is correct |
25 |
Correct |
1 ms |
296 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
304 KB |
Output is correct |
2 |
Correct |
1 ms |
284 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
200 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
328 KB |
Output is correct |
9 |
Correct |
1 ms |
296 KB |
Output is correct |
10 |
Correct |
1 ms |
296 KB |
Output is correct |
11 |
Correct |
1 ms |
280 KB |
Output is correct |
12 |
Correct |
1 ms |
200 KB |
Output is correct |
13 |
Correct |
1 ms |
200 KB |
Output is correct |
14 |
Correct |
0 ms |
200 KB |
Output is correct |
15 |
Correct |
1 ms |
200 KB |
Output is correct |
16 |
Correct |
0 ms |
200 KB |
Output is correct |
17 |
Correct |
1 ms |
288 KB |
Output is correct |
18 |
Correct |
1 ms |
280 KB |
Output is correct |
19 |
Correct |
1 ms |
292 KB |
Output is correct |
20 |
Correct |
1 ms |
304 KB |
Output is correct |
21 |
Correct |
1 ms |
200 KB |
Output is correct |
22 |
Correct |
1 ms |
200 KB |
Output is correct |
23 |
Correct |
1 ms |
304 KB |
Output is correct |
24 |
Correct |
1 ms |
300 KB |
Output is correct |
25 |
Correct |
1 ms |
296 KB |
Output is correct |
26 |
Correct |
1 ms |
284 KB |
Output is correct |
27 |
Correct |
1 ms |
292 KB |
Output is correct |
28 |
Correct |
1 ms |
304 KB |
Output is correct |
29 |
Correct |
1 ms |
200 KB |
Output is correct |
30 |
Correct |
2 ms |
200 KB |
Output is correct |
31 |
Correct |
1 ms |
200 KB |
Output is correct |
32 |
Correct |
1 ms |
200 KB |
Output is correct |
33 |
Correct |
1 ms |
296 KB |
Output is correct |
34 |
Correct |
1 ms |
304 KB |
Output is correct |
35 |
Correct |
1 ms |
284 KB |
Output is correct |
36 |
Correct |
1 ms |
200 KB |
Output is correct |
37 |
Correct |
1 ms |
244 KB |
Output is correct |
38 |
Correct |
1 ms |
200 KB |
Output is correct |
39 |
Correct |
2 ms |
300 KB |
Output is correct |
40 |
Correct |
1 ms |
300 KB |
Output is correct |
41 |
Correct |
1 ms |
288 KB |
Output is correct |
42 |
Correct |
1 ms |
320 KB |
Output is correct |
43 |
Correct |
1 ms |
200 KB |
Output is correct |
44 |
Correct |
1 ms |
296 KB |
Output is correct |
45 |
Correct |
2 ms |
200 KB |
Output is correct |
46 |
Correct |
1 ms |
328 KB |
Output is correct |
47 |
Correct |
1 ms |
292 KB |
Output is correct |
48 |
Correct |
1 ms |
304 KB |
Output is correct |
49 |
Correct |
1 ms |
304 KB |
Output is correct |
50 |
Correct |
2 ms |
308 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
304 KB |
Output is correct |
2 |
Correct |
1 ms |
284 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
200 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
328 KB |
Output is correct |
9 |
Correct |
1 ms |
296 KB |
Output is correct |
10 |
Correct |
1 ms |
296 KB |
Output is correct |
11 |
Correct |
1 ms |
280 KB |
Output is correct |
12 |
Correct |
1 ms |
200 KB |
Output is correct |
13 |
Correct |
1 ms |
200 KB |
Output is correct |
14 |
Correct |
0 ms |
200 KB |
Output is correct |
15 |
Correct |
1 ms |
200 KB |
Output is correct |
16 |
Correct |
0 ms |
200 KB |
Output is correct |
17 |
Correct |
1 ms |
288 KB |
Output is correct |
18 |
Correct |
1 ms |
280 KB |
Output is correct |
19 |
Correct |
1 ms |
292 KB |
Output is correct |
20 |
Correct |
1 ms |
304 KB |
Output is correct |
21 |
Correct |
1 ms |
200 KB |
Output is correct |
22 |
Correct |
1 ms |
200 KB |
Output is correct |
23 |
Correct |
1 ms |
304 KB |
Output is correct |
24 |
Correct |
1 ms |
300 KB |
Output is correct |
25 |
Correct |
1 ms |
296 KB |
Output is correct |
26 |
Correct |
1 ms |
292 KB |
Output is correct |
27 |
Correct |
1 ms |
296 KB |
Output is correct |
28 |
Correct |
2 ms |
292 KB |
Output is correct |
29 |
Correct |
1 ms |
200 KB |
Output is correct |
30 |
Correct |
0 ms |
292 KB |
Output is correct |
31 |
Correct |
1 ms |
200 KB |
Output is correct |
32 |
Correct |
1 ms |
200 KB |
Output is correct |
33 |
Correct |
4 ms |
280 KB |
Output is correct |
34 |
Correct |
1 ms |
300 KB |
Output is correct |
35 |
Correct |
1 ms |
288 KB |
Output is correct |
36 |
Correct |
1 ms |
200 KB |
Output is correct |
37 |
Correct |
1 ms |
200 KB |
Output is correct |
38 |
Correct |
0 ms |
296 KB |
Output is correct |
39 |
Correct |
1 ms |
296 KB |
Output is correct |
40 |
Correct |
1 ms |
300 KB |
Output is correct |
41 |
Correct |
1 ms |
292 KB |
Output is correct |
42 |
Correct |
1 ms |
304 KB |
Output is correct |
43 |
Correct |
1 ms |
200 KB |
Output is correct |
44 |
Correct |
1 ms |
200 KB |
Output is correct |
45 |
Correct |
1 ms |
200 KB |
Output is correct |
46 |
Correct |
1 ms |
288 KB |
Output is correct |
47 |
Correct |
1 ms |
200 KB |
Output is correct |
48 |
Correct |
1 ms |
304 KB |
Output is correct |
49 |
Correct |
1 ms |
328 KB |
Output is correct |
50 |
Correct |
1 ms |
304 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
304 KB |
Output is correct |
2 |
Correct |
1 ms |
284 KB |
Output is correct |
3 |
Correct |
1 ms |
200 KB |
Output is correct |
4 |
Correct |
1 ms |
200 KB |
Output is correct |
5 |
Correct |
0 ms |
200 KB |
Output is correct |
6 |
Correct |
1 ms |
200 KB |
Output is correct |
7 |
Correct |
1 ms |
200 KB |
Output is correct |
8 |
Correct |
1 ms |
328 KB |
Output is correct |
9 |
Correct |
1 ms |
296 KB |
Output is correct |
10 |
Correct |
1 ms |
296 KB |
Output is correct |
11 |
Correct |
1 ms |
280 KB |
Output is correct |
12 |
Correct |
1 ms |
200 KB |
Output is correct |
13 |
Correct |
1 ms |
200 KB |
Output is correct |
14 |
Correct |
0 ms |
200 KB |
Output is correct |
15 |
Correct |
1 ms |
200 KB |
Output is correct |
16 |
Correct |
0 ms |
200 KB |
Output is correct |
17 |
Correct |
1 ms |
288 KB |
Output is correct |
18 |
Correct |
1 ms |
280 KB |
Output is correct |
19 |
Correct |
1 ms |
292 KB |
Output is correct |
20 |
Correct |
1 ms |
304 KB |
Output is correct |
21 |
Correct |
1 ms |
200 KB |
Output is correct |
22 |
Correct |
1 ms |
200 KB |
Output is correct |
23 |
Correct |
1 ms |
304 KB |
Output is correct |
24 |
Correct |
1 ms |
300 KB |
Output is correct |
25 |
Correct |
1 ms |
296 KB |
Output is correct |
26 |
Correct |
1 ms |
284 KB |
Output is correct |
27 |
Correct |
1 ms |
292 KB |
Output is correct |
28 |
Correct |
1 ms |
304 KB |
Output is correct |
29 |
Correct |
1 ms |
200 KB |
Output is correct |
30 |
Correct |
2 ms |
200 KB |
Output is correct |
31 |
Correct |
1 ms |
200 KB |
Output is correct |
32 |
Correct |
1 ms |
200 KB |
Output is correct |
33 |
Correct |
1 ms |
296 KB |
Output is correct |
34 |
Correct |
1 ms |
304 KB |
Output is correct |
35 |
Correct |
1 ms |
284 KB |
Output is correct |
36 |
Correct |
1 ms |
200 KB |
Output is correct |
37 |
Correct |
1 ms |
244 KB |
Output is correct |
38 |
Correct |
1 ms |
200 KB |
Output is correct |
39 |
Correct |
2 ms |
300 KB |
Output is correct |
40 |
Correct |
1 ms |
300 KB |
Output is correct |
41 |
Correct |
1 ms |
288 KB |
Output is correct |
42 |
Correct |
1 ms |
320 KB |
Output is correct |
43 |
Correct |
1 ms |
200 KB |
Output is correct |
44 |
Correct |
1 ms |
296 KB |
Output is correct |
45 |
Correct |
2 ms |
200 KB |
Output is correct |
46 |
Correct |
1 ms |
328 KB |
Output is correct |
47 |
Correct |
1 ms |
292 KB |
Output is correct |
48 |
Correct |
1 ms |
304 KB |
Output is correct |
49 |
Correct |
1 ms |
304 KB |
Output is correct |
50 |
Correct |
2 ms |
308 KB |
Output is correct |
51 |
Correct |
1 ms |
292 KB |
Output is correct |
52 |
Correct |
1 ms |
296 KB |
Output is correct |
53 |
Correct |
2 ms |
292 KB |
Output is correct |
54 |
Correct |
1 ms |
200 KB |
Output is correct |
55 |
Correct |
0 ms |
292 KB |
Output is correct |
56 |
Correct |
1 ms |
200 KB |
Output is correct |
57 |
Correct |
1 ms |
200 KB |
Output is correct |
58 |
Correct |
4 ms |
280 KB |
Output is correct |
59 |
Correct |
1 ms |
300 KB |
Output is correct |
60 |
Correct |
1 ms |
288 KB |
Output is correct |
61 |
Correct |
1 ms |
200 KB |
Output is correct |
62 |
Correct |
1 ms |
200 KB |
Output is correct |
63 |
Correct |
0 ms |
296 KB |
Output is correct |
64 |
Correct |
1 ms |
296 KB |
Output is correct |
65 |
Correct |
1 ms |
300 KB |
Output is correct |
66 |
Correct |
1 ms |
292 KB |
Output is correct |
67 |
Correct |
1 ms |
304 KB |
Output is correct |
68 |
Correct |
1 ms |
200 KB |
Output is correct |
69 |
Correct |
1 ms |
200 KB |
Output is correct |
70 |
Correct |
1 ms |
200 KB |
Output is correct |
71 |
Correct |
1 ms |
288 KB |
Output is correct |
72 |
Correct |
1 ms |
200 KB |
Output is correct |
73 |
Correct |
1 ms |
304 KB |
Output is correct |
74 |
Correct |
1 ms |
328 KB |
Output is correct |
75 |
Correct |
1 ms |
304 KB |
Output is correct |
76 |
Correct |
0 ms |
200 KB |
Output is correct |
77 |
Correct |
1 ms |
300 KB |
Output is correct |
78 |
Correct |
1 ms |
200 KB |
Output is correct |
79 |
Correct |
1 ms |
200 KB |
Output is correct |
80 |
Correct |
1 ms |
200 KB |
Output is correct |
81 |
Correct |
1 ms |
200 KB |
Output is correct |
82 |
Correct |
1 ms |
200 KB |
Output is correct |
83 |
Correct |
1 ms |
200 KB |
Output is correct |
84 |
Correct |
2 ms |
200 KB |
Output is correct |
85 |
Correct |
2 ms |
200 KB |
Output is correct |
86 |
Correct |
1 ms |
300 KB |
Output is correct |
87 |
Correct |
1 ms |
300 KB |
Output is correct |
88 |
Correct |
1 ms |
300 KB |
Output is correct |
89 |
Correct |
1 ms |
200 KB |
Output is correct |
90 |
Correct |
1 ms |
200 KB |
Output is correct |
91 |
Correct |
1 ms |
304 KB |
Output is correct |
92 |
Correct |
1 ms |
300 KB |
Output is correct |
93 |
Correct |
1 ms |
296 KB |
Output is correct |
94 |
Correct |
1 ms |
300 KB |
Output is correct |
95 |
Correct |
1 ms |
200 KB |
Output is correct |
96 |
Correct |
1 ms |
200 KB |
Output is correct |
97 |
Correct |
1 ms |
200 KB |
Output is correct |
98 |
Correct |
1 ms |
200 KB |
Output is correct |
99 |
Correct |
1 ms |
296 KB |
Output is correct |
100 |
Correct |
1 ms |
292 KB |
Output is correct |
101 |
Correct |
1 ms |
200 KB |
Output is correct |
102 |
Correct |
1 ms |
292 KB |
Output is correct |
103 |
Correct |
1 ms |
328 KB |
Output is correct |
104 |
Correct |
1 ms |
292 KB |
Output is correct |
105 |
Correct |
1 ms |
296 KB |
Output is correct |
106 |
Correct |
1 ms |
300 KB |
Output is correct |
107 |
Correct |
1 ms |
304 KB |
Output is correct |
108 |
Correct |
1 ms |
296 KB |
Output is correct |
109 |
Correct |
1 ms |
200 KB |
Output is correct |
110 |
Correct |
1 ms |
292 KB |
Output is correct |
111 |
Correct |
1 ms |
200 KB |
Output is correct |
112 |
Correct |
1 ms |
200 KB |
Output is correct |
113 |
Correct |
1 ms |
320 KB |
Output is correct |
114 |
Correct |
1 ms |
304 KB |
Output is correct |
115 |
Correct |
1 ms |
308 KB |
Output is correct |
116 |
Correct |
1 ms |
304 KB |
Output is correct |
117 |
Correct |
1 ms |
296 KB |
Output is correct |
118 |
Correct |
1 ms |
200 KB |
Output is correct |
119 |
Correct |
1 ms |
200 KB |
Output is correct |
120 |
Correct |
1 ms |
200 KB |
Output is correct |
121 |
Correct |
1 ms |
300 KB |
Output is correct |
122 |
Correct |
1 ms |
300 KB |
Output is correct |
123 |
Correct |
1 ms |
304 KB |
Output is correct |
124 |
Correct |
1 ms |
300 KB |
Output is correct |
125 |
Correct |
1 ms |
200 KB |
Output is correct |
126 |
Correct |
1 ms |
296 KB |
Output is correct |
127 |
Correct |
1 ms |
200 KB |
Output is correct |
128 |
Correct |
1 ms |
296 KB |
Output is correct |
129 |
Correct |
1 ms |
328 KB |
Output is correct |
130 |
Correct |
1 ms |
280 KB |
Output is correct |
131 |
Correct |
1 ms |
200 KB |
Output is correct |
132 |
Correct |
2 ms |
292 KB |
Output is correct |
133 |
Correct |
2 ms |
308 KB |
Output is correct |
134 |
Correct |
1 ms |
300 KB |
Output is correct |
135 |
Correct |
1 ms |
200 KB |
Output is correct |