Problem

题目传送门

n个布丁摆成一行,进行m次操作.每次将某种颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段连续的颜色.

\(n , m< 10^5, a_i, x, y <10^6\)

solution

一种颜色的布丁变成另一种颜色就相当于合并两种颜色的布丁。

并且两种布丁合并了之后显然不可能分开。

所以考虑启发式合并。

对于每种颜色，开一个vector记录当前颜色在哪些位置出现。

合并时小向大合并，计算答案即可。

然后愉快的咕了……

原因是：有时候我们需要将颜色x变成y，而此时siz[x]>siz[y]……直接合并的话显然是有问题的。

所以，我们需要用一个数组记录下每个颜色的真实颜色。

如果siz[x]>siz[y]但需要将颜色x变成y时，我们可以先将y变成x，计算答案之后，再将颜色x的真实颜色改为y即可。

这样复杂度和答案的正确性都有保证。

~(撒花)^_^~

Code

#include <bits/stdc++.h> using namespace std; #define DEBUG(...) fprintf(stderr, __VA_ARGS__) #define mp make_pair #define fst first #define snd second template<typename T> inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; } template<typename T> inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; } inline int read(){ int res = 0, fl = 1; char r = getchar(); for (; !isdigit(r); r = getchar()) if(r == '-') fl = -1; for (; isdigit(r); r = getchar()) res = (res << 3) + (res << 1) + r - 48; return res * fl; } typedef long long LL; typedef pair<int, int> pii; const int Maxn = 1e6 + 10; vector<int> plc[Maxn]; int b[Maxn], col[Maxn], a[Maxn], n, m, ans; struct ASK{ int opt, x, y; }ask[Maxn]; void merge(int x,int y){ for (int i = plc[x].size() - 1; i >= 0; --i) ans -= (a[plc[x][i] - 1] == y) + (a[plc[x][i] + 1] == y); for (int i = plc[x].size() - 1; i >= 0; --i) a[plc[x][i]] = y, plc[y].push_back(plc[x][i]); plc[x].clear(); } int main() { #ifndef ONLINE_JUDGE freopen("a.in", "r", stdin); freopen("a.out", "w", stdout); #endif int n = read(), m = read(); for (int i = 1; i <= n; ++i) a[i] = read(), col[a[i]] = a[i]; for (int i = 1; i <= n; ++i) { plc[a[i]].push_back(i); if(a[i] != a[i - 1]) ans++; } while(m--){ int opt = read(); if(opt == 2) printf("%d\n",ans); else { int x = read(), y = read(); if(x == y) continue; if(plc[col[x]].size() > plc[col[y]].size()) swap(col[x], col[y]); if(plc[col[x]].size()) merge(col[x], col[y]); } } return 0; }

转载于:https://www.cnblogs.com/LZYcaiji/p/10397846.html