题意:一场有向图,询问你第k小的路为多长
思路:首先我们知道所有样例的k的最大值的和为250000,这样我们每次只要暴力的预处理出每条路(注意这里要离线处理,不然每次处理50000会爆炸),然后输出即可。
有个坑点就是不能一次将一个点连的所有的边都更新,这样可能会导致mle(因为可能这个点连了所有的点),我们只要控制每次点更新的边数大小,一种方法就是每次更新只加入下一条最小的边和当前边的下一条边即可。
附上代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
struct inst {
int tt;
int sa;
long long w;
inst() {}
inst(int ttt,int ssa, long long ww) {
this->tt = ttt;
this->sa = ssa;
this->w = ww;
}
bool operator <(const inst &A)const {
return A.w < this->w;
}
};
struct ist {
int tt;
int sa;
int w;
ist() {}
ist(int ttt, int ssa, int ww) {
this->tt = ttt;
this->sa = ssa;
this->w = ww;
}
bool operator >(const ist &A)const {
return A.w > this->w;
}
};
priority_queue<inst>qq;
vector<ist>gx[50010];
long long ans[50010];
int query[50010];
int cmp(ist a, ist b) {
return a.w < b.w;
}
int main(void) {
int t;
scanf("%d", &t);
while (t--) {
while (!qq.empty()) qq.pop();
int n, m, q;
scanf("%d%d%d", &n, &m, &q);
for (int i = 1; i <= n; i++) {
gx[i].clear();
}
int yy = 0;
for (int i = 0; i < m; i++) {
int a, b;
int w;
scanf("%d%d%d", &a, &b, &w);
gx[a].push_back(ist(0,b, w));
}
for (int i = 1; i <= n; i++) {
if (gx[i].size() == 0) continue;
sort(gx[i].begin(), gx[i].end(), cmp);
if (gx[i].size()) {
qq.push(inst(0, i, gx[i][0].w));
}
}
int maxx = -1;
for (int i = 0; i < q; i++) {
scanf("%d", &query[i]);
maxx = max(maxx, query[i]);
}
while (!qq.empty()) {
inst a = qq.top();
qq.pop();
ans[yy++] = a.w;
if (a.tt + 1 < gx[a.sa].size()) {
qq.push(inst(a.tt + 1, a.sa, a.w - gx[a.sa][a.tt].w + gx[a.sa][a.tt + 1].w));
}
if (gx[gx[a.sa][a.tt].sa].size() > 0) {
qq.push(inst(0, gx[a.sa][a.tt].sa, a.w + gx[gx[a.sa][a.tt].sa][0].w));
}
if (yy > maxx+1) break;
}
for (int i = 0; i < q; i++) {
printf("%lld\n", ans[query[i] - 1]);
}
}
}