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

最新推荐文章于 2019-10-17 19:50:49 发布

原创最新推荐文章于 2019-10-17 19:50:49 发布 · 286 阅读

0 ·

CC 4.0 BY-SA版权

bzoj 同时被 3 个专栏收录

452 篇文章

订阅专栏

启发式合并

4 篇文章

订阅专栏

链表

2 篇文章

订阅专栏

针对一组布丁的颜色变更操作，通过链表实现高效更新并计算不同颜色段的数量。该问题涉及数据结构的设计与优化，使用启发式合并策略避免超时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.

范围不给耍流氓。
一开始没认真想直接无脑线段树后来发现好像修改没办法维护，然后认真想了想好像我一定要记录相邻的颜色，那只能链表了啊 = =，那好吧就链表了。
注意启发式合并不然会T。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e6+5;

int n,m,ans;
int a[N],tou[N],wei[N],head[N],size[N],next[N];
inline void merge(int x,int y)
{
    for(int i=head[x];i;i=next[i])
    {
        if (a[i+1]==y)ans--;
        if (a[i-1]==y)ans--;
    }
    for(int i=head[x];i;i=next[i])a[i]=y;
    next[tou[x]]=head[y];
    head[y]=head[x];
    size[y]+=size[x];
    head[x]=tou[x]=size[x]=0;
}
int main()
{
    scanf("%d%d",&n,&m);
    fo(i,1,n)
    {
        scanf("%d",&a[i]);
        wei[a[i]]=a[i];
        if (a[i]!=a[i-1])ans++;
        if (!head[a[i]])tou[a[i]]=i;
        size[a[i]]++;
        next[i]=head[a[i]];
        head[a[i]]=i;
    }
    fo(i,1,m)
    {
        int op;
        scanf("%d",&op);
        if (op==2)printf("%d\n",ans);
        else
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if (x==y)continue;
            if (size[wei[x]]>size[wei[y]])
                swap(wei[x],wei[y]);
            x=wei[x],y=wei[y];
            if (size[x]==0)continue;
            size[y]+=size[x];size[x]=0;
            merge(x,y);
        }
    }
}