求调样例不过55pts
查看原帖
求调样例不过55pts
690243
houluyu楼主2023/11/25 19:28
#include<bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int N=200005;
typedef pair<int,int> PII;
priority_queue<PII,vector<PII>,greater<PII>> q;
vector<vector<PII> > g(N);
int n,m,k,s1,t;
bool vis[N];
int dist[N]; 
void add(int x,int y,int z)
{
	g[x].push_back({y,z});
	g[y].push_back({x,z});
} 
void dij()
{
	memset(dist,127,sizeof dist);
	dist[s1]=0;
	q.push({0,s1});
	while(!q.empty())
	{
		int t=q.top().s;
		q.pop();
		if(vis[t])continue;
		vis[t]=1;
		for(int i=0;i<g[t].size();i++)
		{
			int t1=g[t][i].f,t2=g[t][i].s;
			if(dist[t1]>dist[t]+t2)
			{
				dist[t1]=dist[t]+t2;
				q.push({dist[t1],t1});
			}
		}
	}
}
int main()
{
	scanf("%d%d%d%d%d",&n,&m,&k,&s1,&t);
	for(int i=1;i<=m;i++)
	{
		int x,y,z; 
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
		for(int j=1;j<=k;j++)
		{
			add(x+j*n,y+j*n,z);
			add(y+j*n,x+j*n,z);
			add(x+(j-1)*n,y+j*n,0);
			add(y+(j-1)*n,x+j*n,0);
		} 
	}

	dij();
	int ans=0x7fffffff;
	for(int i=0;i<=k;i++)
	{
		ans=min(ans,dist[t+n*i]);
	} 
	printf("%d\n",ans);
	return 0; 
	
}
2023/11/25 19:28
加载中...