oj.uz

Submission #482724

# Submission time Handle Problem Language Result Execution time Memory
482724 2021-10-26T06:40:26 Z Fysty A Difficult(y) Choice (BOI21_books) C++17
100 / 100
 4 ms 328 KB 
#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]

Subtask #1
5.0 / 5.0

# 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

Subtask #2
15.0 / 15.0

# 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

Subtask #3
10.0 / 10.0

# 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

Subtask #4
15.0 / 15.0

# 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

Subtask #5
15.0 / 15.0

# 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

Subtask #6
20.0 / 20.0

# 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

Subtask #7
20.0 / 20.0

# 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