【洛谷P3939】数颜色

题目背景

大样例下发链接：http://pan.baidu.com/s/1c0LbQ2 密码：jigg

题目描述

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

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

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

输入输出格式

输入格式：
从标准输入中读入数据。 输入第 1 行两个正整数 nn,mm。

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

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

“1\ l_j\ r_j\ c_j1 l
j
​ r
j
​ c
j
​ ” ：询问在区间 [l_j,r_j][l
j
​ ,r
j
​ ] 里有多少只颜色为 c_jc
j
​ 的兔子；

“2\ x_j2 x
j
​ ”： x_jx
j
​ 和 x_j+1x
j
​ +1 两只兔子交换了位置。
输出格式：
输出到标准输出中。

对于每个 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： 复制
1
2
2
3
说明

【样例 1 说明】

前两个 1 操作和后两个 1 操作对应相同；在第三次的 2 操作后，3 号兔子和 4 号兔子

交换了位置，序列变为 1 2 2 3 3 3。

【数据范围与约定】

题解
题解里的解法一，按照颜色与位置进行双关键字排序。
操作1，二分位置。
操作2，如果两个位置颜色不同，则交换两点的坐标。

代码

#include<bits/stdc++.h>
typedef long long ll;
const int mod=1000000007;
const int N=500005;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct node{int x,p;}a[300005];
int b[300005],n,m,L,R,C,X;
bool cmp(node a,node b)
{
    return (a.x!=b.x)?a.x<b.x:a.p<b.p;
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=n;i++) a[i].x=read(),a[i].p=i;
    sort(a+1,a+n+1,cmp);
    for (int i=1;i<=n;i++) b[a[i].p]=i;
    while (m--)
    {
        int opt=read();
        if (opt==1)
        {
            L=read(),R=read(),C=read();
            int l=0,r=n,ans=0;
            while (l<=r)
            {
                int mid=(l+r)>>1;
                if (a[mid].x>C){r=mid-1;continue;}
                if (a[mid].x<C){ans=mid;l=mid+1;continue;}
                if (a[mid].p<L) l=mid+1,ans=mid;else r=mid-1;
            }
            int x=ans;
            l=0,r=n,ans=0;
            while (l<=r)
            {
                int mid=(l+r)>>1;
                if (a[mid].x>C){r=mid-1;continue;}
                if (a[mid].x<C){ans=mid;l=mid+1;continue;}
                if (a[mid].p<=R) l=mid+1,ans=mid;else r=mid-1;
            }
            int y=ans;
            printf("%d\n",y-x);
        }
        else
        {
            X=read();
            int x=b[X],y=b[X+1];
            if (a[b[X]].x==a[b[X+1]].x) continue;
            swap(a[x].p,a[y].p);
            swap(b[X],b[X+1]);
        }
    }
    return 0;
}