Codeforces Round #257 (Div. 1) B. Jzzhu and Cities(多条最短路)

题目链接
在这里插入图片描述
在这里插入图片描述
思路:又是一个阅读理解图。。。那么k条火车站居然是建在图里。。。
spfa一遍同时记录一下到各个点的最短路径有几条,有多条的时候也要删去火车站。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+1;
int n,m,k,u,v,w,x[maxn],y[maxn],cnt=0;
ll d[maxn],num[maxn];
bool vis[maxn];
vector<pair<int,int>>g[maxn<<1];
void spfa(int x)
{
	memset(vis,false,sizeof(vis));
	for(int i=0;i<=n;++i) d[i]=1e18,num[i]=0;
	d[x]=0;num[x]=1;
	queue<int>q;
	q.push(x);
	vis[x]=true;
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		vis[top]=false;
		for(auto v:g[top])
		{
			if(d[top]+v.second<d[v.first])
			{
				d[v.first]=d[top]+v.second;
				num[v.first]=num[top];
				if(!vis[v.first]) q.push(v.first),vis[v.first]=true;
			}
			else if(d[top]+v.second==d[v.first])
			{
				num[v.first]++;
				
			}
		}
	}
}
int main()
{
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1;i<=m;++i)
	{
		scanf("%d %d %d",&u,&v,&w);
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	} 
	for(int i=1;i<=k;++i)
	{
		scanf("%d %d",&x[i],&y[i]);
		g[1].push_back({x[i],y[i]});
		g[x[i]].push_back({1,y[i]});
	}
	spfa(1);
	for(int i=1;i<=k;++i)
	{
		if(d[x[i]]<y[i]) cnt++;
		else if(d[x[i]]==y[i]&&num[x[i]]>1) num[x[i]]--,cnt++;
	}
	printf("%d\n",cnt);
}
发布了283 篇原创文章 · 获赞 0 · 访问量 7296

猜你喜欢

转载自blog.csdn.net/qq_42479630/article/details/105126736