Cod sursa(job #1608196)

Utilizator UngureanuRuxandraUngureanu Andreea Ruxandra UngureanuRuxandra Data 21 februarie 2016 21:46:50
Problema Distante Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int i,n,m,j,mini,x,y,cost,k,t2,nr,p,r,d[5001],s[5001],t[5001],v[5001];
double long a[1001][1001];
int main()
{f>>t2;
for(p=1;p<=t2;++p)
    {f>>n>>m>>r;
for(i=1;i<=n;++i)
    f>>v[i];
for(i=1;i<=n;++i)
    for(j=i+1;j<=n;++j)
{a[i][j]=32000;
 a[j][i]=32000;
}
  for(i=1;i<=m;++i)
  {f>>x>>y>>cost;
   a[x][y]=cost;
   a[y][x]=cost;
  }
  for(i=1;i<=n;++i)
  {t[i]=0;
  d[i]=0;
  s[i]=0;}
  for(i=1;i<=n;++i)
   {d[i]=a[r][i];
   if(d[i]<32000)
     t[i]=r;}
   s[r]=1;
  t[r]=0;
  x=0;
  while(1)
  {mini=32000;
   x++;
   for(i=1;i<=n;++i)
     if(s[i]==0&&d[i]<mini)
   {mini=d[i];
    k=i;

   }
   if(mini==32000||x==n)
     break;
   s[k]=1;
   for(i=1;i<=n;++i)
    if(s[i]==0&&d[i]>mini+a[k][i])
      {d[i]=a[k][i]+mini;
    t[i]=k;}

  }
nr=0;
  for(i=1;i<=n;++i)
    if(d[i]==v[i])
      nr++;
  if(nr==n)
   g<<"DA"<<'\n';
  else
    g<<"NU"<<'\n';
}
    return 0;
}