#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 3e5 + 2;
int n, m, a[maxn], q, tree[maxn], ans[maxn], bad[maxn];
vector <tuple<int, int, int>> query;
vector <int> st[maxn], check[maxn];
pair <int, int> b[maxn];
void update(int pos, int val)
{
for (int i = pos; i <= m; i += (i & (-i)))
{
tree[i] += val;
}
}
int get(int pos)
{
int ret = 0;
for (int i = pos; i >= 1; i -= (i & (-i)))
{
ret += tree[i];
}
return ret;
}
void query_apply(int i)
{
auto [u, v, f] = query[i - 1];
if (u <= v)
{
update(u, f);
update(v + 1, -f);
}
else
{
update(1, f);
update(v + 1, -f);
update(u, f);
}
}
signed main()
{
ios::sync_with_stdio(0); cin.tie(0);
memset(ans, -1, sizeof(ans));
memset(bad, -1, sizeof(bad));
//freopen("input.inp", "r", stdin);
//freopen("output.out", "w", stdout);
cin >> n >> m;
for (int i = 1, t; i <= m; i++)
{
cin >> t;
st[t].push_back(i);
}
for (int i = 1; i <= n; i++)
cin >> a[i];
cin >> q;
for (int i = 1; i <= q; i++)
{
int l, r, w;
cin >> l >> r >> w;
query.push_back({l, r, w});
}
for (int i = 1; i <= n; i++)
{
b[i] = {1, q + 1};
}
int cnt = n;
while (1)
{
bool kt = 0;
memset(tree, 0, sizeof(tree));
for (int i = 1; i <= q; i++)
check[i].clear();
for (int i = 1; i <= n; i++)
{
auto [l, r] = b[i];
if (l < r)
{
int mid = (l + r) / 2;
check[mid].push_back(i);
kt = 1;
}
}
if (!kt)
break;
for (int i = 1; i <= q; i++)
{
query_apply(i);
for (int p : check[i])
{
int &l = b[p].first, &r = b[p].second;
int mid = (l + r) / 2;
int sum = 0;
for (int child : st[p])
{
sum += get(child);
if (sum >= a[p])
break;
}
if (sum < a[p])
{
l = mid + 1;
}
else
{
r = mid;
}
}
}
}
for (int i = 1; i <= n; i++)
if (b[i].second <= q)
cout << b[i].second << '\n';
else
cout << "NIE\n";
}
Compilation message
met.cpp: In function 'int main()':
met.cpp:86:6: warning: unused variable 'cnt' [-Wunused-variable]
86 | int cnt = n;
| ^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
12 ms |
21460 KB |
Output is correct |
2 |
Correct |
13 ms |
21460 KB |
Output is correct |
3 |
Correct |
11 ms |
21472 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
21460 KB |
Output is correct |
2 |
Correct |
11 ms |
21576 KB |
Output is correct |
3 |
Correct |
14 ms |
21608 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
61 ms |
24128 KB |
Output is correct |
2 |
Correct |
107 ms |
27568 KB |
Output is correct |
3 |
Correct |
92 ms |
26264 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
90 ms |
25444 KB |
Output is correct |
2 |
Correct |
110 ms |
25552 KB |
Output is correct |
3 |
Correct |
105 ms |
27676 KB |
Output is correct |
4 |
Correct |
34 ms |
24472 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
24532 KB |
Output is correct |
2 |
Correct |
90 ms |
27804 KB |
Output is correct |
3 |
Correct |
50 ms |
22728 KB |
Output is correct |
4 |
Correct |
97 ms |
26872 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
81 ms |
23280 KB |
Output is correct |
2 |
Correct |
97 ms |
25384 KB |
Output is correct |
3 |
Correct |
73 ms |
23740 KB |
Output is correct |
4 |
Correct |
113 ms |
28992 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
953 ms |
52880 KB |
Output is correct |
2 |
Correct |
611 ms |
34512 KB |
Output is correct |
3 |
Correct |
222 ms |
29972 KB |
Output is correct |
4 |
Runtime error |
955 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
935 ms |
50640 KB |
Output is correct |
2 |
Correct |
657 ms |
34512 KB |
Output is correct |
3 |
Correct |
183 ms |
29612 KB |
Output is correct |
4 |
Runtime error |
938 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Halted |
0 ms |
0 KB |
- |