Cod sursa(job #3277055)

Utilizator puica2018Puica Andrei puica2018 Data 15 februarie 2025 11:49:47
Problema Marbles Scor 20
Compilator cpp-64 Status done
Runda vs11_12_vine_oji_2025 Marime 2.12 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;
unordered_map <int,int> v;

set <int> h;

int cnt[300005];

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[a[i]]=b[i];

    for(int j=1; j<=64; j++)
    {
        v.clear();
        h.clear();
        for(int i=1; i<=n; i++)
        {
            v[a[i]]=b[i];
            if(b[i]==j)
                h.insert(a[i]);
        }

        ind=0;
        for(auto it:s)
        {
            ind++;
            cnt[ind]=cnt[ind-1]+(v[it]==j);
        }

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

                if(color==j)
                {
                    h.erase(l[i]);
                    h.insert(l[i]+r[i]);
                }

                v[l[i]]=0;
                v[l[i]+r[i]]=color;
            }
            else
            {
                auto it1=h.lower_bound(l[i]);
                auto it2=h.upper_bound(r[i]);

                if(it1!=h.end() && it2!=h.begin())
                {
                    it2--;

                    if((*it1)<=(*it2))
                    {
                        int p1=c[*it1],p2=c[*it2];

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

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

    return 0;
}