LOJ10066 新的开始

题目链接 https://loj.ac/problem/10066

题解：这道题只是在普通的prim加了一个条件，就是可以直接花一定的代价直接到达这个点。这时候，我们只要再加一个点，这个点到其他的点的连边的值表示直接到达该点花费。然后再跑一遍prim就好了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int inf=1e9+7;
int vis[510],p[510][510],d[510],minn,ans,n,k,v[510];
void prim(int v0)
{
	memset(vis,0,sizeof(vis));
	for (int i=1;i<=n;i++)	d[i]=inf;
	d[v0]=0;
	ans=0;
	for (int i=1;i<=n+1;i++)
	{
		minn=inf;
		for (int j=1;j<=n+1;j++)
			if (vis[j]==0 && minn>d[j]) {minn=d[j];k=j;}
		vis[k]=1;ans=ans+d[k];//printf("%d\n",ans);
		for (int j=1;j<=n+1;j++)
			if (vis[j]==0 && d[j]>p[k][j]) d[j]=p[k][j];
	}
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&v[i]);
		p[n+1][i]=v[i];
		p[i][n+1]=v[i];
	}
	for (int i=1;i<=n;i++)
		for (int j=1;j<=n;j++)
		{
			scanf("%d",&p[i][j]);
		}
	prim(n+1);
	printf("%d\n",ans);
}