全TLE求助!
查看原帖
全TLE求助!
783336
Earth_Sky楼主2024/5/17 21:19
#include<bits/stdc++.h>
using namespace std;
const int MAX=2e5+5,INF=INT_MAX;
struct node{
	int u,dis;
	bool operator <(const node& rhs)const{
		return dis>rhs.dis;
	}
};
bool vis[10005];
int n,m,k,dis[MAX];
int h[MAX],e[MAX],ne[MAX],edis[MAX],tot=1;
void add(int x,int y,int dis){
	edis[tot]=dis;
	e[tot]=y;
	ne[tot]=h[x];
	h[x]=tot++;
}
void dijkstra(){
	priority_queue<node>q;
	q.push((node){1,0});
	while(!q.empty()){
		int u=q.top().u;q.pop();
		vis[u]=1;
		for(int i=h[u];i!=-1;i=ne[i]){
			if(!vis[e[i]]&&dis[e[i]]>dis[u]+edis[i]){
				dis[e[i]]=dis[u]+edis[i];
				q.push((node){e[i],dis[e[i]]});
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(NULL);
	cin.tie(0);cout.tie(0);
	memset(h,-1,sizeof -1);
	cin>>n>>m>>k;
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		for(int j=0;j<=k;j++) add(j*n+x,j*n+y,z),add(j*n+y,j*n+x,z);
		for(int j=0;j<k;j++) add(j*n+x,(j+1)*n+y,z/2),add(j*n+y,(j+1)*n+x,z/2);
	}
	for(int i=1;i<=n*k+n;i++) dis[i]=INF;
	dis[1]=0;
	dijkstra();
	int minn=INF;
	for(int i=0;i<=k;i++) minn=min(minn,dis[i*n+n]);
	cout<<minn;

	return 0;
}

我实在是不知道哪里错了,帮忙看一看呗

2024/5/17 21:19
加载中...