【CF】1775D-Friendly Spiders 题解

传送门：1775D
标签：图论

题目大意

给出n只蜘蛛，第i只蜘蛛有a_i条腿。蜘蛛i和蜘蛛j是朋友当且仅当a_i和a_j的公因数不为1(即gcd(a_i,a_j)!=1)。现规定只有互为朋友的蜘蛛间能直接传递信息，求第x只蜘蛛传递信息给第y只蜘蛛至少需要经过几只蜘蛛。
输入：n，a_i(1<=n,a_i<=3e5)，x，y。
输出：如果x与y间无法通信，输出-1。否则输出最短路径上的蜘蛛数量，并按路径顺序输出各蜘蛛的原下标。

算法分析

• 题目已经明示了要我们求最短路，只不过没有直接给出各边，而是通过gcd在点和点之间建立联系。我们知道，求多源最短路用Floyd，求单点到多点的最短路一般用堆优化Dijkstra，而求单点到单点的最短路用bfs。这题显然是单点到单点，看数据范围也不会T，那么就可以基本确定用bfs了。
• 但是本题并没有这么简单，我们还需要解决一个关键的问题：如何建边。你可能会觉得：双重for让每两个点都跑一遍gcd。如果这么做，光是预处理的时间复杂度就到了O(n²)，n最大为3e5显然会超时。而且如果是极端情况：所有的蜘蛛腿数都相同，那么邻接表就要存(3e5)²=9e10个点，必然是会MLE。也就是说，将所有的边全部表示出来在时间和空间上都不可行。那我们只好想办法降低复杂度，但又不可能提前知道两点通信需要的是哪些边，也就没办法在原来的算法上剪枝。
• 我们不妨将一些点看作一个整体，准确来说是一个层。同一层的点都有一个共同的质因子，两两间可以直接通信。一个数可能有很多质因子，在很多层，同时一个层又包含很多个数。也就是说，点和层之间并没有严格的映射关系。比如：当n=3，且序列a为[2,3,6]时，2和6都在“2”这一层，3和6都在“3”这一层。这样做就相当于省去了同一层各点间的边，降低了空间复杂度。那要怎么快速地将每个点分到各个层呢？当然是对每个a_i分解质因子，可以预处理范围里所有质数，也可以预处理所有数的最小质因子。我这里采用的是后者。
• 还有最后一个问题，分层后如何bfs。如果以层为单位，就要在层与层间建边，而且也不好判断是否经过了目标点。如果以点为单位，显然会TLE。那就折中一下，让每次搜索的起点为点，终点为层，再以终点层内的每个点为起点继续搜索。剪枝也很简单：经过的层和点都被标记，再次经过就直接continue。而空间上只需建两个动态二维数组，分别存储点所在的多个层以及层包含的多个点。

代码实现

#include <iostream>
using namespace std;
#include <queue>
#include <vector>
queue<int> q;	//bfs用的队列 
vector<int> ans,to[300001],from[300001];//分别存路径、点的层、层的点 
vector<int>::reverse_iterator iter;	//反向迭代器 
int x,y,prime[300001],pm[300001],pre[300001],ranks[300001];
bool visit[300001];
void init(){	//预处理 
	for(int i=2;i<=300000;i++){	//线性筛最小质因子 
		if(!pm[i]){		//如果是质数 
			prime[++prime[0]]=i;
			pm[i]=i;	//最小质因子为本身 
		}
		for(int j=1;j<=prime[prime[0]];j++){
			if(i*prime[j]>300000)break;
			pm[i*prime[j]]=prime[j];
			if(i%prime[j]==0)break;	//防止重复筛 
		}
	}
}
void bfs(){
	ranks[x]=1;	//起始路径只有一个点 
	q.push(x);
	while(!q.empty()){
		x=q.front();
		q.pop();
		if(x==y){	//找到目标点 
			cout<<ranks[y]<<'\n';
			for(int i=0,y1=y;i<ranks[y];i++,y1=pre[y1])
				ans.push_back(y1);	//逆序存储路径 
			for(iter=ans.rbegin();iter!=ans.rend();iter++)
				cout<<(*iter)<<' ';	//输出路径上各点原下标 
			return ;
		}
		for(auto i:to[x]){
			if(visit[i])continue;	//搜索剪枝 
			visit[i]=1;
			for(auto j:from[i]){
				if(j==x||ranks[j])continue;	//搜索剪枝 
				ranks[j]=ranks[x]+1;
				pre[j]=x;	//记录前一个点 
				q.push(j);
			}
		}
	}
	cout<<-1;
}
int main(){
	int i,n;
	std::ios::sync_with_stdio(false);//取消同步流
	std::cin.tie(0);std::cout.tie(0);
	init();
	cin>>n;
	for(i=1;i<=n;i++){
		cin>>x;
		while(x>1){
			to[i].push_back(pm[x]);	//存层 
			from[pm[x]].push_back(i);//存点 
			x/=pm[x];
		}
	}
	cin>>x>>y;	//起点和终点 
	bfs();
	return 0;
}