#include <bits/stdc++.h>
#define X first
#define Y second
#define pb push_back
#define pii pair<int, int>
typedef long long ll;
using namespace std;
const int MOD = 1e9 + 7;
const ll INF = 1e18;
const int OFF = (1 << 20);
const int MAXN = 400005;
int n, m, a, q;
vector<int> v[MAXN];
pair< pii, int > w[MAXN];
ll loga[MAXN];
void update(int x, int y) {
x++;
for (; x <= 300000; x += x & -x) loga[x] += y;
}
ll query(int x) {
x++;
ll out = 0;
for (; x > 0; x -= x & -x) out += loga[x];
return out;
}
vector< pair<pii, int> > gdje[MAXN];
vector< pair<pii, int> > slj[MAXN];
int sol[MAXN];
int treba[MAXN];
int main () {
cin >> n >> m;
for (int i = 0; i < m; i++) {
cin >> a;
v[a-1].push_back(i);
}
for (int i = 0; i < n; i++) cin >> treba[i];
cin >> q;
for (int i = 0; i < q; i++) {
cin >> w[i].X.X >> w[i].X.Y >> w[i].Y;
w[i].X.X --;
w[i].X.Y --;
}
/*for (int i = 0; i < q; i++) {
if (w[i].X.X <= w[i].X.Y) {
update(w[i].X.X, w[i].Y);
update(w[i].X.Y+1, -w[i].Y);
}
else {
update(w[i].X.X, w[i].Y);
update(0, w[i].Y);
update(w[i].X.Y+1, -w[i].Y);
}
}*/
int kol = n;
for (int i = 0; i < n; i++) {
gdje[(q-1)/2].push_back({{0, q-1}, i});
}
memset(sol, -1, sizeof sol);
int br = 0;
while (kol > 0) {
for (int i = 0; i < q; i++) {
if (w[i].X.X <= w[i].X.Y) {
update(w[i].X.X, w[i].Y);
update(w[i].X.Y+1, -w[i].Y);
}
else {
update(w[i].X.X, w[i].Y);
update(0, w[i].Y);
update(w[i].X.Y+1, -w[i].Y);
}
/*cout << "QUERY" << endl;
for (int j = 0; j < m; j++) cout << query(j) << " ";
cout << endl; */
for (int j = 0; j < gdje[i].size(); j++) {
unsigned long long zbr = 0;
for (int k = 0; k < v[gdje[i][j].Y].size(); k++) {
//cout << "dodaj poziciju " << v[gdje[i][j].Y][k] << endl;
zbr += query(v[gdje[i][j].Y][k]);
if (zbr >= treba[gdje[i][j].Y]) break;
}
//cout << gdje[i][j].X.X << " do " << gdje[i][j].X.Y << ": " << gdje[i][j].Y << " " << zbr << endl;
//system("pause");
if (gdje[i][j].X.X == gdje[i][j].X.Y) {
if (zbr >= treba[gdje[i][j].Y]) sol[gdje[i][j].Y] = i+1;
else sol[gdje[i][j].Y] = -1;
kol--;
}
else {
if (zbr < treba[gdje[i][j].Y]) slj[(i+1+gdje[i][j].X.Y)/2].push_back({{i+1, gdje[i][j].X.Y}, gdje[i][j].Y});
else {
slj[(gdje[i][j].X.X+i)/2].push_back({{gdje[i][j].X.X, i}, gdje[i][j].Y});
}
}
}
}
for (int i = 0; i < q; i++) {
gdje[i].clear();
gdje[i] = slj[i];
slj[i].clear();
}
memset(loga, 0, sizeof loga);
br++;
}
for (int i = 0; i < n; i++) {
if (sol[i] == -1) cout << "NIE\n";
else cout << sol[i] << "\n";
}
return 0;
}
Compilation message
met.cpp: In function 'int main()':
met.cpp:88:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<int, int>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
88 | for (int j = 0; j < gdje[i].size(); j++) {
| ~~^~~~~~~~~~~~~~~~
met.cpp:90:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
90 | for (int k = 0; k < v[gdje[i][j].Y].size(); k++) {
| ~~^~~~~~~~~~~~~~~~~~~~~~~~
met.cpp:93:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
93 | if (zbr >= treba[gdje[i][j].Y]) break;
| ~~~~^~~~~~~~~~~~~~~~~~~~~~
met.cpp:100:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
100 | if (zbr >= treba[gdje[i][j].Y]) sol[gdje[i][j].Y] = i+1;
| ~~~~^~~~~~~~~~~~~~~~~~~~~~
met.cpp:106:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
106 | if (zbr < treba[gdje[i][j].Y]) slj[(i+1+gdje[i][j].X.Y)/2].push_back({{i+1, gdje[i][j].X.Y}, gdje[i][j].Y});
| ~~~~^~~~~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
25 ms |
33260 KB |
Output is correct |
| 2 |
Correct |
25 ms |
33388 KB |
Output is correct |
| 3 |
Correct |
24 ms |
33260 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
23 ms |
33388 KB |
Output is correct |
| 2 |
Correct |
24 ms |
33388 KB |
Output is correct |
| 3 |
Correct |
25 ms |
33540 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
144 ms |
36076 KB |
Output is correct |
| 2 |
Correct |
242 ms |
44028 KB |
Output is correct |
| 3 |
Correct |
206 ms |
40496 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
212 ms |
38832 KB |
Output is correct |
| 2 |
Correct |
213 ms |
38828 KB |
Output is correct |
| 3 |
Correct |
254 ms |
44432 KB |
Output is correct |
| 4 |
Correct |
74 ms |
38820 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
182 ms |
36768 KB |
Output is correct |
| 2 |
Correct |
257 ms |
44944 KB |
Output is correct |
| 3 |
Correct |
195 ms |
34156 KB |
Output is correct |
| 4 |
Correct |
247 ms |
42088 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
228 ms |
34296 KB |
Output is correct |
| 2 |
Correct |
224 ms |
39192 KB |
Output is correct |
| 3 |
Correct |
185 ms |
35180 KB |
Output is correct |
| 4 |
Correct |
291 ms |
46056 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
1165 ms |
65540 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
1145 ms |
65536 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
| 2 |
Halted |
0 ms |
0 KB |
- |
