多组数据不清零------见祖宗
「3.3」虫洞 Wormholes
问题背景
「一本通3.3 练习2」
题目描述
John 在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边,并可以使你返回到过去的一个时刻(相对你进入虫洞之前)。John 的每个农场有 M 条小路(无向边)连接着 N(从 1 到 N 标号)块地,并有 W 个虫洞。
现在 John 想借助这些虫洞来回到过去(在出发时刻之前回到出发点),请你告诉他能办到吗。 John 将向你提供 F 个农场的地图。没有小路会耗费你超过 10^4 秒的时间,当然也没有虫洞回帮你回到超过 10^4 秒以前。
输入格式
第一行一个整数 F,表示农场个数;
对于每个农场:
第一行,三个整数 N,M,W;
接下来 M 行,每行三个数 S,E,T,表示在标号为 S 的地与标号为 E 的地中间有一条用时 T 秒的小路;
接下来 W 行,每行三个数 S,E,T,表示在标号为 S 的地与标号为 E 的地中间有一条可以使 John 到达 T 秒前的虫洞。
输出格式
输出共 F 行,如果 John 能在第 i 个农场实现他的目标,就在第 i 行输出 YES,否则输出 NO。
样例输入1
2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8
样例输出1
NO
YES
注释说明
对于全部数据,1≤F≤5, 1≤N≤500, 1≤M≤2500, 1≤W≤200,1≤S,E≤N, ∣T∣≤10^4。
cpp
#include<bits/stdc++.h>
using namespace std;
int n,m,dis[100005],a,b,c,huan[100005],w,t;
bool bl[100005];
struct ed {
int to,w;
};
vector<ed>e[100005];
void spfa(int s){
deque<int>q;
memset(dis,0x3f,sizeof(dis));
memset(bl,0,sizeof(bl));
memset(huan,0,sizeof(huan));
q.push_back(s);
bl[s]=1;huan[s]++;
dis[s]=0;
while(!q.empty()) {
int k=q.front();
q.pop_front();
bl[k]=0;
int o;
for(int i=0; i<e[k].size(); i++){
o=e[k][i].to;
if(e[k][i].w+dis[k]<dis[o]){
dis[o]=e[k][i].w+dis[k];
if(bl[o]==0){
if(q.empty()||dis[o]<q.front())q.push_front(o);
else q.push_back(o);
bl[o]=1;
huan[o]++;
if(huan[o]>n){
puts("YES");
return;
}
}
}
}
}
puts("NO");
}
int main() {
scanf("%d",&t);
while(t--) {
scanf("%d%d%d",&n,&m,&w);
for (int i = 0; i <= 501; i++) e[i].clear();
for(int i=1; i<=m; i++) {
scanf("%d%d%d",&a,&b,&c);
e[a].push_back((ed){b,c});
e[b].push_back((ed){a,c});
}
for(int i=1; i<=w; i++) {
scanf("%d%d%d",&a,&b,&c);
e[a].push_back((ed){b,-c});
}
for (int i=1;i<=n;i++)e[0].push_back((ed){i,0});
spfa(0);
}
}
/*
2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8
NO
YES
*/