1483: [HNOI2009]梦幻布丁

题目链接

题目大意：维护数列，兹瓷两种操作：1.把所有x变成y
2.询问数列中的连续段数，相邻且相同的为一段

题解：先求一下初始的答案，记作ans，然后维护它

由于没有数据范围，考虑用平衡树启发式合并暴力

用若干个链表维护，一个链表维护一种颜色出现的各个位置

操作1：启发式合并链表，同时维护答案

为了方便在启发式合并时因为size而交换链表，记录p[i]表示颜色i所代表的实际颜色,初始p[i]=i，交换p[x],p[y]即可

操作2：输出ans……

这里的多个链表使用前向星实现……

我终于会链表了

我的收获：求初始答案然后维护似乎挺常见的

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>using namespace std;#define M 1000100int n,q,z,ans,t;int c[M],sz[M],head[M],p[M];struct edge{int to,nex;}e[M<<1];void add(int u,int v){e[t]=(edge){v,head[u]};head[u]=t++;}void merge(int x,int y){    if(x==y) return ;//！！！     if(sz[p[x]]>sz[p[y]]) swap(p[x],p[y]);    x=p[x];y=p[y];    if(!sz[x]) return ;//!!!    for(int i=head[x];i!=-1;i=e[i].nex){        if(c[e[i].to-1]==y) ans--;//维护相邻位置对答案的影响         if(c[e[i].to+1]==y) ans--;     }    for(int i=head[x];i!=-1;i=e[i].nex) c[e[i].to]=y;//修改     for(z=head[x];e[z].nex!=-1;z=e[z].nex);//找到链表x     e[z].nex=head[y];head[y]=head[x];head[x]=-1;//因为链表不需要考虑单调性，直接把链表y连到链表x后面     sz[y]+=sz[x];sz[x]=0;}void work(){    int opt,x,y;    while(q--)    {        scanf("%d",&opt);        if(opt==1) scanf("%d%d",&x,&y),merge(x,y);        else printf("%d\n",ans);    }}void init(){    t=0,memset(head,-1,sizeof(head));    scanf("%d%d",&n,&q);    for(int i=1;i<=n;i++)    {        scanf("%d",&c[i]);        if(c[i]!=c[i-1]) ans++;        sz[c[i]]++;p[c[i]]=c[i];        add(c[i],i);//用链表记录每个c[i]的位置     }}int main(){    init();    work();    return 0;}