Cod sursa(job #2829400)

Utilizator bogdan2405Strat Bogdan-Valentin bogdan2405 Data 8 ianuarie 2022 16:20:29
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.02 kb
#include<bits/stdc++.h>

using namespace std;

ifstream f("distante.in");
ofstream g("distante.out");

struct CustomCompare{
    bool operator()(const pair<int,int> &a, const pair<int,int> &b){
        return a.second>b.second;
    }
};
 
vector<pair<int,int>> adjList[50001];
vector<int> isMarked,length;
priority_queue<pair<int,int>,vector<pair<int,int>>,CustomCompare>heap;
 
void Dijkstra(int x){
    int i;
    isMarked[x]++;
    for(i=0;i<adjList[x].size();++i){
        if(isMarked[adjList[x][i].first]==0){
            pair<int,int> aux;
            //aux.push_back(x);
            //aux.push_back(adjList[x][i].first);
            //aux.push_back(adjList[x][i].second+length[x]);
            heap.push(make_pair(adjList[x][i].first,adjList[x][i].second+length[x])); 
            //aux.clear();
        }
        
    }
 
    pair<int,int> frt;
    while(heap.size()>0){
        frt=heap.top();
        heap.pop();
        if(/*isMarked[first[0]]==1 &&*/ isMarked[frt.first]==0){
            length[frt.first]=frt.second;
            Dijkstra(frt.first);
        }
    }
}

int main(){
    int t,a,b,c;
    int n,m,s;
    vector<string> ans;
    vector<int> zaharel,res;
    f>>t;
    int i,j,k;
    for(i=0;i<t;++i){
        length.clear();
        isMarked.clear();

        while(!heap.empty())
            heap.pop();

        for(j=0;j<n;++j){
        isMarked.push_back(0);
        length.push_back(0);
        }

        f>>n>>m>>s;
        zaharel.clear();

        for(j=0;j<n;++j)
        {
            f>>a;
            zaharel.push_back(a);
        }

        for(j=0;j<m;++j){
            f>>a>>b>>c;
            adjList[a-1].push_back(make_pair(b-1,c));
            adjList[b-1].push_back(make_pair(a-1,c));
        }
        
        Dijkstra(s-1);
        

        if(zaharel==length){
            ans.push_back("DA");
        }
        else{
            ans.push_back("NU");
        }

        for(j=0;j<n;++j){
            adjList[i].clear();
        }
    }

    for(i=0;i<ans.size();++i){
        g<<ans[i]<<'\n';
    }

    return 0;
}