求救,堆优化了,最后两个超时
查看原帖
求救,堆优化了,最后两个超时
1252709
young020202楼主2024/3/12 20:22
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
#include<stdio.h>
#include<queue>
using namespace std;
struct node{
	int x,y,t;
};
int n,m;
int TIME[300],mp[300][300],dis[300],f[300];
node ask[50010];
bool xiuhao[300];
int h[300],ver[40010],edge[40010],next[40010],idx=0;
void add(int u,int v,int w)
{
	idx++;
	ver[idx]=v;
	edge[idx]=w;
	next[idx]=h[u];
	h[u]=idx;
}
priority_queue<pair<int,int> >q;
int last=0;
void Dijsktra(int x,int y,int t)
{
	fill(dis,dis+300,0x7fffff);
	for(int i=last;i<n;i++)
	{
		if(TIME[i]<=t)
			xiuhao[i]=1;
		else
		{
			last=i;break;
		}
	}
	if(!xiuhao[x]||!xiuhao[y])	return;
	fill(f,f+n+1,0);
	dis[x]=0;
	q.push(make_pair(0,x));
	while(!q.empty())
	{
		int t=q.top().second;
		q.pop();
		if(f[t])	continue;
		f[t]=1;
		for(int i=h[t];i;i=next[i])
		{
			if(xiuhao[ver[i]]==0)	continue;
			int y=ver[i],w=edge[i];
			if(dis[y]>dis[t]+w&&xiuhao[t]&&f[y]==0)
			{
				dis[y]=dis[t]+w;
				q.push(make_pair(-dis[y],y));
			}
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
		scanf("%d",&TIME[i]);
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	int q;
	cin>>q;
	for(int i=1;i<=q;i++)
		scanf("%d%d%d",&ask[i].x,&ask[i].y,&ask[i].t);
	for(int i=1;i<=q;i++)
	{
		Dijsktra(ask[i].x,ask[i].y,ask[i].t);
		if(dis[ask[i].y]!=0x7fffff)
			printf("%d\n",dis[ask[i].y]);
		else
			printf("-1\n");
	}
}

2024/3/12 20:22
加载中...