求助,找不出哪里错了QAQ
查看原帖
求助,找不出哪里错了QAQ
230804
Durancer楼主2020/10/31 15:33
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
#include<map>
#include<cmath>
#include<algorithm>
using namespace std;
int head[1000009],e_num,dis[1000009],v[1000009],ans;
int n,m,k,s,t;//城市数量,航线数,免费次数,起点,终点 
struct node{
	int last;
	int to;
	int dis;
}e[10000009];
void add(int from,int to,int dis)
{
	e[++e_num].last=head[from];
	e[e_num].to=to;
	e[e_num].dis=dis;
	head[from]=e_num;
}
void djstm(int k)
{
	memset(dis,0x3f3f3f3f,sizeof(dis));
	queue<int> q;
	q.push(k);
	dis[k]=0;
	while(!q.empty())
	{
		int cun=q.front();
		q.pop();
		if(v[cun]==1) continue;
		v[cun]=1;
		for(int i=head[cun];i;i=e[i].last)
		{
			int to=e[i].to;
			int w=e[i].dis;
			if(dis[to]>dis[cun]+w)
			{
				dis[to]=dis[cun]+w;
				q.push(to);
			}
		}
	}
}
int main()
{
	cin>>n>>m>>k>>s>>t;
	s++;
	t++;//0-n-1,转换成-n
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		u++;
		v++;
		add(u,v,w);
		add(v,u,w);
		for(int j=1;j<=k;j++)
		{
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
			add(u+(j-1)*n,v+j*n,0);
			add(v+(j-1)*n,u+j*n,0); 
		}
	}
	for(int i=1;i<=k;i++)
	{
		add(t+(i-1)*n,t+i*n,0);//到最底下 
	}
	djstm(s);
	cout<<dis[t+k*n]<<endl;
	return 0;
}

2020/10/31 15:33
加载中...