Cod sursa(job #3277018)

Utilizator puica2018Puica Andrei puica2018 Data 15 februarie 2025 11:17:58
Problema Marbles Scor 10
Compilator cpp-64 Status done
Runda vs11_12_vine_oji_2025 Marime 1.88 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("marbles.in");
ofstream fout("marbles.out");

int n,m;
int a[100005],b[100005];

int t[100005],l[100005],r[100005];

int ans[100005];

set <int> s;
unordered_map <int,int> c;

int v[300005];

int aib[300005];

void update(int p,int v)
{
    for(int i=p; i<=n+2*m; i+=(i&(-i)))
        aib[i]+=v;
}

int query(int p)
{
    int ans=0;
    for(int i=p; i>=1; i-=(i&(-i)))
        ans+=aib[i];
    return ans;
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; i++)
        fin>>a[i]>>b[i];
    for(int i=1; i<=m; i++)
        fin>>t[i]>>l[i]>>r[i];

    for(int i=1; i<=n; i++)
        s.insert(a[i]);
    for(int i=1; i<=m; i++)
    {
        if(t[i]==0)
        {
            s.insert(l[i]);
            s.insert(l[i]+r[i]);
        }
        else
        {
            s.insert(l[i]);
            s.insert(r[i]);
        }
    }

    int ind=0;
    for(auto it:s)
        c[it]=++ind;

    for(int i=1; i<=n; i++)
        v[c[a[i]]]=b[i];

    for(int j=1; j<=64; j++)
    {
        for(int i=1; i<=n+2*m; i++)
            aib[i]=0;

        for(int i=1; i<=n; i++)
            if(v[c[a[i]]]==j)
                update(c[a[i]],1);

        for(int i=1; i<=m; i++)
        {
            if(t[i]==0)
            {
                int p1=c[l[i]],p2=c[l[i]+r[i]];
                int color=v[p1];

                if(color==j)
                {
                    update(p1,-1);
                    update(p2,1);
                }

                v[p1]=0;
                v[p2]=color;
            }
            else
            {
                int p1=c[l[i]],p2=c[r[i]];

                ans[i]=max(ans[i],query(p2)-query(p1-1));
            }
        }
    }

    for(int i=1; i<=m; i++)
        if(t[i]==1)
            fout<<ans[i]<<"\n";

    return 0;
}