洛谷 P3939 数颜色

洛谷 P3939 数颜色

题意

有 $n$ 只兔子，每个兔子有各种颜色 $a_i$ ，然后给出 $m$ 次操作：

• 1 l r c 表示询问区间 $[l,r]$ 内有多少种不同的颜色；
• 2 x 表示交换 $a_x,a_{x+1}$ 。

解法

用vector 数组记录每个颜色的出现位置即可，这样询问操作可以使用upperbound()-lowerbound() 来查询，因为每次交换一定是相邻两个交换，所以只要二分找到两个颜色的位置，然后位置加一或者减一即可。

代码

#pragma region
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
typedef long long ll;
#define tr t[root]
#define lson t[root << 1]
#define rson t[root << 1 | 1]
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#pragma endregion
const int maxn = 3e5 + 5;
int n, q, a[maxn];
vector<int> v[maxn];
int main() {
    scanf("%d%d", &n, &q);
    rep(i, 1, n) scanf("%d", &a[i]), v[a[i]].push_back(i);
    while (q--) {
        int op, l, r, c, p;
        scanf("%d", &op);
        if (op == 1) {
            scanf("%d%d%d", &l, &r, &c);
            int p1 = lower_bound(v[c].begin(), v[c].end(), l) - v[c].begin();
            int p2 = upper_bound(v[c].begin(), v[c].end(), r) - v[c].begin() - 1;
            printf("%d\n", max(0, p2 - p1 + 1));
        } else {
            scanf("%d", &p);
            if (a[p] == a[p + 1]) continue;
            int p1 = lower_bound(v[a[p]].begin(), v[a[p]].end(), p) - v[a[p]].begin();
            int p2 = lower_bound(v[a[p + 1]].begin(), v[a[p + 1]].end(), p + 1) - v[a[p + 1]].begin();
            v[a[p]][p1]++, v[a[p + 1]][p2]--;
            swap(a[p], a[p + 1]);
        }
    }
}