Cod sursa(job #2866318)

Utilizator Alecu123Alexandra Alecu123 Data 9 martie 2022 16:48:13
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.5 kb
#include <fstream>
#define N 1005
#define oo 1e9

using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");

int t,n,m,a[N][N];
int x;
int D[N];
bool viz[N];
int nod[N];

void Dijkstra(int x)
{

    int dmin,nod,i,j;
    for(int i=1;i<=n;i++) D[i]=a[x][i];
    viz[x]=1;
    for(int i=1;i<n;i++)
    {

      dmin=oo;
      for(int j=1;j<=n;j++)
        if(!viz[j] && D[j]<dmin)
        {
          dmin=D[j];
          nod=j;
        }
        if(dmin==oo) break;
        else
        {

            viz[nod]=1;
            for(int j=1;j<=n;j++)
                if(!viz[j] && D[nod]+a[nod][j]<D[j])
                    D[j]=D[nod]+a[nod][j];
        }
    }
}


bool Afisare()
{

    for(int i=1;i<=n;i++)
        if(D[i]!=nod[i]) return 0;
    return 1;
}


void Citire()
{

    int y,z,c;
    fin>>t;
    for(int i=1;i<=t;i++)
       {fin>>n>>m>>x;
        for(int i=1;i<=n;i++) fin>>nod[i];
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(i==j) a[i][j]=0;
                else a[i][j]=oo;
        for(int i=1;i<=m;i++)
          {
            fin>>y>>z>>c;
            a[y][z]=c;

          }
         Dijkstra(x);
        if(Afisare()) fout<<"DA";
        else fout<<"NU";
        fout<<"\n";

        for(int i=1;i<=n;i++)
        {
            viz[i]=0;
            D[i]=0;
            for(int j=1;j<=n;j++) a[i][j]=0;
        }
      }
}

int main()
{
	Citire();
    return 0;
}