首页 > 代码库 > BZOJ 1483 HNOI2009 梦幻布丁 链表+启发式合并
BZOJ 1483 HNOI2009 梦幻布丁 链表+启发式合并
题目大意:给定n个布丁,每个布丁有一个颜色,多次将某种颜色的所有布丁变为另一种颜色,多次询问颜色段数
数据范围:n<=10W 颜色数<=100W
链表的启发式合并0.0 一直没写明白 其实就是开个链表记录每种颜色的位置,合并时撸一遍短的链看看两边是不是长链的颜色就行
不过交换比较麻烦0.0 要开个数组记录每个数字代表的真实颜色 交换时把数组的这两个位置也交换下就可以了
注意用过的垃圾不要留在原位 size合并掉就清零 head合并走了就弄成NULL 不然会挂 强迫症的福音啊
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 1001001 using namespace std; struct abcd{ abcd *next; int pos; }*head[M]; int n,m,ans,a[M]; int f[M],size[M]; inline void Add(int x,int y) { abcd *temp=new abcd; temp->pos=y; temp->next=head[x]; head[x]=temp; } inline void Merge(int x,int y) { abcd *temp; if(x==y) return ; if(size[f[x]]>size[f[y]]) swap(f[x],f[y]); x=f[x];y=f[y]; if(!head[x]) return ; for(temp=head[x];temp;temp=temp->next) { if(a[temp->pos-1]==y) --ans; if(a[temp->pos+1]==y) --ans; } for(temp=head[x];temp;temp=temp->next) a[temp->pos]=y; for(temp=head[x];temp->next;temp=temp->next); temp->next=head[y];head[y]=head[x];head[x]=0x0; size[y]+=size[x];size[x]=0; } int main() { int i,p,x,y; cin>>n>>m; for(i=1;i<=n;i++) { scanf("%d",&a[i]); if(a[i]!=a[i-1]) ++ans; Add(a[i],i); f[a[i]]=a[i]; size[a[i]]++; } for(i=1;i<=m;i++) { scanf("%d",&p); if(p==1) { scanf("%d%d",&x,&y); Merge(x,y); } else printf("%d\n",ans); } }
BZOJ 1483 HNOI2009 梦幻布丁 链表+启发式合并
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。