填了昨晚BC的坑
链表的启发式合并
合并两个链表的时间代价如果是O(l1长度)或O(l2长度)的时候选择较短的那个长度。若l1>l2,则新链表长度显然大于两倍l2。
对于任意一个元素,若合并的时候对他进行了操作,那么该元素所在链表的长度至少翻了一倍,也就是说对这个元素我们最多进行logN次操作,它就会出现在最长的链表之中。
利用这个特性,在这个题目中,把所有颜色相同的布丁用链表串起来,每次将颜色x的布丁改成颜色y的布丁,就相当于对答案减上两种颜色相邻的对数,我们只需要遍历一次链表就可以了。根据上面的结论,我们选择元素个数较少的链表遍历。最后再将两个链表首尾相接就可以了。
总时间复杂度为O(n*logn),n为元素个数,复杂度与颜色数量无关。**
#include <iostream>
#include <cstdio>
#define N 100050
#define M 1000050
using namespace std;
struct Monster{ int u,next; }e[N];
int head[M],cnt,a[N],siz[M],c[M],n,m,ans;
int main() {
scanf("%d%d",&n,&m); ans = n;
a[0] = -1;
for (int i=1;i<=n;i++) {
scanf("%d",&a[i]);
if (a[i] == a[i-1]) ans--;
e[++cnt].u = i;
e[cnt].next = head[a[i]];
head[a[i]] = cnt;
siz[ a[i] ]++;
}
for (int i=1;i<=1e6;i++) c[i] = i;
while (m--) {
int cmd=0; scanf("%d",&cmd);
if (cmd == 1) {
int x,y,p; scanf("%d%d",&x,&y); if (x == y) continue;
if (siz[c[x]] > siz[c[y]]) swap(c[x],c[y]);
x=c[x]; y=c[y];
if (siz[x] == 0) continue; //重要
for (int i=head[x];i;i=e[i].next) {
int t=e[i].u;
if (a[t-1]==y) ans--;
if (a[t+1]==y) ans--;
}
for (int i=head[x];i;i=e[i].next) a[e[i].u]=y;
for (int i=head[x];i;i=e[i].next) if (!e[i].next) p=i;
e[p].next=head[y]; head[y]=head[x];
siz[y]+=siz[x]; siz[x]=0;
}
if (cmd == 2) printf("%d\n",ans);
}
return 0;
}