이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 4e5 + 5;
int n, q, nxt[N], ku[N][10];
map<ll, int> mp;
void build_rmq() {
for (int i = 1; i <= n; ++i) ku[i][0] = nxt[i];
int cur, ti;
for (int j = 1; j <= 9; ++j)
for (int i = 1; i + (1 << (j * 2)) - 1 <= n; ++i) {
cur = i;
for (int ti = 1; ti <= 4; ++ti) {
cur = ku[cur][j - 1];
if (cur == 0) break;
else cur++;
}
if (cur == 0) {
ku[i][j] = 0;
continue;
}
ku[i][j] = cur - 1;
}
}
void build() {
cin >> n;
ll sum = 0, x;
mp[0] = 0;
for (int i = 1; i <= n; ++i) nxt[i] = -1;
for (int i = 1; i <= n; ++i) {
cin >> x;
sum += x;
if (mp.count(sum)) nxt[mp[sum] + 1] = i;
mp[sum] = i;
}
mp.clear();
int pre = -1;
for (int i = n; i >= 1; --i) {
if (nxt[i] != -1) pre = (pre == -1 ? nxt[i] : min(pre, nxt[i]));
else nxt[i] = 0;
nxt[i] = (pre != -1 ? pre : 0);
}
build_rmq();
}
int ans;
int que(int l, int r) {
ans = 0;
for (int i = 9; i >= 0; --i)
while (ku[l][i] != 0 && ku[l][i] <= r) {
ans += 1 << (i * 2);
l = ku[l][i] + 1;
}
return ans;
}
void ouput() {
cin >> q;
while (q--) {
static int l, r; cin >> l >> r;
cout << que(l, r) << '\n';
}
}
void solve() {
build();
ouput();
}
signed main() {
cin.tie(0) -> sync_with_stdio(0);
solve();
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
sumzero.cpp: In function 'void build_rmq()':
sumzero.cpp:12:12: warning: unused variable 'ti' [-Wunused-variable]
12 | int cur, ti;
| ^~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |