O2 (#2 #9 WA) 不开O2(#2 #9 #10 T飞)
查看原帖
O2 (#2 #9 WA) 不开O2(#2 #9 #10 T飞)
364847
_Kouki_楼主2022/2/5 23:15
#include<bits/stdc++.h>
using namespace std;
struct edge
{
	int to,dis,next;
};
edge e[100001];
int head[1001],dis[1001],cnt;
bool vis[1001];
int n,m;
void add_edge(int u,int v,int w)
{
	++cnt;
	e[cnt].dis=w;
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt; 
}
struct node
{
	int dis;
	int pos;
	bool operator <(const node &x)const
	{
		return x.dis<dis;
	}
};
void dij(int s)
{
	priority_queue<node>q;
	for(int i=1;i<=n;i++) dis[i]=0x7fffffff,vis[i]=0;
	dis[s]=0;
	q.push((node){0,s});
	while(!q.empty())
	{
		node tmp=q.top();
		q.pop();
		int x=tmp.pos;
		if(vis[x]) continue;
		vis[x]=1;
		for(int i=head[x];i;i=e[i].next)
		{
			int y=e[i].to;
			if(dis[y]>dis[x]+e[i].dis)
			{
				dis[y]=dis[x]+e[i].dis;
				if(!vis[y])
				{
					q.push((node){dis[y],y});
				}
			}
		}
	}
}
int ms[10001];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++) scanf("%d",&ms[i]);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			int tx;
			scanf("%d",&tx);
			add_edge(i,j,tx);
		}
	}
	int ans=0;
	ms[0]=1;ms[m+1]=n;
	for(int i=1;i<=m+1;i++)
	{

		dij(ms[i-1]);
		ans+=dis[ms[i]];
	}
	printf("%d",ans);
	return 0;
}
2022/2/5 23:15
加载中...