Submission #1353233

#TimeUsernameProblemLanguageResultExecution timeMemory
1353233bakhtiyarnSum Zero (RMI20_sumzero)C++20
61 / 100
241 ms30580 KiB
// GPT changed it to ternery lifting
#include <bits/stdc++.h>
using namespace std;

const int N = 4e5 + 5;
const int LOG = 13; // 3^12 > 4e5
int up[N][LOG];
int idx[N];
long long pw[LOG];

void solve() {
    int n; 
    cin >> n;

    pw[0] = 1;
    for (int i = 1; i < LOG; i++) pw[i] = pw[i - 1] * 3;

    for (int i = 0; i <= n; i++) idx[i] = 1e9;

    unordered_map<int, int> last;
    int pre = 0;

    for (int i = 0; i <= n; i++) {
        if (i) {
            int x; cin >> x;
            pre += x;
        }

        if (last.count(pre)) up[i][0] = last[pre];
        else up[i][0] = -1;

        last[pre] = i;

        if (up[i][0] != -1) idx[up[i][0]] = min(idx[up[i][0]], i);
    }

    int mn = 1e9;
    for (int i = n; i >= 0; i--) {
        if (idx[i] != 1e9) mn = min(mn, idx[i]);
        up[i][0] = (mn == 1e9 ? n + 1 : mn);

        for (int k = 1; k < LOG; k++) {
            int a = up[i][k - 1];
            if (a > n) up[i][k] = n + 1;
            else {
                int b = up[a][k - 1];
                if (b > n) up[i][k] = n + 1;
                else up[i][k] = up[b][k - 1];
            }
        }
    }

    int q; cin >> q;
    while (q--) {
        int l, r; cin >> l >> r;
        l--;

        long long ans = 0;
        for (int b = LOG - 1; b >= 0; b--) {
            for (int rep = 0; rep < 2; rep++) { // base 3 digit can be 0,1,2
                if (up[l][b] > r) break;
                ans += pw[b];
                l = up[l][b];
            }
        }

        cout << ans << '\n';
    }
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    solve();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...