采用经过堆优化的Dijkstra算法+优先级队列只能拿到60分,运行超时。
解题思路:
如果对所有顶点使用Dijkstra,复杂度无疑要上升至n*m*log(m)级别。如果我们只对每个发动机节点使用Dijkstra,复杂度可以下降到k*m*log(m)级别,就可以通过子任务中百分之60的数据。
建立一个n级别大小的优先级队列数组,每个优先级队列按照编号存放到发动机据点的最短距离,这样经过k次Dijkstra算法之后,每个优先级队列将从小到大存放到发动机据点的所有距离,按照题意输出即可。
#include <bits/stdc++.h>
#define debug puts("-----")
#define pi (acos(-1.0))
#define eps (1e-8)
#define inf (1<<30)
using namespace std;
const int maxn = 10001;
int n, m, k;
int d[maxn], w[maxn];
//用于创建邻接表的节点
struct Node
{
int w, v;
};
vector<Node> G[maxn];
//堆优化使用的重载括号运算符
struct cmp
{
bool operator() (const int& a, const int& b) {
return d[a] > d[b];
}
};
//用于存放