当前位置: 代码迷 >> 综合 >> POJ 3259 - Wormholes
  详细解决方案

POJ 3259 - Wormholes

热度:92   发布时间:2023-12-24 11:20:38.0

题目大意:有一个农夫有F个农场,每个农场里有N个农田,农田之间有M条道路,每条道路都是双向的,每条道路信息包括起点,终点,从起点到终点需要的时间。有W个虫洞,进入虫洞所在的农田就会穿越到指定的农田,并且时间回到x秒以前。农夫是个狂热的时空旅行迷,他想知道他能否借助虫洞碰到他自己。

解题思路:floyd是可以ac的,但是稍不限制就会超时。用bellman ford的算法,任意一个起点出发,虫洞就当作负权值的道路,问题就变成了是否存在负环了,注意M条道路双向的,而虫洞是单向的,数组一次输入包括了道路和虫洞,所以要开大一些。

ac代码:

#include <iostream>
#define INF 5000005
using namespace std;
int F, N, M, W, u[3005], v[3005];
int w[3005], dis[505], t1, t2, flag;
int main()
{scanf("%d", &F);while (F--){scanf("%d%d%d", &N, &M, &W);flag = 0;for (int i=1; i<=N; i++)dis[i] = INF;for (int i=0; i<M+W; i++){scanf("%d%d%d", &u[i], &v[i], &w[i]);if (i >= M)w[i] = -w[i];}dis[1] = 0;for (int i=0; i<N-1; i++){for (int j=0; j<M+W; j++){t1 = u[j], t2 = v[j];if (j < M && dis[t1] > dis[t2] + w[j])dis[t1] = dis[t2] + w[j];if (dis[t2] > dis[t1] + w[j])dis[t2] = dis[t1] + w[j];}}for (int i=0; i<M+W; i++){t1 = u[i], t2 = v[i];if (i < M && dis[t1] > dis[t2] + w[i])flag = 1;if (dis[t2] > dis[t1] + w[i])flag = 1;}if (flag)printf("YES\n");elseprintf("NO\n");}
return 0;
}