This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "biscuits.h"
#include "bits/stdc++.h"
using namespace std;
#define forR(i, x) for(int i = 0; i < (x); ++i)
#define REP(i, a, b) for(int i = (a); i < (b); ++i)
#define all(x) x.begin(), x.end()
typedef long long ll;
typedef vector<ll> vll;
ll valOf(ll msk, vll& dc) {
ll ret = 0;
forR(i, dc.size()) if(msk & (1 << i)) ret += dc[dc.size() - 1 - i];
return ret;
}
long long count_tastiness(long long x, std::vector<long long> a) {
a.insert(a.begin(), (ll) 0);
int k = a.size();
vll ftp(k);
vll sm(k);
sm[0] = a[0];
REP(i, 1, k) sm[i] = sm[i-1] + (1 << (i-1)) * a[i];
forR(i, k) {
if(i == 0) ftp[i] = 1;
else {
cerr << i << ": \n";
ll st = sm[i];
vll dc = {(1 << (i-1)) * x};
ll msk = 0;
for(int j = i-1; j >= 0; --j) {
// st - dc(lo) >= sm[j], st - dc(hi) < sm[j]
if(st - valOf(msk, dc) >= sm[j]) {
ll lo=msk, hi=(1 << dc.size());
while(hi - lo > 1) {
int mid = (lo+hi)/2;
if(st - valOf(mid, dc) >= sm[j]) lo = mid;
else hi = mid;
}
ftp[i] += (hi - msk) * ftp[j];
cerr << j << ' ' << hi << ' ' << ftp[i] << '\n';
if(hi == (1 << dc.size())) break;
msk = hi << 1;
dc.push_back((1 << (j-1)) * x);
}
}
}
}
forR(i, k) cerr << ftp[i] << ' ';
cerr << '\n';
return ftp[k-1];
}
Compilation message (stderr)
biscuits.cpp: In function 'll valOf(ll, vll&)':
biscuits.cpp:4:37: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
4 | #define forR(i, x) for(int i = 0; i < (x); ++i)
| ^
biscuits.cpp:12:2: note: in expansion of macro 'forR'
12 | forR(i, dc.size()) if(msk & (1 << i)) ret += dc[dc.size() - 1 - i];
| ^~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |