|
100鱼币
本帖最后由 798236606 于 2020-2-22 17:20 编辑
题目:https://pintia.cn/problem-sets/9 ... /994805523835109376
之前用迪杰斯特拉算法和BF算法成功通过,但是BF算法的优化算法SPFA算法总是有两个点没通过。
看了一下午了,不知道哪里有问题
C++代码:
- #include <cstdio>
- #include <climits>
- #include <cstring>
- #include <vector>
- #include <queue>
- #include <set>
- #include <algorithm>
- using namespace std;
- const int maxn = 510;
- typedef struct node{
- int id, dis;
- node (int _id, int _dis) : id(_id), dis(_dis) {};
- }Node;
- int n, m, c1, c2;
- int weight[maxn];//存放点权,即救援队数
- int num[maxn];//存放到每个点的最短路径数
- int w[maxn];//存放可聚集的队伍数
- int d[maxn];//存放最短路径长度
- bool inq[maxn];//是否在队列内
- vector<Node> adj[maxn];//邻接表
- set<int> pre[maxn];//存放顶点在最短路径中的前驱
- void SPFA(int s)
- {
- fill(d, d + n, INT_MAX);
- memset(w, 0, sizeof(w));
- memset(num, 0, sizeof(num));
- memset(inq, false, sizeof(inq));
- queue<int> q;
- q.push(s);
- w[s] = weight[s];
- d[s] = 0;
- num[s] = 1;
- inq[s] = true;
- while (!q.empty())
- {
- int u = q.front();
- q.pop();
- inq[u] = false;
- for (int v = 0; v < adj[u].size(); v++)//遍历u连接的每一条边
- {
- int id = adj[u][v].id, dis = adj[u][v].dis;
- if (dis + d[u] < d[id])//如果以u为前驱路径更短
- {
- d[id] = dis + d[u];
- num[id] = num[u];
- w[id] = weight[id] + w[u];
- pre[id].clear();
- pre[id].insert(u);
- if (!inq[id])//如果id不在队列中
- {
- q.push(id);
- inq[id] = true;
- }
- }
- else if (dis + d[u] == d[id])//如果存在等价路径
- {
- if (weight[id] + w[u] > w[id])
- w[id] = weight[id] + w[u];
- num[id] = 0;
-
- pre[id].insert(u);
- for (set<int>::iterator it = pre[id].begin(); it != pre[id].end(); ++it)
- num[id] += num[*it];
- }
- }
- }
- }
- int main(void)
- {
- // freopen("input.txt", "r", stdin);
- scanf("%d %d %d %d", &n, &m, &c1, &c2);
- for (int i = 0; i < n; i++)
- scanf("%d", weight + i);
-
- while (m--)
- {
- int a, b, l;
- scanf("%d %d %d", &a, &b, &l);
- adj[a].push_back(node(b, l));
- adj[b].push_back(node(a, l));
- }
- SPFA(c1);
- printf("%d %d", num[c2], w[c2]);
- return 0;
- }
复制代码 |
|