树状数组+启发式合并

重要的是事情说三遍，我是傻逼我是傻逼我是傻逼

 

 

 

 

 

 

 

 

 

傻逼。。。。

说说这个题吧，hdu-5997

 

题意：

给出一个颜色的序列，每次有两种操作：

• 1 x y：将所有的颜色x变为颜色y；
• 2 l r：查询[l,r]之间有多少个连续的颜色段。

对于每次查询操作，输出查询的结果。

 

思路

 

用树状数组c维护当前点是不是新的颜色的开始，然后可以用前缀和算出区间内颜色的段数。对于修改就需要进行启发式合并，储存每种颜色的位置，对于合并，将少的颜色合并到多的颜色上。比如某种操作将颜色5改成颜色10，而5有8个，10有2个，所以你可以将颜色10换成颜色5，然后再将颜色5和10互换，p[i]就继续颜色i

代码：

 

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <queue>
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
int p[1000000+10];
int c[1000000+10];
bool ok[1000000+10];
int  a[1000000+10];
vector<int>g[1000000+10];
void add(int x,int d)
{
    while(x<1000000+10)
    {
        c[x]+=d;
        x+=x&-x;
    }
}
int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=c[x];
        x-=x&-x;
    }
    return ans;
}
void init()
{
    for(int i=0;i<1000000+10;i++)
    {
        p[i]=i;
        g[i].clear();
    }
    memset(ok,false,sizeof(ok));
    memset(c,0,sizeof(c));
}
void procc(int x)
{
    if(a[x]==a[x-1]&&ok[x])
    {
        ok[x]=false;
        add(x,-1);
    }
}
void proc(int &x,int &y)
{
    //for(int i=0;i<g[x].size();i++)
    for(int &v:g[x])
    {
        //int v=g[x][i];
        a[v]=y;
        g[y].push_back(v);
        procc(v);
        procc(v+1);
    }
    g[x].clear();
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        int n,q;
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            g[a[i]].push_back(i);
            if(a[i]!=a[i-1])
            {
                ok[i]=true;
                add(i,1);
            }
        }
        for(int i=1;i<=q;i++)
        {
            int c,x,y;
            scanf("%d%d%d",&c,&x,&y);
            if(c==1&&p[x]!=p[y])
            {
                if(g[p[x]].size()<g[p[y]].size())
                {
                    proc(p[x],p[y]);
                }
                else
                {
                    proc(p[y],p[x]);
                    swap(p[x],p[y]);
                }
            }
            else if(c==2)
            {
                int ans=sum(y)-sum(x-1);
                if(!ok[x])
                    ++ans;
                printf("%d\n",ans);
            }
        }

    }

    return 0;
}