首页 > 代码库 > poj3013 Big Christmas Tree --- 最短路
poj3013 Big Christmas Tree --- 最短路
我都不好意思在标题上写这是最短路
这题挺有意思,关键在于把题目所求的量转换为最短路问题。
题意:
给一个无向图,每个结点有权值p[i],每条边有权值w[i]
求使这颗树所有顶点与根节点1联通的最小花费,
最小花费=∑w[i]×∑p[i]
第一个∑是所有边,第二个∑是该边下所有结点的权值和
思路:
通过推导可以发现,对于每个结点,它被算入的花费为 p[i]*d[i],
d[i]为该结点到根结点的距离。
于是豁然开朗了吧。。水题一道
唯一的坑点是要int64,而且inf要足够大
#include<cstdio> #include<cstring> #include<queue> #include<iostream> using namespace std; const ll inf=1ll<<60;//坑死!这题inf必须足够大 #define N 50010 #define M 100010 struct node { int v,next; ll w; }e[M]; int vis[N],n,m,head[N],h; ll d[N],p[N]; void addedge(int a,int b,ll c) { e[h].v=b; e[h].w=c; e[h].next=head[a]; head[a]=h++; } int spfa(int st) { memset(vis,0,sizeof vis); for(int i=0;i<=n;i++) d[i]=inf; d[st]=0; vis[st]=1; queue<int> q; q.push(st); int x,i,v; while(!q.empty()) { x=q.front(); q.pop(); vis[x]=0; for(i=head[x];i!=-1;i=e[i].next) { v=e[i].v; if(d[v]>d[x]+e[i].w) { d[v]=d[x]+e[i].w; if(!vis[v]) { vis[v]=1; q.push(v); } } } } } void init() { memset(head,-1,sizeof head); h=0; } int main() { int t,a,b,i; ll ans,c; scanf("%d",&t); while(t--) { init(); scanf("%d%d",&n,&m); for(i=1;i<=n;i++) cin>>p[i]; while(m--) { scanf("%d%d",&a,&b); cin>>c; addedge(a,b,c); addedge(b,a,c); } spfa(1); ans=0; int flag=1; for(i=2;i<=n;i++) { if(d[i]==inf) { flag=0; break; } ans+=(d[i]*p[i]); } if(!flag) printf("No Answer\n"); else cout<<ans<<endl; } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。