CodeForces-940F Machine Learning

Machine Learning

询问一个区间，求区间内数的桶的 MEX，还有单点修改

带修莫队

直接离线用带修莫队，然后维护一个桶的桶，每次询问答案的时候直接找 MEX 就行了

一开始想复杂了，一直想维护 MEX 的值，然后用了一个 set 去维护，但是每次修改的时候都会乘上一个 logn 级别，导致超时

看了答案后，发现直接每次询问的时候查询就行，因为 MEX 的值不可能超过 \(2 * \sqrt{n}\)，所以这里询问的复杂度也就 \(2* m * \sqrt{n}\)，还不如带修莫队本身的 \(n^{\frac{5}{3}}\) 来的高

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <queue>
#include <functional>
#include <map>
#include <set>
#include <cmath>
#include <cstring>
#include <deque>
#include <stack>
using namespace std;
typedef long long ll;
#define pii pair<int, int>
#define endl '\n'
const ll maxn = 2e5 + 10;
const ll inf = 1e17 + 10;
int num[maxn], cnt[maxn], a[maxn];
int alp[maxn], pos[maxn];
pii change[maxn];
int last[maxn];

struct node
{
    int l, r, t, id;
    node(){}
    node(int _l, int _r, int _t, int _id){l = _l, r = _r, t = _t, id = _id;}
}seg[maxn];

bool cmp(const node& a, const node& b)
{
    return pos[a.l] ^ pos[b.l] ? pos[a.l] < pos[b.l] : (pos[a.r] ^ pos[b.r] ? pos[a.r] < pos[b.r] : a.t < b.t);
}

inline void add(int x)
{
    if(alp[num[x]]) cnt[alp[num[x]]]--;
    alp[num[x]]++;
    if(alp[num[x]]) cnt[alp[num[x]]]++;
}

inline void del(int x)
{
    if(alp[num[x]]) cnt[alp[num[x]]]--;
    alp[num[x]]--;
    if(alp[num[x]]) cnt[alp[num[x]]]++;
} 

inline void update(int x, int way)
{
    if(way)
    {
        del(change[x].first);
        swap(change[x].second, num[change[x].first]);
        add(change[x].first);
    }
    else swap(change[x].second, num[change[x].first]);
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, m;
    cin >> n >> m;
    for(int i=1; i<=n; i++)
    {
        cin >> num[i];
        a[i] = num[i];
    }
    int ta = n;
    
    int tq = 0, tc = 0;
    for(int i=0; i<m; i++)
    {
        int t, l, r;
        cin >> t >> l >> r;
        if(t == 1)
        {
            seg[tq] = node(l, r, tc, tq);
            tq++;
        }
        else
        {
            change[tc] = make_pair(l, r);
            a[++ta] = r;
            tc++;
        }
    }
    sort(a + 1, a + ta + 1);
    int tp = unique(a + 1, a + 1 + ta) - a - 1;
    for(int i=1; i<=n; i++) num[i] = lower_bound(a + 1, a + 1 + tp, num[i]) - a - 1;
    for(int i=0; i<tc; i++) change[i].second = lower_bound(a + 1, a + 1 + tp, change[i].second) - a - 1;
    cnt[0] = tp;
    int nn = pow(n, 2.0 / 3.0);
    for(int i=1; i<=n; i++) pos[i] = i / nn;
    sort(seg, seg + tq, cmp);

    int l = 1, r = 0, t = 0;
    for(int i=0; i<tq; i++)
    {
        while(seg[i].t < t) {--t; update(t, (change[t].first <= r && change[t].first >= l));}
        while(seg[i].t > t) {update(t, (change[t].first <= r && change[t].first >= l)); t++;}
        while(r < seg[i].r) add(++r);
        while(l < seg[i].l) del(l++);
        while(l > seg[i].l) add(--l);
        while(r > seg[i].r) del(r--);
        for(int j=1; j<=ta; j++) if(cnt[j] == 0) {last[seg[i].id] = j; break;}
    }
    for(int i=0; i<tq; i++)
        cout << last[i] << endl;
    return 0;
}

标签：int,CodeForces,pos,Machine,num,change,Learning,include,alp
来源： https://www.cnblogs.com/dgsvygd/p/16320684.html