Pagini recente » Cod sursa (job #638237) | Cod sursa (job #1587994) | Cod sursa (job #3190236) | Cod sursa (job #2110201) | Cod sursa (job #566726)
Cod sursa(job #566726)
#include<stdio.h>
#include<queue>
#include<vector>
#include<algorithm>
#include<string.h>
using namespace std;
const int N=50715;
const int INF=1000000000;
struct succesor
{
int nod,cost;
};
vector<succesor> a[N];
priority_queue<pair<int,int> > h;
int n,d[N],x[N],t;
bool s[N];
void citire()
{
int m,i,x,y,c;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&c);
a[x].push_back((succesor){y,c});
}
}
void actualizare(int x)
{
int c,y;
for(size_t i=0 ; i<a[x].size() ; i++)
{
y = a[x][i].nod;
c = a[x][i].cost;
if(d[x] + c < d[y])
{
d[y] = d[x] + c;
h.push(make_pair(-d[y],y));
}
}
}
void init()
{
for(int i=0 ; i<=n ; i++)
d[i] = INF;
}
int vf_cost_min()
{
while(!h.empty())
{
if(!s[h.top().second])
return h.top().second;
h.pop();
}
return 0;
}
void dijkstra(int x0)
{
int x;
d[x0] = 0;
h.push(make_pair(d[x0],x0));
for(int i=1 ; i<n ; i++)
{
x = vf_cost_min();
if(d[x] == INF)
return;
s[x] = true;
actualizare(x);
}
}
int verificare()
{
int i;
for(i=1;i<=n;i++)
{
if(d[i] != x[i])
return 1;
}
return 0;
}
int main()
{
freopen("distante.in","r",stdin);
freopen("distante.out","w",stdout);
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&x0);
for(i=1;i<=n;i++)
scanf("%d",&x[i]);
a.swap(vector<int>());
memset(d,0,sizeof(d));
memset(s,0,sizeof(s));
citire();
dijkstra(x0);
if(verificare()==1) printf("NU");
else
printf("DA");
}
return 0;
}