【题解】数颜色 STL vector数组

小 C 的兔子不是雪白的，而是五彩缤纷的。

题目

题目描述

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

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

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

输入输出格式

输入格式：

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

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

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

“1 lj rj cj” ：询问在区间 [lj,rj]里有多少只颜色为 cj 的兔子；
“2 xj”： xj 和 xj+1 两只兔子交换了位置。

输出格式：

输出到标准输出中。

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

输入输出样例

输入样例

6 5
1 2 3 2 3 3
1 1 3 2
1 4 6 3
2 3
1 1 3 2
1 4 6 3

输出样例

1
2
2
3

题解

分析

维护每个数字出现的位置就可以了，用vector存，1操作就输出尾位置在数字出现位置的位置减去首位置的就可以了。至于2操作，先判断2个位置上的数字是否相等，相等就跳过，不然就直接swap就可以了

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=300010;
const int inf=0xFFFFFFF;
vector<int>r[maxn];

int a[maxn],n,m;
inline int read() {
    int  x=0,w=1;
    char ch=getchar();
    while(ch!='-'&&(ch<'-'||ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-48,ch=getchar();
    return x*w;
}

void solve()
{
    int cases,l,R,x;
    while(m--)
    {
        cases=read();
        if(cases==1)
        {
            l=read();R=read();x=read();
            printf("%d\n",distance(lower_bound(r[x].begin(),r[x].end(),l),upper_bound(r[x].begin(),r[x].end(),R)));
        }
        else
        {
            x=read();
            if(a[x]==a[x+1])continue;
            int u=distance(r[a[x]].begin(),lower_bound(r[a[x]].begin(),r[a[x]].end(),x));

            int v=distance(r[a[x+1]].begin(),lower_bound(r[a[x+1]].begin(),r[a[x+1]].end(),x+1));
            r[a[x]][u]=x+1;
            r[a[x+1]][v]=x;
            swap(a[x],a[x+1]);
        }
    }
}
int main()
{
    freopen("color.in","r",stdin);
    freopen("color.out","w",stdout);
    int i;
    n=read();m=read();
    for(register int i=1;i<=n;++i)
    {
        a[i]=read();
        r[a[i]].push_back(i);
    }
    for(register int i=1;i<=300000;++i) r[i].push_back(inf);
    solve();
    return 0;
}

​