#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){
if(l==nullptr)return 0;
return l->query(L,R);
}
else if(L>tm){
if(r==nullptr)return 0;
return r->query(L,R);
}else{
ll res = 0;
if(l)res+=l->query(L,tm);
if(r)res+=r->query(tm+1,R);
return res;
}
}
};
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
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 |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Runtime error |
514 ms |
524292 KB |
Execution killed with signal 9 |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
486 ms |
34308 KB |
Output is correct |
7 |
Correct |
439 ms |
33896 KB |
Output is correct |
8 |
Correct |
457 ms |
34376 KB |
Output is correct |
9 |
Correct |
743 ms |
44152 KB |
Output is correct |
10 |
Correct |
688 ms |
48272 KB |
Output is correct |
11 |
Correct |
165 ms |
33304 KB |
Output is correct |
12 |
Correct |
166 ms |
33288 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Runtime error |
514 ms |
524292 KB |
Execution killed with signal 9 |
7 |
Halted |
0 ms |
0 KB |
- |