OJ P1222 Watering Hole

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
int dis[320];
bool vis[320];
int sumn=0,c;
int a[310][310];
int v=1,v1;
int n;
void init()
{
 cin>>n;
 memset(a,10,sizeof(a));
 a[0][0]=0;
 for(int i=1;i<=n;i++)
 {
  cin>>a[0][i];
  a[i][0]=a[0][i];
 }
 for(int i=1;i<=n;i++)
 { 
  vis[i]=false;
  for(int j=1;j<=n;j++)
   cin>>a[i][j];
 }
}
void MST_Prim(int s)
{
 memset(dis,10,sizeof(dis));
 for(int i=0;i<=n;i++)
  dis[i]=a[s][i];
 memset(vis,0,sizeof(vis));
 vis[s]=1;sumn=0;
 for(int i=1;i<=n;i++)
 {
  int minn=21479565,c=0;
  for(int j=0;j<=n;j++)
   if((!vis[j])&&(dis[j]<minn))
   {
    minn=dis[j];
    c=j;
   }
  vis[c]=1;
  sumn+=minn;
  for(int j=0;j<=n;j++)
   if((a[c][j]<dis[j])&&(!vis[j]))
    dis[j]=a[c][j];
 }
}
int main()
{
 
 init();
 MST_Prim(0);
 cout<<sumn;
return 0;
}