bzoj1483: [HNOI2009]梦幻布丁

传送门
话说好像可以用平衡树+启发式合并+set来水
当然还是队列+启发式合并好呀。
1:将两个队列合并,有若干队列,总长度为n,直接合并,最坏O(N),
会被用心良苦的出题人卡到 O(NM)
2.考虑启发式合并。每次我们把短的合并到长的上面去,O(短的长度)
每次合并后,队列长度一定大于等于原来短的长度的两倍。
这样相当于每次合并都会让短的长度扩大一倍以上,
最多扩大logN次,所以总复杂度O(NlogN),每次O(logN)。
我们先求出原序列的答案
每一种颜色搞一条链把该色结点串起来,记录下链条尾结点
把一种颜色的染成另一种,很简单把它合并过去,然后处理下对于答案的影响
但是只是这么弄会出现各种bug:
比如把1染成2,但是s[1]>s[2],这时我们应该将2合并到1的链后面,但是会遇到一个麻烦的问题,就是这个链头是接1下的,也就是说以后找颜色2,发现没有颜色2只有颜色1
于是我们应该开一个数组f,表示我们寻找一种颜色时,实际应该找哪个颜色下的链,遇到上面那种情况要交换f[1]和f[2]
然后九水过了。

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream> 
#define N 100005
#define M 1000005
int n,m,ans,x,a,b;
int c[N],ne[N],ft[M],he[M],s[M],st[M];
using namespace std;
inline void read(int &x){
    char ch=getchar();
    x=0;
    int f=1;
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    x*=f;
}
void solve(int a,int b){
    for (int i=he[a];i;i=ne[i]){
        if (c[i+1]==b) ans--;
        if (c[i-1]==b) ans--;
    }
    for (int i=he[a];i;i=ne[i]) c[i]=b;
    ne[st[a]]=he[b];
    he[b]=he[a];
    s[b]+=s[a];
    he[a]=st[a]=s[a]=0;
}
int main(){
    read(n);
    read(m);
    for (int i=1;i<=n;i++){
        read(c[i]);
        ft[c[i]]=c[i];
        if (c[i]!=c[i-1]) ans++;
        if (!he[c[i]]) st[c[i]]=i;
        s[c[i]]++;
        ne[i]=he[c[i]];
        he[c[i]]=i;
    }
    for (int i=1;i<=m;i++){
        read(x);
        if (x==2) printf("%d\n",ans);
        else{
            read(a);
            read(b);
            if (a==b) continue;
            if (s[ft[a]]>s[ft[b]]) swap(ft[a],ft[b]);
            a=ft[a];
            b=ft[b];
            if (!s[a]) continue;
            s[b]+=s[a];
            s[a]=0;
            solve(a,b);
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值