oj.uz

Submission #464022

# Submission time Handle Problem Language Result Execution time Memory
464022 2021-08-12T08:46:48 Z AriaH A Difficult(y) Choice (BOI21_books) C++17
100 / 100
 17 ms 328 KB 
/** work as hard as you can and keep the first place **/

#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
#include "books.h"

using namespace std;

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

#define F					first
#define S					second
#define all(x)				x.begin(), x.end()
#define SZ(x)				(int)x.size()
#define Mp					make_pair

const int N = 1e6 + 10;
const ll mod = 1e9 + 7;
const ll inf = 8e18;
const int LOG = 20;

ll pw(ll a, ll b, ll M = mod, ll ret = 1) { while(b) { ret = ret * (b & 1? a : 1) % M, a = a * a % M, b >>= 1; } return ret; }

map < ll, ll > mp;

/*ll skim(int i)
{
	return 1;
}*/

ll ask(int i)
{
	if(mp.count(i))
	{
		return mp[i];
	}
	return mp[i] = skim(i);
}

/*void answer(vector < int > vec)
{

}

void impossible()
{

}
*/

void solve(int n, int k, ll A, int s)
{
	mp.clear();
	int d = 0, up = n + 1;
	while(up - d > 1)
	{
		int mid = (up + d) >> 1;
		if(ask(mid) < A) d = mid;
		else up = mid;
	}
	if(up > n) up --;
	if(up < k) impossible();
	vector < pll > vec;
	for(int i = 1; i <= min(k, up - k); i ++)
	{
		vec.push_back(Mp(ask(i), i));
	}
	for(int i = up - k + 1; i <= up; i ++)
	{
		vec.push_back(Mp(ask(i), i));
	}
	sort(all(vec));
	int sz = SZ(vec);
	for(int mask = 0; mask < 1 << sz; mask ++)
	{
		if(__builtin_popcount(mask) != k) continue;
		ll sum = 0;
		for(int i = 0; i < sz; i ++)
		{
			if(mask >> i & 1)
			{
				sum += vec[i].F;
			}
		}
		if(A <= sum && sum <= 2ll * A)
		{
			vector < int > ret;
			for(int i = 0; i < sz; i ++)
			{
				if(mask >> i & 1) ret.push_back(vec[i].S);
			}
			answer(ret);
			return;
		}
	}
	impossible();
}

/*int main()
{
	return 0;
}
*/
/** test corner cases(n = 1?) watch for overflow or minus indices **/

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 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 1 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 200 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 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 1 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 200 KB Output is correct
9 Correct 1 ms 292 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 200 KB Output is correct
12 Correct 16 ms 296 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 9 ms 200 KB Output is correct
15 Correct 9 ms 200 KB Output is correct
16 Correct 16 ms 200 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 320 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 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 16 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct

Subtask #4
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 320 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 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 16 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 268 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 200 KB Output is correct
15 Correct 0 ms 200 KB Output is correct
16 Correct 1 ms 200 KB Output is correct
17 Correct 17 ms 268 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 276 KB Output is correct
20 Correct 1 ms 320 KB Output is correct
21 Correct 1 ms 200 KB Output is correct
22 Correct 1 ms 200 KB Output is correct
23 Correct 8 ms 292 KB Output is correct
24 Correct 8 ms 320 KB Output is correct
25 Correct 8 ms 200 KB Output is correct

Subtask #5
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 320 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 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 16 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 268 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 200 KB Output is correct
15 Correct 0 ms 200 KB Output is correct
16 Correct 1 ms 200 KB Output is correct
17 Correct 17 ms 268 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 276 KB Output is correct
20 Correct 1 ms 320 KB Output is correct
21 Correct 1 ms 200 KB Output is correct
22 Correct 1 ms 200 KB Output is correct
23 Correct 8 ms 292 KB Output is correct
24 Correct 8 ms 320 KB Output is correct
25 Correct 8 ms 200 KB Output is correct
26 Correct 1 ms 200 KB Output is correct
27 Correct 1 ms 200 KB Output is correct
28 Correct 1 ms 272 KB Output is correct
29 Correct 1 ms 200 KB Output is correct
30 Correct 1 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 17 ms 272 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 200 KB Output is correct
36 Correct 1 ms 272 KB Output is correct
37 Correct 1 ms 200 KB Output is correct
38 Correct 1 ms 200 KB Output is correct
39 Correct 8 ms 288 KB Output is correct
40 Correct 8 ms 200 KB Output is correct
41 Correct 8 ms 200 KB Output is correct
42 Correct 1 ms 200 KB Output is correct
43 Correct 1 ms 200 KB Output is correct
44 Correct 2 ms 200 KB Output is correct
45 Correct 1 ms 200 KB Output is correct
46 Correct 1 ms 260 KB Output is correct
47 Correct 2 ms 200 KB Output is correct
48 Correct 8 ms 328 KB Output is correct
49 Correct 9 ms 280 KB Output is correct
50 Correct 16 ms 200 KB Output is correct

Subtask #6
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 320 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 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 16 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 268 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 200 KB Output is correct
15 Correct 0 ms 200 KB Output is correct
16 Correct 1 ms 200 KB Output is correct
17 Correct 17 ms 268 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 276 KB Output is correct
20 Correct 1 ms 320 KB Output is correct
21 Correct 1 ms 200 KB Output is correct
22 Correct 1 ms 200 KB Output is correct
23 Correct 8 ms 292 KB Output is correct
24 Correct 8 ms 320 KB Output is correct
25 Correct 8 ms 200 KB Output is correct
26 Correct 1 ms 200 KB Output is correct
27 Correct 1 ms 200 KB Output is correct
28 Correct 1 ms 200 KB Output is correct
29 Correct 1 ms 200 KB Output is correct
30 Correct 1 ms 256 KB Output is correct
31 Correct 1 ms 200 KB Output is correct
32 Correct 1 ms 200 KB Output is correct
33 Correct 17 ms 200 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 200 KB Output is correct
36 Correct 1 ms 200 KB Output is correct
37 Correct 1 ms 200 KB Output is correct
38 Correct 1 ms 200 KB Output is correct
39 Correct 9 ms 200 KB Output is correct
40 Correct 10 ms 292 KB Output is correct
41 Correct 11 ms 200 KB Output is correct
42 Correct 1 ms 200 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 200 KB Output is correct
47 Correct 17 ms 200 KB Output is correct
48 Correct 9 ms 200 KB Output is correct
49 Correct 9 ms 200 KB Output is correct
50 Correct 9 ms 288 KB Output is correct

Subtask #7
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 320 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 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 16 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 268 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 200 KB Output is correct
15 Correct 0 ms 200 KB Output is correct
16 Correct 1 ms 200 KB Output is correct
17 Correct 17 ms 268 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 276 KB Output is correct
20 Correct 1 ms 320 KB Output is correct
21 Correct 1 ms 200 KB Output is correct
22 Correct 1 ms 200 KB Output is correct
23 Correct 8 ms 292 KB Output is correct
24 Correct 8 ms 320 KB Output is correct
25 Correct 8 ms 200 KB Output is correct
26 Correct 1 ms 200 KB Output is correct
27 Correct 1 ms 200 KB Output is correct
28 Correct 1 ms 272 KB Output is correct
29 Correct 1 ms 200 KB Output is correct
30 Correct 1 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 17 ms 272 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 200 KB Output is correct
36 Correct 1 ms 272 KB Output is correct
37 Correct 1 ms 200 KB Output is correct
38 Correct 1 ms 200 KB Output is correct
39 Correct 8 ms 288 KB Output is correct
40 Correct 8 ms 200 KB Output is correct
41 Correct 8 ms 200 KB Output is correct
42 Correct 1 ms 200 KB Output is correct
43 Correct 1 ms 200 KB Output is correct
44 Correct 2 ms 200 KB Output is correct
45 Correct 1 ms 200 KB Output is correct
46 Correct 1 ms 260 KB Output is correct
47 Correct 2 ms 200 KB Output is correct
48 Correct 8 ms 328 KB Output is correct
49 Correct 9 ms 280 KB Output is correct
50 Correct 16 ms 200 KB Output is correct
51 Correct 1 ms 200 KB Output is correct
52 Correct 1 ms 200 KB Output is correct
53 Correct 1 ms 200 KB Output is correct
54 Correct 1 ms 200 KB Output is correct
55 Correct 1 ms 256 KB Output is correct
56 Correct 1 ms 200 KB Output is correct
57 Correct 1 ms 200 KB Output is correct
58 Correct 17 ms 200 KB Output is correct
59 Correct 1 ms 200 KB Output is correct
60 Correct 1 ms 200 KB Output is correct
61 Correct 1 ms 200 KB Output is correct
62 Correct 1 ms 200 KB Output is correct
63 Correct 1 ms 200 KB Output is correct
64 Correct 9 ms 200 KB Output is correct
65 Correct 10 ms 292 KB Output is correct
66 Correct 11 ms 200 KB Output is correct
67 Correct 1 ms 200 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 200 KB Output is correct
72 Correct 17 ms 200 KB Output is correct
73 Correct 9 ms 200 KB Output is correct
74 Correct 9 ms 200 KB Output is correct
75 Correct 9 ms 288 KB Output is correct
76 Correct 0 ms 200 KB Output is correct
77 Correct 3 ms 200 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 1 ms 200 KB Output is correct
85 Correct 1 ms 200 KB Output is correct
86 Correct 1 ms 296 KB Output is correct
87 Correct 16 ms 288 KB Output is correct
88 Correct 1 ms 296 KB Output is correct
89 Correct 9 ms 200 KB Output is correct
90 Correct 10 ms 328 KB Output is correct
91 Correct 16 ms 200 KB Output is correct
92 Correct 1 ms 200 KB Output is correct
93 Correct 1 ms 200 KB Output is correct
94 Correct 1 ms 200 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 16 ms 272 KB Output is correct
100 Correct 1 ms 200 KB Output is correct
101 Correct 1 ms 264 KB Output is correct
102 Correct 1 ms 272 KB Output is correct
103 Correct 1 ms 200 KB Output is correct
104 Correct 1 ms 200 KB Output is correct
105 Correct 9 ms 292 KB Output is correct
106 Correct 9 ms 200 KB Output is correct
107 Correct 9 ms 292 KB Output is correct
108 Correct 1 ms 200 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 2 ms 200 KB Output is correct
114 Correct 8 ms 292 KB Output is correct
115 Correct 9 ms 200 KB Output is correct
116 Correct 16 ms 200 KB Output is correct
117 Correct 1 ms 200 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 200 KB Output is correct
122 Correct 16 ms 200 KB Output is correct
123 Correct 9 ms 292 KB Output is correct
124 Correct 8 ms 320 KB Output is correct
125 Correct 9 ms 268 KB Output is correct
126 Correct 1 ms 200 KB Output is correct
127 Correct 1 ms 200 KB Output is correct
128 Correct 1 ms 200 KB Output is correct
129 Correct 1 ms 200 KB Output is correct
130 Correct 2 ms 200 KB Output is correct
131 Correct 16 ms 200 KB Output is correct
132 Correct 1 ms 200 KB Output is correct
133 Correct 9 ms 288 KB Output is correct
134 Correct 10 ms 200 KB Output is correct
135 Correct 16 ms 200 KB Output is correct