Submission #646455

#	Time	Username	Problem	Language	Result	Execution time	Memory
646455		borgar02	Sum Zero (RMI20_sumzero)	C++17	61 / 100	335 ms	31256 KiB

sumzero

#include <bits/stdc++.h>

using namespace std;
unsigned seed = chrono :: system_clock :: now().time_since_epoch().count();
mt19937 rng(seed);
using ll = long long;
const int N = 4e5 + 5;
ll c[N];
int dp[N][7];
pair <int, int> que[N];
int res[N];

int main()
{
    ios_base :: sync_with_stdio(false); cin.tie(0);
    int n, q;
    cin >> n;
    for(int i = 1; i <= n; i++)
        cin >> c[i];
    cin >> q;
    for(int i = 1; i <= q; i++) {
        cin >> que[i].first >> que[i].second;
        que[i].first--;
    }
    partial_sum(c + 1, c + n + 1, c + 1);
    map <ll, int> m;
    dp[n + 1][0] = n + 1;
    for(int i = n; i >= 0; i--) {
        dp[i][0] = (m.count(c[i]) ? min(m[c[i]], dp[i + 1][0]) : dp[i + 1][0]);
        m[c[i]] = i;
    }
    auto precalc = [&](int to, int from) {
        dp[n + 1][to] = n + 1;
        for(int i = n; i >= 0; i--)
            dp[i][to] = dp[dp[i][from]][from];
    };
    for(int k = 3; k >= 1; k--) {
        int kk = 6 * (k - 1);
        for(int j = 1; j < 7; j++) {
            dp[n + 1][j] = n + 1;
            precalc(j, j - 1);
        }
        for(int j = 2; j <= k; j++) {
            precalc(1, 6);
            for(int  j = 2; j < 7; j++)
                precalc(j, j - 1);
        }
        for(int i = 1; i <= q; i++) {
            for(int j = 6; j >= (k != 1); j--)
                if(dp[que[i].first][j] <= que[i].second)
                    res[i] |= (1 << (kk + j)),
                    que[i].first = dp[que[i].first][j];
            if(k == 1) cout << res[i] << "\n";
        }
    }
    return 0;
}

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