[BZOJ1483] [HNOI2009]梦幻布丁链表启发式合并

最新推荐文章于 2019-03-19 21:59:11 发布

原创最新推荐文章于 2019-03-19 21:59:11 发布 · 405 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种链表启发式合并算法，用于优化不同颜色布丁的转换过程。通过选择较短链表进行操作，确保每个元素的操作次数不超过logN次，实现O(n*logn)的时间复杂度。

填了昨晚BC的坑
链表的启发式合并

合并两个链表的时间代价如果是O(l1长度)或O(l2长度)的时候选择较短的那个长度。若l1>l2，则新链表长度显然大于两倍l2。

对于任意一个元素，若合并的时候对他进行了操作，那么该元素所在链表的长度至少翻了一倍，也就是说对这个元素我们最多进行logN次操作，它就会出现在最长的链表之中。

利用这个特性，在这个题目中，把所有颜色相同的布丁用链表串起来，每次将颜色x的布丁改成颜色y的布丁，就相当于对答案减上两种颜色相邻的对数，我们只需要遍历一次链表就可以了。根据上面的结论，我们选择元素个数较少的链表遍历。最后再将两个链表首尾相接就可以了。

总时间复杂度为O(n*logn)，n为元素个数，复杂度与颜色数量无关。**

#include <iostream>
#include <cstdio>
#define N 100050
#define M 1000050
using namespace std;
struct Monster{ int u,next; }e[N];
int head[M],cnt,a[N],siz[M],c[M],n,m,ans;
int main() {
    scanf("%d%d",&n,&m); ans = n;
    a[0] = -1;
    for (int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        if (a[i] == a[i-1]) ans--;
        e[++cnt].u = i;
        e[cnt].next = head[a[i]];
        head[a[i]] = cnt;
        siz[ a[i] ]++;
    }
    for (int i=1;i<=1e6;i++) c[i] = i;
    while (m--) {
        int cmd=0; scanf("%d",&cmd);
        if (cmd == 1) {
            int x,y,p; scanf("%d%d",&x,&y); if (x == y) continue;
            if (siz[c[x]] > siz[c[y]]) swap(c[x],c[y]);
            x=c[x]; y=c[y];
            if (siz[x] == 0) continue; //重要 
            for (int i=head[x];i;i=e[i].next) {
                int t=e[i].u;
                if (a[t-1]==y) ans--;
                if (a[t+1]==y) ans--;
            }
            for (int i=head[x];i;i=e[i].next) a[e[i].u]=y;
            for (int i=head[x];i;i=e[i].next) if (!e[i].next) p=i;
            e[p].next=head[y]; head[y]=head[x];
            siz[y]+=siz[x]; siz[x]=0;
        }
        if (cmd == 2) printf("%d\n",ans);
    } 
    return 0;
}