Sweet Butter

最新推荐文章于 2025-12-07 17:14:34 发布

原创最新推荐文章于 2025-12-07 17:14:34 发布 · 172 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#图论 #算法

简单的一道单源最小路径问题，算出不同的牧场放置糖时奶牛要走过的不同距离，再进行比较，最小的即可。

用弗洛伊德和最基本的迪杰斯特拉肯定是超时的，这儿用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;
	}

有点搞不懂。