「Codeforces 940F」Machine Learning

最新推荐文章于 2021-11-15 22:15:04 发布

原创最新推荐文章于 2021-11-15 22:15:04 发布 · 514 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用带修改的莫队算法解决特定数据结构问题的方法。该算法能在O(n^(5/3))的时间复杂度内处理包括询问和修改操作在内的各种任务。

初始时有一个长度为 $n$ 数字串。有以下 $q$ 个操作：
1.询问。定义 $c_i$ 为区间 $[l,r]$ 中 $i$ 出现的次数，求出 $\mathrm{Mex} \{c_0,c_1,...\}$ ，其中 $\mathrm{Mex}$ 表示集合中未出现的最小的正整数。
2.修改。将第 $p$ 个数改成 $x$ 。
$(1\le n,q\le 100 000)$

这题一眼看上去就是一道数据结构题（没想到居然这么暴力，不过我居然还是没在赛场上码完）。
秒解：带修改的莫队算法 $O(n^{\frac{5}{3}})$ 。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <map>
#define R register
int n, q, a[100010], t[100010], d[3000], s[200010], S[100010], u[100010], Ans[100010];
std::map<int, int> M; int cnt;
int F()
{
    R int x; R char ch;
    while(ch = getchar(), ch < '0' || ch > '9'); x = ch - '0';
    while(ch = getchar(), ch >= '0' && ch <= '9') x = x * 10 + ch - '0';
    return x;
}
struct Data1{ int i, l, r, t; } A[100010];
struct Data2{ int i, p, x, t; } B[100010];
int cntA, cntB, w;
bool cmp(R Data1 i, R Data1 j)
{
    if(u[i.l] != u[j.l]) return i.l < j.l;
    if(u[i.r] != u[j.r]) return u[i.l] & 1 ? i.r < j.r : i.r > j.r;
    return (u[i.l] ^ u[i.r]) & 1 ? i.i < j.i : i.i > j.i;
}
void Add(R int i)
{
    if(i == 0) return ;
    if(s[i]) 
    {
        S[s[i]]--;
        if(!S[s[i]]) d[u[s[i]]]--;
    }
    ++s[i];
    if(!S[s[i]]) d[u[s[i]]]++;
    S[s[i]]++;
}
void Del(R int i)
{
    if(i == 0) return ;
    S[s[i]]--;
    if(!S[s[i]]) d[u[s[i]]]--;
    --s[i];
    if(s[i])
    {
        if(!S[s[i]]) d[u[s[i]]]++;
        S[s[i]]++;
    }
}
int Query()
{
    for(R int i = 0;; i++) if(d[i] != w)
        for(R int j = i * w;; j++) if(!S[j])
            return j;
}
int main()
{
    scanf("%d %d", &n, &q);
    for(R int i = 1; i <= n; i++) 
    {
        R int t = F();
        if(M[t] == 0) a[i] = M[t] = ++cnt;
        else a[i] = M[t];
    }
    for(R int i = 1; i <= n; i++) t[i] = a[i];
    for(R int i = 1; i <= q; i++)
    {
        R int opt = F();
        if(opt == 1) A[++cntA] = (Data1){cntB, F(), F(), cntA};
        else B[++cntB] = (Data2){cntB, F(), F()};
    }
    for(R int i = 1; i <= cntB; i++)
    {
        if(M[B[i].x] == 0) B[i].x = M[B[i].x] = ++cnt;
        else B[i].x = M[B[i].x];
        B[i].t = t[B[i].p]; t[B[i].p] = B[i].x;
    }
    w = pow(n, 2.0 / 3.0);
    for(R int i = 1; i <= q; i++) u[i] = i / w;
    R int x = 0, y = 0, z = 0; 
    std::sort(A + 1, A + 1 + cntA, cmp); S[0] = d[0] = 1;
    for(R int i = 1; i <= cntA; i++) //边界！！！ 
    {
        while(y <= A[i].r) Add(a[y++]);
        while(x >= A[i].l) Add(a[x--]);
        while(z <= A[i].i) 
        {
            if(x < B[z].p && B[z].p < y) Del(B[z].t), Add(B[z].x);
            a[B[z].p] = B[z].x; z++;
        }
        while(z > A[i].i + 1) 
        {
            --z;
            if(x < B[z].p && B[z].p < y) Del(B[z].x), Add(B[z].t);
            a[B[z].p] = B[z].t;
        }
        while(x < A[i].l - 1) Del(a[++x]);
        while(y > A[i].r + 1) Del(a[--y]);
        Ans[A[i].t] = Query();
    }
    for(R int i = 1; i <= cntA; i++) printf("%d\n", Ans[i]);
    return 0;
}