Cod sursa(job #2306229)

Utilizator maria_sinteaMaria Sintea maria_sintea Data 21 decembrie 2018 19:50:25
Problema Drumuri minime Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.62 kb
#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
#define mod 104659
#define inf 0x3f3f3f3f
#define N 1505

using namespace std;

vector <pair <int, int> > g[N];
vector <pair <int, int> > ::iterator it;
queue <int> q;
int n, m, cost[N], viz[N], nr[N];

void citire()
{
    scanf("%d %d\n", &n, &m);
    for(int i=1;i<=n;i++)
        cost[i]=inf;
    for(int i=0;i<m;i++)
    {
        int x, y, c;
        scanf("%d %d %d\n", &x, &y, &c);
        g[x].push_back({y, c});
        g[y].push_back({x, c});
    }
}

void parcurg(int nod)
{
    for(it=g[nod].begin();it!=g[nod].end();it++)
        if(it->second+cost[nod]<cost[it->first])
        {
            cost[it->first]=it->second+cost[nod];
            if(!viz[it->first])
            {
                q.push(it->first);
                viz[it->first]=1;
            }
        }
}

void afisare()
{
    for(int i=2;i<=n;i++)
        printf("%d ", nr[i]);
}

void bellman()
{
    q.push(1);
    cost[1]=1;
    viz[1]=0;
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        viz[x]=0;
        parcurg(x);
    }
}

int calcul(int x)
{
    if(nr[x]!=0)
        return nr[x];
    for(it=g[x].begin();it!=g[x].end();it++)
        if(it->second+cost[it->first]==cost[x])
            nr[x]=(1LL*nr[x]+calcul(it->first))%mod;
    return nr[x];
}

int main()
{
    freopen("dmin.in", "r", stdin);
    freopen("dmin.out", "w", stdout);

    citire();
    bellman();
    nr[1]=1;
    for(int i=2;i<=n;i++)
        if(!nr[i])
            nr[i]=calcul(i);
    afisare();
    return 0;
}