oj.uz

Submission #1016256

# Submission time Handle Problem Language Result Execution time Memory
1016256 2024-07-07T15:01:51 Z VMaksimoski008 Poklon (COCI17_poklon) C++17
140 / 140
 302 ms 42068 KB 
#include <bits/stdc++.h>

#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;

struct BIT {
    int n;
    vector<int> tree, val;

    BIT(int _n) : n(_n+10), tree(_n+60), val(_n+60) {}

    void update(int p, int v) {
        val[p] += v;
        for(p++; p<n; p+=p&-p) tree[p] += v;
    }

    ll query(int p) {
        ll ans = 0;
        for(p++; p>0; p-=p&-p) ans += tree[p];
        return ans;
    }

    ll query(int l, int r) { return query(r) - query(l-1); }
    void set(int p, int v) { update(p, v - val[p]); }
};


signed main() {
    ios_base::sync_with_stdio(false);
    cout.tie(0); cin.tie(0);

    int n, q;
    cin >> n >> q;

    vector<int> v(n);
    for(int &x : v) cin >> x;

    vector<pii> qus[n];
    for(int i=0; i<q; i++) {
        int l, r;
        cin >> l >> r;
        qus[r-1].push_back({ l-1, i });
    }

    vector<int> ans(q);
    map<int, vector<int> > mp;
    BIT bit(n);

    for(int i=0; i<n; i++) {
        mp[v[i]].push_back(i);
        int sz = mp[v[i]].size();
        if(sz >= 2) bit.set(mp[v[i]][sz-2], 1);
        if(sz >= 3) bit.set(mp[v[i]][sz-3], -1);
        if(sz >= 4) bit.set(mp[v[i]][sz-4], 0);
        for(auto &[l, id] : qus[i]) ans[id] = bit.query(l, i); 
    }

    for(int &x : ans) cout << x << '\n';
    return 0;
}

Subtask #1
140.0 / 140.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 2 ms 860 KB Output is correct
5 Correct 40 ms 8320 KB Output is correct
6 Correct 43 ms 8472 KB Output is correct
7 Correct 119 ms 17032 KB Output is correct
8 Correct 188 ms 26192 KB Output is correct
9 Correct 242 ms 34460 KB Output is correct
10 Correct 302 ms 42068 KB Output is correct