【链表】梦幻布丁

题意
N个布丁摆成一行,进行M次操作。每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色。
例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.。
n,m<=1000000
【思路】
对于每次操作,我们把小的合并到大的里面。
举个例子:颜色为1,2,2时要把颜色2改为颜色1,暴力的方法就是直接将所有颜色2改为1,但是将1改为2耗时更短。因此我们以f[i]表示i真正的颜色,那么若被替换的颜色数量大于将其替换的颜色数量,为了节省时间将替换颜色改为被替换颜色以后,要交换f[替换颜色]与f[被替换颜色]。
然后对于每种颜色都使用链表储存,将颜色1改为颜色2就将颜色1的结尾接到颜色2的头。
【代码】

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 1100000
using namespace std;
int n,m,ans,val[MAXN],f[MAXN],head[MAXN],st[MAXN],cnt[MAXN],nex[MAXN];
void solve(int x,int y)
{
    cnt[y]+=cnt[x];
    cnt[x]=0;
    for(int i=head[x];i;i=nex[i])
    {
        if(val[i-1]==y)ans--;
        if(val[i+1]==y)ans--;
    }
    for(int i=head[x];i;i=nex[i])
        val[i]=y;

    nex[st[x]]=head[y];
    head[y]=head[x];
    head[x]=0;
    return ;
}
int main()
{
    scanf("%d%d",&n,&m);

    memset(head,0,sizeof(head));
    memset(st,0,sizeof(st));
    ans=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&val[i]);
        f[val[i]]=val[i];
        if(val[i]!=val[i-1])ans++;
        if(!head[val[i]])st[val[i]]=i;
        cnt[val[i]]++;
        nex[i]=head[val[i]];
        head[val[i]]=i;
    }

    while(m--)
    {
        int op,x,y;
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d",&x,&y);
            if(x==y)continue;
            if(cnt[f[x]]>cnt[f[y]])
                swap(f[x],f[y]);
            x=f[x],y=f[y];
            if(!cnt[x])continue;
            solve(x,y); 
        }
        else printf("%d\n",ans);
    }
    return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值