许久未更新,今天做到一道有意思的题目,先上图,然后细说
这道题大体意思是给定一张无向图,其边权为正,要你求出总代价最小的生成树,生成树上
每条边 v(e)的代价为 w(e)*s(j),w(e)为边(i,j)的权值,s(j)是 对于 j 节点所在子树的节点
数总和。
题解:
该题看起来是最小生成树的题目,实际上只需从顶点跑一遍最短路后,然后将顶点到每个点的最短
路之和相加即可。
同时还可以拓展下,假设顶点不是固定的,只需对每个点跑一遍最短路算法,然后找出最短路总和
最小的那个顶点,输出该顶点最短路之和。
随后附上AC代码:
#include <bits/stdc++.h>
using namespace std;
using namespace std;
const int maxn=50007;
struct edge{
int v,w,nxt;
}e[maxn<<1]; //邻接表要开两倍
int p[maxn],n,m,eid,dist[maxn],r[maxn];
bool vst[maxn];
void init(){
memset(p,-1,sizeof(p));
eid=0;
}
typedef pair<int,int> PII;
set<PII,less<PII> > min_heap; //second:顶点下标 first:该顶点dist值
bool dij(int S){
memset(vst,0,sizeof(vst));
memset(dist,0x3f,sizeof(dist));
min_heap.insert(make_pair(0,S));
dist[S]=0;
for(int i=1;i<=n;i++){
if(min_heap.size()==0) return false;
auto iter=min_heap.begin();
int v=iter->second;
min_heap.erase(*iter);
vst[v]=true; //bfs
for(int j=p[v];j!=-1;j=e[j].nxt){
int t=e[j].v;
if(!vst[t]&&dist[v]+e[j].w<dist[t]){
//先将对应的pair从堆中删除 再将更新后的pair入堆
min_heap.erase(make_pair(dist[t],t));
dist[t]=dist[v]+e[j].w;//此时再更新
min_heap.insert(make_pair(dist[t],t));
//cout<<dist[t]<<" "<<t<<endl;
}
}
}
//n个点都可以从源点到达 返回true
return true;
}
void ins(int u,int v,int w){
e[eid].v=v;
e[eid].w=w;
e[eid].nxt=p[u];
p[u]=eid++;
}
void ins2(int u,int v,int w){ins(u,v,w);ins(v,u,w);}
int main(){
scanf("%d%d",&n,&m);
init();
for(int i=1;i<=n;i++)
scanf("%d",&r[i]);
for(int i=1;i<=m;i++) {int a,b,c;scanf("%d%d%d",&a,&b,&c);ins2(a,b,c);}
if(dij(1)){
long long ans=0;
for(int i=1;i<=n;i++) ans+=dist[i]*r[i];
cout<<ans;
}
else printf("No Answer");
return 0;
}
最后:欢迎转载,需要转载的话需要在转载文章中写明转载的源地址以及作者,如意疑问欢迎提出,如果有什么地方写错了欢迎指正。