// https://oj.uz/problem/view/Balkan15_HAPPINESS
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pii;
typedef long double ld;
typedef vector<ll> vi;
typedef vector<pii> vpi;
const ll maxn = 2e5 + 10;
const ll inf = LLONG_MAX;
#define pb push_back
#define ppb pop_back
#define ff first
#define ss second
#define sza(x) ((int)(x).size())
#define all(a) (a).begin(), (a).end()
#define smax(s, v) s = max(s, (v))
#define smin(s, v) s = min(s, (v))
#define FF(i, n) for(int i = 0; i < (n); i++)
#define FF1(i, n) for(int i = 1; i <= (n); i++)
#define FFZ(i, v) FF(i, sza(v))
#define FFS(i, s, n) for(int i = s; i <= (n); i++)
#define FFD(i, n) for(int i = (n)-1; i >= 0; i--)
#define F(n) FF(i, n)
#define F1(n) FF1(i, n)
#define FZ(v) FFZ(i, v)
#define FS(s, n) FFS(i, s, n)
#define FD(n) FFD(i, n)
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << "(" << p.ff << ", " << p.ss << ")"; }
template<typename A> ostream& operator<<(ostream& os, const vector<A>& v) { os << "["; for(const auto& x: v) os << x << ", "; return os << "]"; }
//#define LOCAL
#ifdef LOCAL
void dbg_out() { cerr << endl; }
template<typename H, typename... T> void dbg_out(H h, T... t) { cerr << " " << h; dbg_out(t...); }
#define dbg(...) cerr << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif
#ifndef LOCAL
#include "happiness.h"
#endif
ll n, m, p, q, k;
// 1 indexado
// Note que EL, ER mudaram um pouco
#define EL &(dt[v->i]), l, (l + r) / 2
#define ER &(dt[v->i+1]), (l + r) / 2 + 1, r
#define INI &(seg.dt[0]), 1, n
struct Node;
struct Node {
// cnt eh util so para folha
ll sum, rem;
int i, cnt;
Node() : i(0), sum(0), rem(0), cnt(0) {}
void merge(Node l, Node r) {
sum = l.sum + r.sum;
rem = max(l.rem, r.rem - l.sum);
}
void update(ll v, int d) {
cnt += d;
sum = v * cnt;
if(cnt == 0) rem = 0;
else rem = v-1;
}
};
struct Seg {
vector<Node> dt;
Seg() { dt.pb(Node()); }
// Cuidado se for fazer lazy, tem que verificar se nao eh folha antes
inline void extend(Node *v) {
if(v->i == 0) {
v->i = sza(dt);
dt.pb(Node());
dt.pb(Node());
//dbg("Extend", v->i, v->i+1);
}
}
void update(ll pos, ll x, Node *v, ll l, ll r) {
//dbg("Upd in", pos, x, l, r, v);
if(r < pos || l > pos) {
//dbg("Fora");
return; }
if(l == r) {
v->update(r, x);
// dbg("Folha", v->cnt, v->sum, v->rem);
return;
}
extend(v);
/*if(l == 1 && r == 8) {
dbg("Left addr", &(dt[v->i]));
}*/
update(pos, x, EL); update(pos, x, ER);
v->merge(dt[v->i], dt[v->i+1]);
/*if(l == 1 && r == 8) {
dbg("Left", v->i, &(dt[v->i]), dt[v->i].sum, dt[v->i].rem);
dbg("Right", v->i, dt[v->i+1].sum, dt[v->i+1].rem);
}*/
//dbg("Merge", l, r, v->sum, v->rem, v->i, v);
}
};
Seg seg;
bool init(int coinsCount, long long maxCoinSize, long long coins[]) {
n = maxCoinSize;
seg.dt.reserve(n);
F(coinsCount) seg.update(coins[i], 1, INI);
return seg.dt[0].rem == 0;
}
bool is_happy(int event, int coinsCount, long long coins[]) {
F(coinsCount) seg.update(coins[i], event, INI);
return seg.dt[0].rem == 0;
}
#ifdef LOCAL
ll _coins[maxn];
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
ll n, m, q;
cin >> n >> m;
F(n) cin >> _coins[i];
cout << init(n, m, _coins) << '\n';
cin >> q;
FF(__i, q) {
ll d, k;
cin >> d >> k;
F(k) cin >> _coins[i];
cout << is_happy(d, k, _coins) << "\n";
}
}
#endif
Compilation message
happiness.cpp: In constructor 'Node::Node()':
happiness.cpp:66:6: warning: 'Node::i' will be initialized after [-Wreorder]
66 | int i, cnt;
| ^
happiness.cpp:65:8: warning: 'll Node::sum' [-Wreorder]
65 | ll sum, rem;
| ^~~
happiness.cpp:68:5: warning: when initialized here [-Wreorder]
68 | Node() : i(0), sum(0), rem(0), cnt(0) {}
| ^~~~
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 |
0 ms |
344 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
440 KB |
Execution killed with signal 11 |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
344 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
440 KB |
Execution killed with signal 11 |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
344 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
440 KB |
Execution killed with signal 11 |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
344 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
440 KB |
Execution killed with signal 11 |
| 3 |
Halted |
0 ms |
0 KB |
- |
