luogu P3939 数颜色 |vector

题目描述

小 C 的兔子不是雪白的，而是五彩缤纷的。每只兔子都有一种颜色，不同的兔子可能有 相同的颜色。小 C 把她标号从 1 到 n 的 n 只兔子排成长长的一排，来给他们喂胡萝卜吃。 排列完成后，第 i 只兔子的颜色是 ai

俗话说得好，“萝卜青菜，各有所爱”。小 C 发现，不同颜色的兔子可能有对胡萝卜的 不同偏好。比如，银色的兔子最喜欢吃金色的胡萝卜，金色的兔子更喜欢吃胡萝卜叶子，而 绿色的兔子却喜欢吃酸一点的胡萝卜……为了满足兔子们的要求，小 C 十分苦恼。所以，为 了使得胡萝卜喂得更加准确，小 C 想知道在区间 [l_j,r_j]里有多少只颜色为 cj 的兔子。

不过，因为小 C 的兔子们都十分地活跃，它们不是很愿意待在一个固定的位置；与此同 时，小 C 也在根据她知道的信息来给兔子们调整位置。所以，有时编号为 x_j+1 的两 只兔子会交换位置。 小 C 被这一系列麻烦事给难住了。你能帮帮她吗？

输入格式

从标准输入中读入数据。 输入第 1 行两个正整数 n,m。

输入第 2 行 n 个正整数，第 ii 个数表示第 ii 只兔子的颜色 a_i。

输入接下来 m 行，每行为以下两种中的一种：

输出格式

输出到标准输出中。

对于每个 1 操作，输出一行一个正整数，表示你对于这个询问的答案。

---

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
struct node{
    int col,w;
}a[3*110000];
vector <int> c[3*110000];
int find(int id,int x){
    int l=1,r=c[id].size()-1;
    int ans=0;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(c[id][mid]>=x) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main(){
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        c[x].push_back(i);
        a[i].col=x,a[i].w=c[x].size()-1;
    }
    for(int i=1;i<=q;i++){
        int opt;
        scanf("%d",&opt);
        if(opt==1){
            int l,r,cr;
            scanf("%d%d%d",&l,&r,&cr);
            if(c[cr].size()==0)
            {
                printf("0\n");
                continue;
            }
            int pos1=lower_bound(c[cr].begin(),c[cr].end(),l)-c[cr].begin();
            int pos2=upper_bound(c[cr].begin(),c[cr].end(),r)-c[cr].begin();
            printf("%d\n",pos2-pos1);
        }
        if(opt==2){
            int x;
            scanf("%d",&x);
            if(a[x].col==a[x+1].col) continue;
            int w1=a[x].w,w2=a[x+1].w;
            int c1=a[x].col,c2=a[x+1].col;
            a[x].col=c2,a[x+1].col=c1;
            a[x].w=w2,a[x+1].w=w1;
            c[c1][w1]=x+1;
            c[c2][w2]=x;
        }
    }
    return 0;
}