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 "happiness.h"
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define sz(a) (int)(a.size())
#define all(a) a.begin(),a.end()
#define lb lower_bound
#define ub upper_bound
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
#define MOD (ll)(1e9+7)
#define INF (ll)(1e9+1)
#define debug(...) fprintf(stderr, __VA_ARGS__),fflush(stderr)
#define time__(d) for(long blockTime = 0; (blockTime == 0 ? (blockTime=clock()) != 0 : false);\
debug("%s time : %.4fs\n", d, (double)(clock() - blockTime) / CLOCKS_PER_SEC))
typedef long long int ll;
typedef long double ld;
typedef pair<ll,ll> PII;
typedef pair<int,int> pii;
typedef vector<vector<int>> vii;
typedef vector<vector<ll>> VII;
ll gcd(ll a,ll b){if(!b)return a;else return gcd(b,a%b);}
struct node{
ll lf,rg,sum;
node *l,*r;
node(ll _lf,ll _rg){
l = nullptr;
r = nullptr;
sum = 0;
lf = _lf;
rg = _rg;
}
void upd(ll val,ll c){
sum+=val*c;
if(lf==rg)return;
int tm = (lf+rg)/2;
if(val<=tm){
if(l == nullptr)l = new node(lf,tm);
l->upd(val,c);
}else{
if(r == nullptr)r = new node(tm+1,rg);
r->upd(val,c);
}
}
ll query(int L,int R){
if(lf == L && R == rg)return sum;
int tm = (lf+rg)/2;
if(R<=tm){
return l->query(L,R);
}
else if(L>tm){
return r->query(L,R);
}else{
return l->query(L,tm) + r->query(tm+1,R);
}
}
};
node *root;
ll m;
bool check(){
ll cur = 1,mx = root->sum;
while(cur<mx){
ll t = root->query(1,min(m,cur));
if(cur>t)return false;
cur = t+1;
}
return true;
}
bool init(int coinsCount, long long maxCoinSize, long long coins[]) {
m = maxCoinSize;
root = new node(1,m);
for(int i=0;i<coinsCount;i++)root->upd(coins[i],1);
return check();
}
bool is_happy(int event, int coinsCount, long long coins[]) {
for(int i=0;i<coinsCount;i++)root->upd(coins[i],event);
return check();
}
Compilation message (stderr)
grader.cpp: In function 'int main()':
grader.cpp:16:12: warning: unused variable 'max_code' [-Wunused-variable]
16 | long long max_code;
| ^~~~~~~~
# | 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... |