「Codeforces 940F」Machine Learning

初始时有一个长度为 n 数字串。有以下 q 个操作:
1.询问。定义 ci 为区间 [l,r] i 出现的次数,求出 Mex{c0,c1,...} ,其中 Mex 表示集合中未出现的最小的正整数。
2.修改。将第 p 个数改成 x
(1n,q100000)

这题一眼看上去就是一道数据结构题(没想到居然这么暴力,不过我居然还是没在赛场上码完)。
秒解:带修改的莫队算法 O(n53)

#include 
#include 
#include 
#include 
#include 
#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;
}

你可能感兴趣的:(数据结构)