bzoj1483 [HNOI2009]梦幻布丁

题目

我们先考虑暴力的做，显然会T，如何优化呢？

把相同的合成一段，用链表连起来，修改时，按照长度合并就好了。

#include<bits/stdc++.h>
#define MAXN 2000000
using namespace std;
int m,n,id,ans,a,b;
int A[MAXN+1],first[MAXN+1],nex[MAXN+1];
int num[MAXN+1],len[MAXN+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    char c=nc();
    int x=0,b=1;
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(int x)
{
    int L=0;
    char s[15];
    if(!x)putchar('0');
    else while(x)s[++L]=x%10+'0',x/=10;
    for(int i=L;i>=1;i--)putchar(s[i]);
    putchar('\n');
}
int main()
{
    //freopen("in.txt","r",stdin);
    n=read(),m=read();
    memset(first,-1,sizeof(first));
    ans=0;
    for(register int i=1;i<=n;i++)
    {
        A[i]=read();
        nex[i]=first[A[i]];
        first[A[i]]=i;
        num[A[i]]=A[i];
        if(A[i]!=A[i-1])ans++;
        len[A[i]]++;
    }
    for(int i=1;i<=m;i++)
    {
        id=read();
        if(id==2)write(ans);
        else
        {
            a=read(),b=read();
            if(a==b)continue;
            if(len[num[a]]>len[num[b]])swap(num[a],num[b]);
            a=num[a],b=num[b];
            if(len[a]==0)continue;
            for(int j=first[a];j!=-1;j=nex[j])
            {
                if(A[j-1]==b)ans--;
                if(A[j+1]==b)ans--;
            }
            int pos;
            for(int j=first[a];j!=-1;j=nex[j])
            {
                A[j]=b;
                pos=j;
            }
            nex[pos]=first[b];
            first[b]=first[a];
            len[b]+=len[a];
            len[a]=0;
            first[a]=-1;
        }
    }
    return 0;
}