重要的是事情说三遍,我是傻逼我是傻逼我是傻逼
傻逼。。。。
说说这个题吧,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;
}