最小生成树变型——[Usaco2008 Oct]灌水

#include<stdio.h>

int map[309][309];
int dis[309];
int p[309];
bool use[309];
int n;

void krus()
{
int i,min,rj,j;
for(i=0;i<=n;i++)
{
use[i]=0;
dis[i]=map[0][i];
}
use[0]=1;
int add=0;

for(i=0;i<n;i++)
{
min=99999999;
for(j=0;j<=n;j++)
{
if(use[j]==1)continue;
if(dis[j]<min)
{
min=dis[j];
rj=j;
}
}
use[rj]=1;
add+=min;

for(j=0;j<=n;j++)
{
if(use[j]==1)continue;
if(dis[j]>map[rj][j])
dis[j]=map[rj][j];
}
}

printf("%d\n",add);
}

int main()
{
int i,j;
while(scanf("%d",&n)!=EOF)
{
for(i=1;i<=n;i++)
scanf("%d",&p[i]);

for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
scanf("%d",&map[i][j]);
}
}
for(i=1;i<=n;i++)
{
map[0][i]=p[i];
map[i][0]=p[i];
}

krus();
}
}