Pagini recente » Kcity | Istoria paginii utilizator/schumi | poarta2 | Monitorul de evaluare | Cod sursa (job #2576540)
#include<fstream>
#include<vector>
#include<set>
#include<cstring>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int T,n,m,root,x,y,c;
int D[50010],bronzarel[50010];
char numar[50];
vector < pair<int,int> > L[50010];
multiset < pair<int,int> > s;
int read(){
fin>>numar;
int val=0;
int n=strlen(numar);
for(int i=0;i<n;i++)
val=val*10+(numar[i]-'0');
return val;
}
int main(){
fin>>T;
for(int t=1;t<=T;t++){
s.clear();
n=read();
for(int i=1;i<=n;i++){
if(!L[i].empty())
L[i].clear();
}
m=read();
root=read();
for(int i=1;i<=n;i++){
bronzarel[i]=read();
D[i]=1000000000;
}
for(int i=1;i<=m;i++){
x=read();
y=read();
c=read();
L[x].push_back( make_pair(y,c) );
L[y].push_back( make_pair(x,c) );
}
D[root]=0;
s.insert( make_pair(0,root) );
int ok=0;
while(!s.empty()){
int nod=s.begin()->second;
if(D[nod]<bronzarel[nod]){
ok=1;
break;
}
s.erase( s.begin() );
for(int i=0;i<L[nod].size();i++){
int vecin=L[nod][i].first;
int cost=L[nod][i].second;
if(D[vecin]>D[nod]+cost){
s.erase({D[vecin],vecin});
D[vecin]=D[nod]+cost;
s.insert({D[vecin],vecin});
}
}
}
if(ok){
fout<<"NU"<<"\n";
continue;
}
for(int i=1;i<=n;i++){
if(D[i]==1000000000)
D[i]=0;
if(D[i]!=bronzarel[i]){
fout<<"NU"<<"\n";
ok=1;
break;
}
}
if(!ok)
fout<<"DA"<<"\n";
}
return 0;
}