P3201 [HNOI2009]梦幻布丁
题目描述
N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.
输入输出格式
输入格式:
第一行给出N,M表示布丁的个数和好友的操作次数. 第二行N个数A1,A2…An表示第i个布丁的颜色从第三行起有M行,对于每个操作,若第一个数字是1表示要对颜色进行改变,其后的两个整数X,Y表示将所有颜色为X的变为Y,X可能等于Y. 若第一个数字为2表示要进行询问当前有多少段颜色,这时你应该输出一个整数. 0
输出格式:
针对第二类操作即询问,依次输出当前有多少段颜色.
输入输出样例
输入样例#1:
4 3
1 2 2 1
2
1 2 1
2
输出样例#1:
3
1
说明
1<=n,m<=100,000; 0<Ai,x,y<1,000,000
这道题一眼看上去很简单 暴力模拟就好了 没有任何难度
代码送上
#include <iostream>
using namespace std;
int n,m;
int a[250000];
int cha()
{
int sum=0;
for(int i=1;i<=n;i++)
{
if(a[i]!=a[i-1]) sum++;
}
return sum;
}
void bian(int x,int y)
{
for(int i=1;i<=n;i++)
{
if(a[i]==x) a[i]=y;
}
}
int main() {
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=m;i++)
{
int hao;
cin>>hao;
if(hao==2)
{
cout<<cha()<<endl;
}else{
int yi,er;
cin>>yi>>er;
bian(yi,er);
}
}
return 0;
}
但当你提交时,你会惊喜的发现 这只有一部分分
其他点超时了
那么怎么办呢??????
分析一下 我们发现 在这种暴力合并的模式下 最坏的时间复杂度是o(n)
看一下数据范围 1<=n,m<=100,000 多来几次这样的操作 你不超时谁超时
现在让我们了解一下一种新的合并方法 启发式合并
它的本质很简单:每次把短的合并到长的上面,那么合并一次的复杂度为
O
(
∣
短
的
队
列
∣
)
O(|短的队列|)
O(∣短的队列∣)。这样看上去貌似没有什么差别嘛 QAQ,接下来我们分析一下均摊复杂度。
考虑用贡献法来分析。我们令两个集合的分别为 A 和 B,且 |A|<|B|,那么我们把 A 暴力加入到 B 中。那么 A 中的元素所在的集合大小变成 |A|+|B|,也就是说至少变成了原来的两倍。所以每个元素至多被加入 nlogn 次,总的复杂度为O(nlogn)。
然后我看了好久好久终于发现这要用链表~~(好吧大佬告诉我的)~~
对于这道题目,我们先求出原序列的答案,对于每一种颜色都用类似链表的数据结构串起来,并记录下尾节点。每次修改,都根据启发式合并的方法来暴力合并,然后处理一下此次合并对答案的影响(显然答案是不增的)。
但是如果我们把 1 染成 2 并且 |S1|>|S2|,那么我们应该把 2 接到 1 的后面。这样会有一个问题:本次修改后这个链的颜色是 1(颜色为 2 的链被删除了),如果接下来修改颜色 22(显然这是合法的),会使得找不到颜色 2 而只能找到颜色 1 了。所以我们需要使用一个 f 数组,表示当我们要寻找颜色 x 时,实际上需要寻找颜色为f[x] 的链。如果遇到上面这种情况就要交换交换 f[x] 和 f[y]。
时间复杂度:O(nlogn)
详解键代码
#include<cstdio>
#include<iostream>
#include<cstring>
#define N 100005
#define M 1000005
using namespace std;
int n,m,cnt,ans;
int c[N],next[N],ft[M],head[M],s[M],st[M];
void solve(int a,int b)
{
for(int i=head[a];i;i=next[i])//变化颜色所在链 从头到尾遍历一遍
{
if(c[i+1]==b)ans--;// 链与目标链相邻 每改动一次端点则颜色段数-1
if(c[i-1]==b)ans--;
}
for(int i=head[a];i;i=next[i])c[i]=b;//遍历颜色所在链 ,改变值
next[st[a]]=head[b];head[b]=head[a];s[b]+=s[a];//此时颜色a的最早出现位置为颜色b所在链的头端点
head[a]=st[a]=s[a]=0;//删除颜色a 所在链 清空所有数据
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&c[i]);ft[c[i]]=c[i];//表示 链 f[c[i]] 的颜色为c[i]
if(c[i]!=c[i-1])ans++;//颜色段数增加
if(!head[c[i]])st[c[i]]=i;//st[]表示颜色c[i]出现的最早位置
s[c[i]]++;next[i]=head[c[i]];head[c[i]]=i;//s[]表示颜色为c[i]的数量增加 将端点i加入该颜色所处链表
} // head[c[i]]表示 颜色为c[i]的链的尾端点 next表示头端点
for(int i=1;i<=m;i++)
{
int x,a,b;
scanf("%d",&x);//输入命令
if(x==2)printf("%d\n",ans);//输出颜色段数
else
{
scanf("%d%d",&a,&b);//输入变化颜色和目标
if(a==b)continue;// 颜色和目标相同则继续输入命令
if(s[ft[a]]>s[ft[b]])//若变化颜色所在链的长度大于目标链 则交换所代表链的颜色
swap(ft[a],ft[b]);
a=ft[a];b=ft[b];
if(s[a]==0)continue;//若变换颜色所在链的长度为0 则继续输入命令
s[b]+=s[a];s[a]=0;//目标链所在长度增加 变化颜色所在链清空
solve(a,b);
}
}
return 0;
}
这是讲线性数据结构的时候老师给的题目 两次测试,都没过
有兴趣的可以好好研究一下