[P3939]数颜色

原题链接

把每个颜色的每个位置记下来
lower_bound是神器

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define LL long long
#define MOD 1000000007
using namespace std;

vector<int>p[300005];
int n,m,a[300005],opt,l,r,c,ll,rr,p1,p2,x,i;

int main()
{
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) 
    {
        scanf("%d",&a[i]);  
        p[a[i]].push_back(i);
    }

    for(i=1;i<=m;i++)
    {
        scanf("%d",&opt);

        if(opt==1)
        {
            scanf("%d%d%d",&l,&r,&c);

            ll=lower_bound(p[c].begin(),p[c].end(),l)-p[c].begin();
            rr=upper_bound(p[c].begin(),p[c].end(),r)-p[c].begin();
            printf("%d\n",rr-ll);
        }

        if(opt==2)
        {
            scanf("%d",&x);
            if(a[x]!=a[x+1])
            {
                p1=lower_bound(p[a[x]].begin(),p[a[x]].end(),x)-p[a[x]].begin();
                p[a[x]][p1]=x+1;
                p2=lower_bound(p[a[x+1]].begin(),p[a[x+1]].end(),x+1)-p[a[x+1]].begin();
                p[a[x+1]][p2]=x;
                swap(a[x],a[x+1]);  
            }
        }
    }

    return 0;
}