Cod sursa(job #403371)

Utilizator cristiprgPrigoana Cristian cristiprg Data 24 februarie 2010 21:38:51
Problema Drumuri minime Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.57 kb
#include <cstdio>
#include <set>
#include <cmath>
#define DIM 50005
#define mp make_pair
#define MOD 104659

using namespace std;
const int INF = 1<<30;
const double eps = 1e-10;
struct nod
{
    int x;
	double cost;
    nod *next;
};
nod *G[DIM];

int n, m, uz[DIM], v[DIM];
double d[DIM];
 set< pair<double, int>  > S;



void addMuchie(int i, int j, int cost)
{
    nod *p = new nod;
    p->x = j;
    p->cost = log10(cost);
    p->next = G[i];
    G[i] = p;

}

void dijkstra()
{
   S.insert(mp(0, 1));

   d[1] = 0;
   v[1] = 1;

   int i;
   while (S.size() > 0)
   {
      i = (*S.begin()).second;
      v[i] = 1;

      S.erase(*S.begin());

      for (nod *p = G[i]; p; p = p -> next)
       if (!v[p->x])
      {
            if   (abs(d[p->x] - (d[i] + p->cost)) <= eps)
					uz[p->x] += uz[i] , uz[p->x] %= MOD;
            else
                if (d[p->x] > d[i] + p->cost )
                    d[p->x] = d[i] + p->cost, S.insert(mp(d[p->x], p->x)),uz[p->x] = uz[i];

       //     printf("%d %d %lf %d\n", i, p->x, d[p->x], uz[p->x]);

      }

   }

}

int main()
{
    FILE *f = fopen("dmin.in", "r");
    fscanf(f, "%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
        d[i] = INF, uz[i] = 1;

    for (int x, y, c; m; --m)
    {
        fscanf(f, "%d%d%d", &x, &y, &c);
        addMuchie(x, y, c);
		addMuchie(y, x, c);

    }
    fclose(f);

    dijkstra();
    f = fopen("dmin.out", "w");

    for (int i = 2; i <= n; ++i)
        fprintf(f, "%d ", uz[i]);
    fclose(f);

    return 0;
}