oj.uz

Submission #556982

# Submission time Handle Problem Language Result Execution time Memory
556982 2022-05-04T13:47:03 Z kym A Difficult(y) Choice (BOI21_books) C++14
100 / 100
 52 ms 428 KB 
#include <bits/stdc++.h>
using namespace std;
#include "books.h"
#define all(x) (x).begin(), (x).end()
#define ll long long
#define s second 
#define f first 
#define pb push_back
#define FOR(i,s,e) for(ll i = s; i <= (ll)e; ++i)
typedef pair <long long ,int> pi;
#ifdef LOCAL
#define db(x) cerr << #x << "=" << x << "\n"
#define db2(x, y) cerr << #x << "=" << x << " , " << #y << "=" << y << "\n"
#define db3(a,b,c) cerr<<#a<<"="<<a<<","<<#b<<"="<<b<<","<<#c<<"="<<c<<"\n"
#define dbv(v) cerr << #v << ":"; for (auto ite : v) cerr << ite << ' '; cerr <<"\n"
#define dbvp(v) cerr << #v << ":"; for (auto ite : v) cerr << "{"  << ite.f << ',' << ite.s << "} "; cerr << "\n"
#define dba(a,ss,ee) cerr << #a << ":"; FOR(ite,ss,ee) cerr << a[ite] << ' '; cerr << "\n"
#define reach cerr << "LINE: " << __LINE__ << "\n";
#else
#define db(x)
#define db2(x,y)
#define db3(a,b,c)
#define dbv(v)
#define dbvp(v)
#define dba(a,ss,ee)
#define reach
#endif
//
// --- Sample implementation for the task books ---
//
// To compile this program with the sample grader, place:
//     books.h books_sample.cpp sample_grader.cpp
// in a single folder and run:
//     g++ books_sample.cpp sample_grader.cpp
// in this folder.
//


/*
SAMPLE GRADER for task BOOKS

USAGE:
place together with your solution and books.h in the same directory, then:
g++ <flags> sample_grader.cpp <solution_file>
e.g.:
g++ -std=c++17 sample_grader.cpp books.cpp

INPUT/OUTPUT:
The sample grader expects on standard input two lines. The first line should
contain the four integers N, K, A and S. The second line should contain a list
of N integers, the sequence of difficulties x_1 x_2 ... x_N which has to be
strictly increasing. Then, the grader writes to standard output a protocol of
all grader functions called by your program.
At the end, the grader prints your verdict.
*/
map<int,ll> memo;
ll query(int x) {
	if(memo.find(x) != memo.end())return memo[x];
	else return memo[x] = skim(x);
}
void solve(int n, int k, long long a, int s) {
    ll lo = 0, hi = n+1;
    while(lo<hi-1) {
    	ll mid = (lo+hi)/2;
    	ll x = query(mid);
    	if(x >= a)hi=mid;
    	else lo=mid;
    }
    vector<pi> v;
    ll r1 = 0;
    FOR(i,1,k) {
    	ll x = query(i);
    	r1 += x;
    	v.pb(pi(x,i));
    }
    reach
    if(r1 > 2*a)impossible();
    if(r1 >= a){
    	vector<int> ans;
    	FOR(i,1,k)ans.pb(i);
    	answer(ans);
    }
    if(hi <= n &&  (hi>=k)&& query(hi) >= a && query(hi) <= 2*a) {
    	ll tot = 0;
    	FOR(i,1,k-1) tot+=query(i);
    	tot+=query(hi);
    	if(tot <= 2*a) {
    		vector<int> ans;
    		FOR(i,1,k-1)ans.pb(i);
    		ans.pb(hi);
    		answer(ans);
    	}
    }
    //we don't use hi
    FOR(i,hi-k,hi-1) {
    	if(i>0ll && i<=n)v.pb(pi(skim(i),i));
    }
    sort(all(v));
    v.resize(unique(all(v)) - v.begin());
    //assert(hi>=10 && hi<=n-10);
    FOR(i,0,(1<<(v.size()))-1) {
    	if(__builtin_popcount(i) != k)continue;
    	ll tot = 0;
    	vector<int> res;
    	FOR(j,0,v.size()-1) {
    		if(i&(1<<j)) {
    			tot += v[j].f;
    			res.pb(v[j].s);
    		}
    	}
    	if(tot >= a && tot <= 2*a){
    		answer(res);
    	}
    }
    impossible();
    
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 208 KB Output is correct
2 Correct 1 ms 208 KB Output is correct
3 Correct 1 ms 208 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 0 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 1 ms 208 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 208 KB Output is correct
2 Correct 1 ms 208 KB Output is correct
3 Correct 1 ms 208 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 0 ms 208 KB Output is correct
6 Correct 1 ms 256 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 1 ms 208 KB Output is correct
9 Correct 1 ms 208 KB Output is correct
10 Correct 1 ms 208 KB Output is correct
11 Correct 1 ms 300 KB Output is correct
12 Correct 42 ms 208 KB Output is correct
13 Correct 1 ms 288 KB Output is correct
14 Correct 1 ms 428 KB Output is correct
15 Correct 1 ms 292 KB Output is correct
16 Correct 41 ms 208 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 208 KB Output is correct
2 Correct 1 ms 280 KB Output is correct
3 Correct 1 ms 276 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 1 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 41 ms 296 KB Output is correct
9 Correct 1 ms 208 KB Output is correct

Subtask #4
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 208 KB Output is correct
2 Correct 1 ms 280 KB Output is correct
3 Correct 1 ms 276 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 1 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 41 ms 296 KB Output is correct
9 Correct 1 ms 208 KB Output is correct
10 Correct 1 ms 208 KB Output is correct
11 Correct 2 ms 208 KB Output is correct
12 Correct 1 ms 208 KB Output is correct
13 Correct 1 ms 208 KB Output is correct
14 Correct 1 ms 208 KB Output is correct
15 Correct 1 ms 208 KB Output is correct
16 Correct 1 ms 256 KB Output is correct
17 Correct 41 ms 208 KB Output is correct
18 Correct 1 ms 208 KB Output is correct
19 Correct 1 ms 300 KB Output is correct
20 Correct 1 ms 208 KB Output is correct
21 Correct 1 ms 208 KB Output is correct
22 Correct 1 ms 208 KB Output is correct
23 Correct 1 ms 300 KB Output is correct
24 Correct 1 ms 336 KB Output is correct
25 Correct 1 ms 304 KB Output is correct

Subtask #5
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 208 KB Output is correct
2 Correct 1 ms 280 KB Output is correct
3 Correct 1 ms 276 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 1 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 41 ms 296 KB Output is correct
9 Correct 1 ms 208 KB Output is correct
10 Correct 1 ms 208 KB Output is correct
11 Correct 2 ms 208 KB Output is correct
12 Correct 1 ms 208 KB Output is correct
13 Correct 1 ms 208 KB Output is correct
14 Correct 1 ms 208 KB Output is correct
15 Correct 1 ms 208 KB Output is correct
16 Correct 1 ms 256 KB Output is correct
17 Correct 41 ms 208 KB Output is correct
18 Correct 1 ms 208 KB Output is correct
19 Correct 1 ms 300 KB Output is correct
20 Correct 1 ms 208 KB Output is correct
21 Correct 1 ms 208 KB Output is correct
22 Correct 1 ms 208 KB Output is correct
23 Correct 1 ms 300 KB Output is correct
24 Correct 1 ms 336 KB Output is correct
25 Correct 1 ms 304 KB Output is correct
26 Correct 1 ms 208 KB Output is correct
27 Correct 1 ms 208 KB Output is correct
28 Correct 1 ms 208 KB Output is correct
29 Correct 1 ms 208 KB Output is correct
30 Correct 1 ms 208 KB Output is correct
31 Correct 1 ms 208 KB Output is correct
32 Correct 1 ms 208 KB Output is correct
33 Correct 40 ms 208 KB Output is correct
34 Correct 1 ms 296 KB Output is correct
35 Correct 1 ms 208 KB Output is correct
36 Correct 1 ms 208 KB Output is correct
37 Correct 1 ms 208 KB Output is correct
38 Correct 1 ms 208 KB Output is correct
39 Correct 1 ms 276 KB Output is correct
40 Correct 1 ms 296 KB Output is correct
41 Correct 1 ms 324 KB Output is correct
42 Correct 1 ms 208 KB Output is correct
43 Correct 1 ms 208 KB Output is correct
44 Correct 2 ms 208 KB Output is correct
45 Correct 1 ms 208 KB Output is correct
46 Correct 1 ms 208 KB Output is correct
47 Correct 3 ms 296 KB Output is correct
48 Correct 1 ms 304 KB Output is correct
49 Correct 1 ms 304 KB Output is correct
50 Correct 41 ms 208 KB Output is correct

Subtask #6
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 208 KB Output is correct
2 Correct 1 ms 280 KB Output is correct
3 Correct 1 ms 276 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 1 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 41 ms 296 KB Output is correct
9 Correct 1 ms 208 KB Output is correct
10 Correct 1 ms 208 KB Output is correct
11 Correct 2 ms 208 KB Output is correct
12 Correct 1 ms 208 KB Output is correct
13 Correct 1 ms 208 KB Output is correct
14 Correct 1 ms 208 KB Output is correct
15 Correct 1 ms 208 KB Output is correct
16 Correct 1 ms 256 KB Output is correct
17 Correct 41 ms 208 KB Output is correct
18 Correct 1 ms 208 KB Output is correct
19 Correct 1 ms 300 KB Output is correct
20 Correct 1 ms 208 KB Output is correct
21 Correct 1 ms 208 KB Output is correct
22 Correct 1 ms 208 KB Output is correct
23 Correct 1 ms 300 KB Output is correct
24 Correct 1 ms 336 KB Output is correct
25 Correct 1 ms 304 KB Output is correct
26 Correct 1 ms 208 KB Output is correct
27 Correct 1 ms 280 KB Output is correct
28 Correct 1 ms 208 KB Output is correct
29 Correct 1 ms 276 KB Output is correct
30 Correct 1 ms 208 KB Output is correct
31 Correct 1 ms 208 KB Output is correct
32 Correct 1 ms 208 KB Output is correct
33 Correct 40 ms 208 KB Output is correct
34 Correct 2 ms 208 KB Output is correct
35 Correct 3 ms 296 KB Output is correct
36 Correct 1 ms 208 KB Output is correct
37 Correct 1 ms 208 KB Output is correct
38 Correct 1 ms 208 KB Output is correct
39 Correct 1 ms 296 KB Output is correct
40 Correct 1 ms 280 KB Output is correct
41 Correct 2 ms 276 KB Output is correct
42 Correct 1 ms 208 KB Output is correct
43 Correct 1 ms 208 KB Output is correct
44 Correct 1 ms 208 KB Output is correct
45 Correct 1 ms 208 KB Output is correct
46 Correct 1 ms 208 KB Output is correct
47 Correct 43 ms 348 KB Output is correct
48 Correct 1 ms 220 KB Output is correct
49 Correct 1 ms 316 KB Output is correct
50 Correct 2 ms 312 KB Output is correct

Subtask #7
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 208 KB Output is correct
2 Correct 1 ms 280 KB Output is correct
3 Correct 1 ms 276 KB Output is correct
4 Correct 1 ms 208 KB Output is correct
5 Correct 1 ms 208 KB Output is correct
6 Correct 1 ms 208 KB Output is correct
7 Correct 1 ms 208 KB Output is correct
8 Correct 41 ms 296 KB Output is correct
9 Correct 1 ms 208 KB Output is correct
10 Correct 1 ms 208 KB Output is correct
11 Correct 2 ms 208 KB Output is correct
12 Correct 1 ms 208 KB Output is correct
13 Correct 1 ms 208 KB Output is correct
14 Correct 1 ms 208 KB Output is correct
15 Correct 1 ms 208 KB Output is correct
16 Correct 1 ms 256 KB Output is correct
17 Correct 41 ms 208 KB Output is correct
18 Correct 1 ms 208 KB Output is correct
19 Correct 1 ms 300 KB Output is correct
20 Correct 1 ms 208 KB Output is correct
21 Correct 1 ms 208 KB Output is correct
22 Correct 1 ms 208 KB Output is correct
23 Correct 1 ms 300 KB Output is correct
24 Correct 1 ms 336 KB Output is correct
25 Correct 1 ms 304 KB Output is correct
26 Correct 1 ms 208 KB Output is correct
27 Correct 1 ms 208 KB Output is correct
28 Correct 1 ms 208 KB Output is correct
29 Correct 1 ms 208 KB Output is correct
30 Correct 1 ms 208 KB Output is correct
31 Correct 1 ms 208 KB Output is correct
32 Correct 1 ms 208 KB Output is correct
33 Correct 40 ms 208 KB Output is correct
34 Correct 1 ms 296 KB Output is correct
35 Correct 1 ms 208 KB Output is correct
36 Correct 1 ms 208 KB Output is correct
37 Correct 1 ms 208 KB Output is correct
38 Correct 1 ms 208 KB Output is correct
39 Correct 1 ms 276 KB Output is correct
40 Correct 1 ms 296 KB Output is correct
41 Correct 1 ms 324 KB Output is correct
42 Correct 1 ms 208 KB Output is correct
43 Correct 1 ms 208 KB Output is correct
44 Correct 2 ms 208 KB Output is correct
45 Correct 1 ms 208 KB Output is correct
46 Correct 1 ms 208 KB Output is correct
47 Correct 3 ms 296 KB Output is correct
48 Correct 1 ms 304 KB Output is correct
49 Correct 1 ms 304 KB Output is correct
50 Correct 41 ms 208 KB Output is correct
51 Correct 1 ms 208 KB Output is correct
52 Correct 1 ms 280 KB Output is correct
53 Correct 1 ms 208 KB Output is correct
54 Correct 1 ms 276 KB Output is correct
55 Correct 1 ms 208 KB Output is correct
56 Correct 1 ms 208 KB Output is correct
57 Correct 1 ms 208 KB Output is correct
58 Correct 40 ms 208 KB Output is correct
59 Correct 2 ms 208 KB Output is correct
60 Correct 3 ms 296 KB Output is correct
61 Correct 1 ms 208 KB Output is correct
62 Correct 1 ms 208 KB Output is correct
63 Correct 1 ms 208 KB Output is correct
64 Correct 1 ms 296 KB Output is correct
65 Correct 1 ms 280 KB Output is correct
66 Correct 2 ms 276 KB Output is correct
67 Correct 1 ms 208 KB Output is correct
68 Correct 1 ms 208 KB Output is correct
69 Correct 1 ms 208 KB Output is correct
70 Correct 1 ms 208 KB Output is correct
71 Correct 1 ms 208 KB Output is correct
72 Correct 43 ms 348 KB Output is correct
73 Correct 1 ms 220 KB Output is correct
74 Correct 1 ms 316 KB Output is correct
75 Correct 2 ms 312 KB Output is correct
76 Correct 2 ms 220 KB Output is correct
77 Correct 1 ms 208 KB Output is correct
78 Correct 1 ms 208 KB Output is correct
79 Correct 1 ms 208 KB Output is correct
80 Correct 1 ms 208 KB Output is correct
81 Correct 1 ms 208 KB Output is correct
82 Correct 1 ms 208 KB Output is correct
83 Correct 1 ms 208 KB Output is correct
84 Correct 1 ms 296 KB Output is correct
85 Correct 1 ms 288 KB Output is correct
86 Correct 2 ms 288 KB Output is correct
87 Correct 44 ms 304 KB Output is correct
88 Correct 1 ms 208 KB Output is correct
89 Correct 1 ms 208 KB Output is correct
90 Correct 1 ms 296 KB Output is correct
91 Correct 42 ms 300 KB Output is correct
92 Correct 1 ms 288 KB Output is correct
93 Correct 1 ms 296 KB Output is correct
94 Correct 1 ms 208 KB Output is correct
95 Correct 1 ms 208 KB Output is correct
96 Correct 1 ms 288 KB Output is correct
97 Correct 1 ms 292 KB Output is correct
98 Correct 1 ms 292 KB Output is correct
99 Correct 44 ms 288 KB Output is correct
100 Correct 1 ms 288 KB Output is correct
101 Correct 1 ms 208 KB Output is correct
102 Correct 1 ms 280 KB Output is correct
103 Correct 1 ms 336 KB Output is correct
104 Correct 1 ms 208 KB Output is correct
105 Correct 1 ms 336 KB Output is correct
106 Correct 1 ms 292 KB Output is correct
107 Correct 1 ms 208 KB Output is correct
108 Correct 1 ms 208 KB Output is correct
109 Correct 1 ms 208 KB Output is correct
110 Correct 1 ms 280 KB Output is correct
111 Correct 1 ms 208 KB Output is correct
112 Correct 1 ms 208 KB Output is correct
113 Correct 3 ms 208 KB Output is correct
114 Correct 1 ms 208 KB Output is correct
115 Correct 1 ms 292 KB Output is correct
116 Correct 45 ms 292 KB Output is correct
117 Correct 1 ms 208 KB Output is correct
118 Correct 1 ms 208 KB Output is correct
119 Correct 1 ms 292 KB Output is correct
120 Correct 1 ms 288 KB Output is correct
121 Correct 1 ms 276 KB Output is correct
122 Correct 42 ms 292 KB Output is correct
123 Correct 1 ms 296 KB Output is correct
124 Correct 1 ms 288 KB Output is correct
125 Correct 1 ms 288 KB Output is correct
126 Correct 1 ms 292 KB Output is correct
127 Correct 1 ms 208 KB Output is correct
128 Correct 1 ms 284 KB Output is correct
129 Correct 1 ms 208 KB Output is correct
130 Correct 3 ms 312 KB Output is correct
131 Correct 52 ms 296 KB Output is correct
132 Correct 1 ms 208 KB Output is correct
133 Correct 1 ms 208 KB Output is correct
134 Correct 1 ms 296 KB Output is correct
135 Correct 43 ms 296 KB Output is correct