Problema #591 Firma
De pe pbinfo

Cerința

Într-o țară sunt n orașe, numerotate de la 1 la n, unite între ele prin m șosele bidirecționale de lungimi cunoscute, între oricare două orașe existând drum, fie șosea directă, fie prin alte orașe. O firmă dorește să-și stabilească sediul în unul dintre orașe, astfel încât suma lungimilor drumurilor minime de la orașul în care se află sediul la toate celelaltele orașe să fie minimă. Determinați orașul care va fi ales pentru sediul firmei. Dacă sunt mai multe orașe care pot fi alese, se va alege cel cu numărul de ordine mai mic.

Date de intrare
Fișierul de intrare firma.in conține pe prima linie numerele n m, iar următoarele m linii câte trei numere i j L, cu semnificatia: între orașele i și j există o șosea directă de lungime L.

Date de ieșire
Fișierul de ieșire firma.out va conține pe prima linie numărul X, reprezentând numărul de ordine al orașului care va ales ca sediu pentru firmă.

Restricții și precizări
1 ≤ n ≤ 100


Răspuns :

Răspuns:

#include <fstream>

#include <queue>

#include <climits>

using namespace std;

ifstream fin( "firma.in" );

ofstream fout( "firma.out" );

bool                viz[ 101 ];

int                 dist[ 101 ], mat[ 101 ][ 101 ], n = 0, m = 0;

struct nod

{

   int             cost = 0;

   int             x = 0;

   bool operator<( const nod& rhs ) const

   {

       return cost > rhs.cost;

   }

};

priority_queue <nod> q;

void showqueue()

{

   while( !q.empty() )

   {

       fout << q.top().x << " " << q.top().cost << " ";

       q.pop();

   }

}

void dijkstra( int start )

{

   for( int i = 1; i <= n; ++i )

   {

       viz[ i ] = false;

       if( mat[ start ][ i ] != INT_MAX )

       {

           dist[ i ] = mat[ start ][ i ];

           nod             a;

           a.cost = dist[ i ];

           a.x = i;

           q.push( a );

       }

       else

       {

           dist[ i ] = INT_MAX;

       }

   }

   dist[ start ] = 0;

   viz[ start ] = true;

   while( !q.empty() )

   {

       nod             a;

       a = q.top();

       q.pop();

       viz[ a.x ] = true;

       for( int i = 1; i <= n; ++i )

       {

           if( !viz[ i ] && mat[ a.x ][ i ] != INT_MAX && dist[ i ] >= dist[ a.x ] + mat[ a.x ][ i ] )

           {

               dist[ i ] = dist[ a.x ] + mat[ a.x ][ i ];

               nod             curent;

               curent.x = i;

               curent.cost = dist[ i ];

               q.push( curent );

           }

       }

   }

}

int main()

{

   fin >> n >> m;

   for( int i = 1; i <= n; ++i )

   {

       for( int j = 1; j <= n; ++j )

       {

           mat[ i ][ j ] = INT_MAX;

       }

   }

   int             x = 0, y = 0, cost = 0, minim = 100000, indice = 0;

   for( int i = 1; i <= m; ++i )

   {

       fin >> x >> y >> cost;

       mat[ x ][ y ] = cost;

       mat[ y ][ x ] = cost;

   }

   for( int i = 1; i <= n; ++i )

   {

       int             suma = 0;

       dijkstra( i );

       for( int j = 1; j <= n; ++j )

       {

           suma += dist[ j ];

       }

       if( suma < minim )

       {

           minim = suma;

           indice = i;

       }

   }

   fout << indice;

   return 0;

}

Explicație: