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

题目:

我是超链接

题解:

我们可以给每一种颜色建立链表,因为链上的颜色总是一起变的,那么修改的时候只需要将链表合并就行了,把小的合并到大的里面去,那么均摊的复杂度就是O(nlogn)
动态的维护答案,当合并的时候看看左看看右,如果是要变成一样的颜色就答案- -
这里还有一个小细节,我们把小的合并进入大的,但并不一定是小的的颜色变成大的颜色。怎么办呢?我们设置一个id[i]表示i这个颜色目前对应的颜色,然后如果我们要把大的的颜色改成小的,就把两个id换一下就好了
还要注意因为链表是前向星要特别注意head和tail。每次变颜色的时候要先看看需要改变的颜色是不是没有,如果没有就不需要改

代码:

#include <cstdio>
#include <iostream>
using namespace std;
const int N=1000005;
int a[N],tail[N],f[N],cnt[N],head[N],nxt[N];
int main()
{
    int n,m,ans=0,maxx=0;scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) 
    {
        scanf("%d",&a[i]);maxx=max(maxx,a[i]);
        if (a[i]!=a[i-1]) ans++;
        if (!cnt[a[i]]) tail[a[i]]=i;
        nxt[i]=head[a[i]]; head[a[i]]=i;
        cnt[a[i]]++;
    }
    for (int i=1;i<=maxx;i++) f[i]=i;
    while (m--)
    {
        int id,x,y;
        scanf("%d",&id);
        if (id==1)
        {
            scanf("%d%d",&x,&y);
            if (f[x]==f[y]) continue;
            if (cnt[f[x]]>cnt[f[y]]) swap(f[x],f[y]); 
            x=f[x]; y=f[y];
            if (cnt[x]==0) continue;///

            for (int i=head[x];i;i=nxt[i])
            {
                if (a[i-1]==y) ans--;
                if (a[i+1]==y) ans--;
            }
            for (int i=head[x];i;i=nxt[i]) a[i]=y;
            nxt[tail[x]]=head[y]; head[y]=head[x]; head[x]=tail[x]=0;
            cnt[y]+=cnt[x]; cnt[x]=0;
        }
        else printf("%d\n",ans);
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值