求助,本题80最后两个点WA
查看原帖
求助,本题80最后两个点WA
602527
star_maelstorm楼主2022/9/11 13:20
#include<bits/stdc++.h>
using namespace std;
const int maxn=10010;
struct node
{
	int u,v,w;
}edge[maxn];
int head[maxn];
int nxt[maxn];
int dis[maxn];
bool vis[maxn];
int edge_cnt; 
int n,k,m,s,t;
int c[maxn];
int dislike[maxn][maxn];
int inq[maxn];
void add(int u,int v,int w)
{
	edge[++edge_cnt].u=u;
	edge[edge_cnt].v=v;
	edge[edge_cnt].w=w;
	nxt[edge_cnt]=head[u];
	head[u]=edge_cnt;
}
void spfa()
{
	queue<int> q;
	memset(dis,0x7f,sizeof(dis));
	dis[s]=0;
	vis[s]=true;
	q.push(s);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		vis[u]=false;
		for(int i=head[u];i;i=nxt[i])
		{
			int v=edge[i].v;
			if(dis[v]>dis[u]+edge[i].w&&!dislike[c[u]][c[v]])
			{
				dis[v]=dis[u]+edge[i].w;
				if(!vis[v])
				{
					vis[v]=true;
					q.push(v);
				}
			}
		}
	}
}
int main()
{
	cin>>n>>k>>m>>s>>t;
	for(int i=1;i<=n;i++)
	{
	    cin>>c[i];
	}
	for(int i=1;i<=k;i++)
	{
		for(int j=1;j<=k;j++)
		{
			int num;
			cin>>num;
			dislike[j][i]=num;
		}
	}
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
	spfa();
	if(dis[t]==2139062143)
	{
		cout<<-1<<endl;
	} 
	else cout<<dis[t]<<endl;
	return 0;
}
2022/9/11 13:20
加载中...