简单的一道单源最小路径问题,算出不同的牧场放置糖时奶牛要走过的不同距离,再进行比较,最小的即可。
用弗洛伊德和最基本的迪杰斯特拉肯定是超时的,这儿用spfa就可以。
话不多说,直接上图
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
struct Edge{
int to,next,dis;
}edge[500005];
int head[10005],dis[10005],vis[10005],d[10005];
int pos=0;
int n,p,c;
void addedge(int u,int v,int w)
{
edge[++pos].next=head[u];
edge[pos].to=v;
edge[pos].dis=w;
head[u]=pos;
}
int spfa(int s)
{
queue<int >q;
memset(dis,INF,sizeof dis);
q.push(s);
vis[s]=1;
dis[s]=0;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
for(int i=head[u];i;i=edge[i].next)
{
int v=edge[i].to;
if(dis[v]>dis[u]+edge[i].dis)
{
dis[v]=dis[u]+edge[i].dis;
if(!vis[v])
{
vis[v]=1;
q.push(v);
}
}
}
}
int path=0;
for(int i=0;i<n;i++)
{
int t=d[i];
if(dis[t]==INF) return INF;
path+=dis[t];
}
return path;
}
int main()
{
scanf("%d %d %d",&n,&p,&c);
for(int i=0;i<n;i++)
{
scanf("%d",&d[i]);
}
int u,v,w;
for(int i=0;i<c;i++)
{
scanf("%d %d %d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
int ans=INF;
for(int i=1;i<=p;i++)
{
//cout<<spfa(i)<<endl;
ans=min(ans,spfa(i));
}
cout<<ans<<endl;
return 0;
}
用memset初始化是正确的,但是用for循环初始化就得不到正确的结果,也不知道为什么
memset(dis,INF,sizeof dis);//这是正确的
q.push(s);
vis[s]=1;
dis[s]=0;
for(int i=1;i<=n;i++)//这样子就不对了
{
dis[i]=INF;
//vis[i]=0;
}
有点搞不懂。