【BZOJ1483】【链表启发式合并】梦幻布丁

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

HINT

Source

【分析】

转来的启发式合并的复杂度均摊分析：orzz

每次我们把短的合并到长的上面去，O(短的长度) 

咋看之下没有多大区别，

下面让我们看看均摊的情况：

1：每次O(N)

2：每次合并后，队列长度一定大于等于原来短的长度的两倍。

这样相当于每次合并都会让短的长度扩大一倍以上，

最多扩大logN次，所以总复杂度O(NlogN)，每次O(logN)。

 

就是裸题了，搞一个链表把每种颜色段的开头位置记录下来，然后每次修改暴力修改+启发式合并就行。

还有就是记得把因为大小而导致错误的颜色用一个数组映射。

 /*
 纳兰性德
 人生若只如初见，何事秋风悲画扇。
 等闲变却故人心，却道故人心易变。
 骊山语罢清宵半，泪雨霖铃终不怨。
 何如薄幸锦衣郎，比翼连枝当日愿。
 */
 #include <iostream>
 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 #include <vector>
 #include <utility>
 #include <iomanip>
 #include <string>
 #include <cmath>
 #include <queue>
 #include <assert.h>
 #include <map>
 #include <ctime>
 #include <cstdlib>
 #include <stack>
 #define LOCAL
 const int MAXN =  + ;
 const int INF = ;
 const int SIZE = ;
 const int MAXM =  + ;
 const int maxnode =  0x7fffffff + ;
 using namespace std;
 struct Node{
        int num;
        Node *next;
 }*head[MAXM];//head为表头
 int cnt[MAXM], rem[MAXM], data[MAXM];
 int tot, n, m;

 //在链表中加入颜色为x的节点
 void add(int x, int d){//d代表位置
      if (cnt[x] == ){
         head[x] = new Node;
         head[x]->num = d;
         head[x]->next = NULL;
      }else{
         //在表头插入
         Node *p = new Node;
         p->num = d;
         p->next = head[x];
         head[x] = p;
      }
 }
 void init(){
      tot = ;//记录颜色的总数
      memset(cnt, , sizeof(cnt));//记录颜色的数量
      scanf("%d%d", &n, &m);
      //for (int i = 1; i <= n; i++) rem[i] = i;
      data[] = -INF;
      for (int i = ; i <= n; i++){
          scanf("%d", &data[i]);//输入颜色
          if (data[i] != data[i - ]) tot++;
          cnt[data[i]]++;
          add(data[i], i);
          rem[data[i]] = data[i];//防错数组初始化
      }

 }
 //将a颜色变成b颜色
 void change(int a, int b){
      //不要搞错了是比较正确颜色的个数
      if (cnt[rem[a]] > cnt[rem[b]]) swap(rem[a], rem[b]);
      a = rem[a];//总是让颜色数量少的变成多的
      b = rem[b];
      if (cnt[a] == ) return;
      cnt[b] += cnt[a];
      cnt[a] = ;
      Node *cur;
      for (cur = head[a]; cur != NULL; cur = cur->next){
          if (data[cur->num + ] == b) tot--;
          if (data[cur->num - ] == b) tot--;
      }
      for (cur = head[a]; cur->next != NULL; cur = cur->next) data[cur->num] = b;
      data[cur->num] = b;
      //最后将a插在b后面
      cur->next = head[b];
      head[b] = head[a];
 }
 void work(){
      for (int i = ; i <= m; i++){
          int t, a, b;
          scanf("%d", &t);
          if (t == ) printf("%d\n", tot);
          else{
               scanf("%d%d", &a, &b);
               if (a == b) continue;//两种颜色相同
               change(a, b);
          }
      }
      //for (int i = 1; i <= n; i++) printf("%d", data[i]);
 }

 int main(){

     init();
     work();
     return ;
 }