bzoj1483 [HNOI2009]梦幻布丁
原题地址:http://www.lydsy.com/JudgeOnline/problem.php?id=1483
题意:
N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.
数据范围
N,M<=1e6
题解:
启发式合并->时间复杂度O(mlogn)
首先用链表存储不同的颜色,这里使用了类似链式前向星的存储方式。记录每个颜色的链的起始位置,以及元素的个数。(便于之后合并)
为了实现启发式合并,我们必然把小的合并到大的中去。但是显然如果只是把颜色 i,颜色 j(size[i] < size[j])都变成颜色j,而原操作是颜色 j 变成颜色 i,显然是不行的。因此,我们交换i,j所代表的颜色,从此i代表j,j代表 i,用一个数组维护。
这样思路就很清晰了,我们每次通过链表枚举数目较小的那个颜色的每个元素,统计看其旁边(左+右)有没有与要改成的颜色相接,若有,则ans- -。然后再一次通过链表枚举数目较小的那个颜色的每个元素,在原数组中改变他们的颜色。
关于合并,因为之前记录了每个颜色的链的起始位置,于是这样处理:
head[y] (new)-> head[x] -> … … … -> fir[x] ->head[y] (previous)
记得更新大颜色的size,并把被合并的颜色size和起始元素都清空。
关于启发式合并的复杂度分析:
显然合并的集合两个大小差距越悬殊,只枚举小的的元素的优势越明显。我们举最坏情况为例:
若每次合并的都是大小相同的两个集合,每一次的大小成两倍扩展,一个集合最多被扩展logn次,n*logn -> O(nlogn)
代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1000006;
int n,m,ans=0,head[N],nxt[N],a[N],col[N],size[N],fir[N];
void merge(int x,int y)
{
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[fir[x]]=head[y];
head[y]=head[x];
size[y]+=size[x];
head[x]=size[x]=fir[x]=0;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]!=a[i-1]) ans++;
col[a[i]]=a[i];
if(!head[a[i]]) fir[a[i]]=i;
size[a[i]]++;
nxt[i]=head[a[i]];
head[a[i]]=i;
}
while(m--)
{
int opt,x,y;
scanf("%d",&opt);
if(opt==1)
{
scanf("%d%d",&x,&y);
if(x==y) continue;
if(size[col[x]]>size[col[y]]) swap(col[x],col[y]);
x=col[x];y=col[y];
if(size[x]==0) continue;
merge(x,y);
}
else printf("%d\n",ans);
}
return 0;
}