Cod sursa(job #3287478)

Utilizator InfinitumDanila Laurentiu Infinitum Data 18 martie 2025 13:07:58
Problema Drumuri minime Scor 5
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("dmin.in");
ofstream g("dmin.out");
const int INF = 1e9;
bitset<1500> viz;
int main()
{
    ios_base::sync_with_stdio(false);
    f.tie(nullptr);

    int n, m;
    f >> n >> m;
    vector<vector<pair<int,int>>> v(n+1, vector<pair<int,int>>());
    vector<int> d(n+1, INF);
    vector<int> cnt(n+1,1);
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
    for(int i=1; i<=m; i++)
    {
        int x, y, c;
        f >> x >> y >> c;
        v[x].push_back({y,c});
        v[y].push_back({x,c});
    }
    pq.push({1,1});
    d[1]=1;
    while(!pq.empty())
    {
        int dist, nod;
        tie(dist,nod) = pq.top();
        pq.pop();

            for(auto vc : v[nod])
            {
                if(d[vc.first]>d[nod]*vc.second)
                {
                    d[vc.first]=d[nod]*vc.second;
                    pq.push({d[vc.first],vc.first});
                    cnt[vc.first]=1;
                }else if(d[vc.first]==d[nod]*vc.second)
                {
                    cnt[vc.first]++;
                    pq.push({d[vc.first],vc.first});
                }
            }
    }
    for(int i=2; i<=n; i++)
        g << cnt[i] << " ";

}