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
1 2 2 1
2
1 2 1
2
Sample Output
3
1
1
HINT
Source
【分析】
转来的启发式合并的复杂度均摊分析:orzz
每次我们把短的合并到长的上面去,O(短的长度)
咋看之下没有多大区别,
下面让我们看看均摊的情况:
1:每次O(N)
2:每次合并后,队列长度一定大于等于原来短的长度的两倍。
这样相当于每次合并都会让短的长度扩大一倍以上,
最多扩大logN次,所以总复杂度O(NlogN),每次O(logN)。
就是裸题了,搞一个链表把每种颜色段的开头位置记录下来,然后每次修改暴力修改+启发式合并就行。
还有就是记得把因为大小而导致错误的颜色用一个数组映射。
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
1 /* 2 纳兰性德 3 人生若只如初见,何事秋风悲画扇。 4 等闲变却故人心,却道故人心易变。 5 骊山语罢清宵半,泪雨霖铃终不怨。 6 何如薄幸锦衣郎,比翼连枝当日愿。 7 */ 8 #include <iostream> 9 #include <cstdio> 10 #include <algorithm> 11 #include <cstring> 12 #include <vector> 13 #include <utility> 14 #include <iomanip> 15 #include <string> 16 #include <cmath> 17 #include <queue> 18 #include <assert.h> 19 #include <map> 20 #include <ctime> 21 #include <cstdlib> 22 #include <stack> 23 #define LOCAL 24 const int MAXN = 100000 + 10; 25 const int INF = 100000000; 26 const int SIZE = 450; 27 const int MAXM = 1000000 + 10; 28 const int maxnode = 0x7fffffff + 10; 29 using namespace std; 30 struct Node{ 31 int num; 32 Node *next; 33 }*head[MAXM];//head为表头 34 int cnt[MAXM], rem[MAXM], data[MAXM]; 35 int tot, n, m; 36 37 //在链表中加入颜色为x的节点 38 void add(int x, int d){//d代表位置 39 if (cnt[x] == 1){ 40 head[x] = new Node; 41 head[x]->num = d; 42 head[x]->next = NULL; 43 }else{ 44 //在表头插入 45 Node *p = new Node; 46 p->num = d; 47 p->next = head[x]; 48 head[x] = p; 49 } 50 } 51 void init(){ 52 tot = 0;//记录颜色的总数 53 memset(cnt, 0, sizeof(cnt));//记录颜色的数量 54 scanf("%d%d", &n, &m); 55 //for (int i = 1; i <= n; i++) rem[i] = i; 56 data[0] = -INF; 57 for (int i = 1; i <= n; i++){ 58 scanf("%d", &data[i]);//输入颜色 59 if (data[i] != data[i - 1]) tot++; 60 cnt[data[i]]++; 61 add(data[i], i); 62 rem[data[i]] = data[i];//防错数组初始化 63 } 64 65 } 66 //将a颜色变成b颜色 67 void change(int a, int b){ 68 //不要搞错了是比较正确颜色的个数 69 if (cnt[rem[a]] > cnt[rem[b]]) swap(rem[a], rem[b]); 70 a = rem[a];//总是让颜色数量少的变成多的 71 b = rem[b]; 72 if (cnt[a] == 0) return; 73 cnt[b] += cnt[a]; 74 cnt[a] = 0; 75 Node *cur; 76 for (cur = head[a]; cur != NULL; cur = cur->next){ 77 if (data[cur->num + 1] == b) tot--; 78 if (data[cur->num - 1] == b) tot--; 79 } 80 for (cur = head[a]; cur->next != NULL; cur = cur->next) data[cur->num] = b; 81 data[cur->num] = b; 82 //最后将a插在b后面 83 cur->next = head[b]; 84 head[b] = head[a]; 85 } 86 void work(){ 87 for (int i = 1; i <= m; i++){ 88 int t, a, b; 89 scanf("%d", &t); 90 if (t == 2) printf("%d\n", tot); 91 else{ 92 scanf("%d%d", &a, &b); 93 if (a == b) continue;//两种颜色相同 94 change(a, b); 95 } 96 } 97 //for (int i = 1; i <= n; i++) printf("%d", data[i]); 98 } 99 100 int main(){ 101 102 init(); 103 work(); 104 return 0; 105 }