CF986A Fair

题意:你有n个物品,m个连接关系,k个标号,n个物品每个都有一个不大于k的编号,你的任务是求出n个点,每个点到k个标号中的s个最短路,每条边的边权都是1

解法:普通的最短路肯定会T,这道题用的是多源BFS,每次讲1~k中一个编号的所有点加入队列去更新其他点,那么就变成了多源bfs,从一个点扩张到整张图变成了从多个点扩散到整张图,复杂度瞬间降低

代码

//By AcerMo
#include<cmath>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int M=2e5+50;
int n,m,k,s;
vector<int>v[M];
vector<int>G[M];
int dis[M/2][105];
int vis[M/2][105];
inline int read()
{
	int x=0;char ch=getchar();
	while (ch>'9'||ch<'0') ch=getchar();
	while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x;
}
inline void bfs(int x)
{
	queue<int>q;
	for (int i=0;i<G[x].size();i++)
	q.push(G[x][i]),dis[G[x][i]][x]=0;
	while (q.size())
	{
		int u=q.front();q.pop();vis[u][x]=0;
		for (int i=0;i<v[u].size();i++)
		{
			int go=v[u][i];
			if (dis[go][x]>dis[u][x]+1)
			{
				dis[go][x]=dis[u][x]+1;
				if (!vis[go][x])
				{
					vis[go][x]=1;q.push(go);
				}
			}
		}
	}
	return ;
}
int main()
{
	memset(dis,0x3f,sizeof(dis));
	n=read();m=read();k=read();s=read();
	for (int i=1;i<=n;i++) 
	{
		int a=read();
		G[a].push_back(i);
	}
	for (int i=1;i<=m;i++)
	{
		int a=read(),b=read();
		v[a].push_back(b);
		v[b].push_back(a);
	}
	for (int i=1;i<=k;i++) bfs(i);
	for (int i=1;i<=n;i++)
	{
		sort(dis[i]+1,dis[i]+1+k);
		int ans=0;
		for (int j=1;j<=s;j++) ans+=dis[i][j];
		cout<<ans<<" ";
	}
	return 0;
}	

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值