Description
N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.
Input
第一行给出N,M表示布丁的个数和好友的操作次数. 第二行N个数A1,A2...An表示第i个布丁的颜色从第三行起有M行,对于每个操作,若第一个数字是1表示要对颜色进行改变,其后的两个整数X,Y表示将所有颜色为X的变为Y,X可能等于Y. 若第一个数字为2表示要进行询问当前有多少段颜色,这时你应该输出一个整数. 0
Output
针对第二类操作即询问,依次输出当前有多少段颜色.
Sample Input
4 3
1 2 2 1
2
1 2 1
2
Sample Output
3
1
将所有一样的颜色用链表连接,先计算出初始答案,当每次修改时相当于将两个不同的颜色合并。但是一定要将个数少的颜色合并到个数多的颜色的后面,这样可以保证每次新链的长度一定>=较小的子链的两倍。相当于每次长度*2这样最多也就修改logn次,因此复杂度是nlogn。
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define swap(x,y){int temp=x;x=y;y=temp;}
const int maxm = 1000005;
int last[maxm], ne[maxm], s[maxm], id[maxm], a[maxm], p[maxm], fa[maxm];
int main()
{
int n, i, j, k, sum, m, ord, ans = 0;
scanf("%d%d", &n, &m);
for (i = 1;i <= n;i++)
{
scanf("%d", &a[i]);
if (a[i] != a[i - 1]) ans++;
if (!fa[a[i]]) fa[a[i]] = i;
id[a[i]] = i;s[a[i]]++;
ne[i] = last[a[i]];
last[a[i]] = i;p[a[i]] = a[i];
}
while (m--)
{
scanf("%d", &ord);
if (ord == 1)
{
int x, y;
scanf("%d%d", &x, &y);
if (x == y) continue;
if (s[p[x]] > s[p[y]]) swap(p[x], p[y]);
x = p[x], y = p[y];
if (!s[x]) continue;
for (i = last[x];i >= 1;i = ne[i])
{
if (a[i - 1] == y) ans--;
if (a[i + 1] == y) ans--;
}
for (i = last[x];i >= 1;i = ne[i])
a[i] = y;
s[y] += s[x];s[x] = 0;
ne[fa[x]] = last[y];last[y] = last[x];
last[x] = fa[x] = 0;
}
else
printf("%d\n", ans);
}
return 0;
}