Submission #1171615

#	Time	Username	Problem	Language	Result	Execution time	Memory
1171615		quannnguyen2009	Sum Zero (RMI20_sumzero)	C++20	100 / 100	274 ms	19880 KiB

sumzero

#include<bits/stdc++.h>
// #define int long long
#define fi first
#define se second
#define pb push_back
#define ii pair<int, int>
#define sz(v) (int)v.size()
#define all(v) v.begin(), v.end()
using namespace std;

const int N=4e5+5, mod = 1e9+7, L = 7;

int n, q;
pair<long long, int> a[N];
int up[N][L];
int pw[L];

signed main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n;
    pw[0] = 1;
    for (int i=1; i<L; i++) pw[i] = pw[i-1]*7;
    for (int i=1; i<=n; i++) {
        cin >> a[i].fi;
        a[i].fi += a[i-1].fi;
        a[i].se = i;
    }
    sort(a, a+n+1);
    up[n+1][0] = n+1;
    a[n+1].fi = 100000000000000000;
    for (int i=0; i<=n; i++) {
        if(a[i].fi==a[i+1].fi) up[a[i].se][0] = a[i+1].se;
        else up[a[i].se][0] = n+1;
    }
    for (int i=n; i>=0; i--) up[i][0] = min(up[i][0], up[i+1][0]);
    for (int i=1; i<L; i++) {
        up[n+1][i] = n+1;
        for (int j=0; j<=n; j++) up[j][i] = up[up[up[up[up[up[up[j][i-1]][i-1]][i-1]][i-1]][i-1]][i-1]][i-1];
    }
    cin >> q;
    while(q--) {
        int l, r; cin >> l >> r;
        int idx = l-1;
        long long ans = 0;
        for (int i=L-1; i>=0; i--) {
            while(up[idx][i]<=r) {
                idx = up[idx][i];
                ans += pw[i];
            }
        }
        cout << ans << '\n';
    }
}

• Subtask 1: 22 / 22
• Subtask 2: 39 / 39
• Subtask 3: 39 / 39