图论-生成树 新的开始

第一次遇到这种加点的题目。
本题的不同之处就在于在每个点上建发电站需要费用，不能直接用最小生成树的模型来做，其他办法也不知道该如何处理。
于是考虑对题目的模型进行转化，将它化归成最小生成树模型。
因此有加点的技巧：原图有0至n-1共n个点，加入一个虚点n,map[n][i]即为建立发电站费用的v[i]，对0至n这(n+1)个点做最小生成树得到的结果就是答案。
另外，听闻图论中还有一些题目要用到拆点什么的技巧，暂时还不懂。

#include<bits/stdc++.h>
#define rep(i,l,r) for(int i=(l);i<=(r);i++)
#define per(i,r,l) for(int i=(r);i>=(l);i--)
using namespace std;
const int inf=1e9+10,N=400;
int n,a,mymap[N][N],dis[N],ans;
bool vis[N];
int main(){
	ios::sync_with_stdio(false); cin.tie(0);
	cin>>n;
	rep(i,0,n-1) cin>>a,mymap[n][i]=mymap[i][n]=a;
	rep(i,0,n-1) rep(j,0,n-1) cin>>mymap[i][j];
	memset(dis,127,sizeof(dis)); mymap[n][n]=0;
	dis[0]=0;
	rep(i,0,n){
		int k,now=inf;
		rep(j,0,n) if((vis[j]==false)&&(dis[j]<now)) now=dis[j],k=j;
		vis[k]=true; ans+=now;
		rep(j,0,n) dis[j]=min(dis[j],mymap[k][j]);	
	}
	cout<<ans;
	return 0;
}