hdu6705
题目链接
大意:给你一张有向加权图,让你求出第k小的路径
思路:显然我们从大到小处理到最大的询问就可以了。
先对所有点的出边按权值从小到大排序,然后塞到优先队列中,每次取出队头最小的元素更新答案,但是如果直接遍历点所有出边的话是会tle的,我们需要一个multiset来记录优先队列中的答案,如果multiset的元素等于最大的询问的话,那么我们只要更新到一个出边,使得这个路径长度和大于multiset最大的元素的话,我们就直接break出去,不然就更新multiset。
细节见代码:
#include<bits/stdc++.h>
#define fi first
#define se second
#define LL long long
#define mp make_pair
#define pb push_back
#define pii pair<LL,LL>
#define SZ(x) (int)x.size()
#define all(x) x.begin(),x.end()
using namespace std;
LL gcd(LL a, LL b) {return b ? gcd(b, a % b) : a;}
LL lcm(LL a, LL b) {return a / gcd(a, b) * b;}
LL powmod(LL a, LL b, LL MOD) {LL ans = 1; while (b) {if (b % 2)ans = ans * a % MOD; a = a * a % MOD; b /= 2;} return ans;}
const int N = 2e5 + 11;
int t;
int n, m, Q;
vector<pii>v[N];
struct uzi {
int a, i;
bool operator <(const uzi &t)const {
return a < t.a;
}
} p[N];
LL ans[N];
int cmp(pii l, pii r) {
return l.se < r.se;
}
int an[N];
multiset<LL>ANS;
priority_queue<pii>q;
int main() {
ios::sync_with_stdio(false);
for (cin >> t; t; t--) {
cin >> n >> m >> Q;
for (int i = 1; i <= n; i++)v[i].clear();
ANS.clear();
while (!q.empty())q.pop();
for (int i = 1; i <= m; i++) {
int s, t, w;
cin >> s >> t >> w;
v[s].pb({t, w});
ANS.insert(w);
}
for (int i = 1; i <= n; i++) {
sort(all(v[i]), cmp);
}
int MAX = 0;
for (int i = 1; i <= Q; i++) {
cin >> p[i].a;
MAX = max(MAX, p[i].a);
p[i].i = i;
}
for (int i = 1; i <= n; i++) {
for (auto k : v[i]) {
q.push({ -k.se, k.fi});
}
}
int NOW = 0;
while (NOW < MAX) {
auto x = q.top();
q.pop();
ans[++NOW] = -x.fi ;
if (NOW >= MAX)break;
for (auto k : v[x.se]) {
if (SZ(ANS) >= p[Q].a) {
auto need = --ANS.end();
if (-x.fi + k.se < *need) {
ANS.erase(need);
ANS.insert(-x.fi + k.se);
} else break;
} else {
ANS.insert({ -x.fi + k.se});
}
q.push({ x.fi - k.se, k.fi});
}
}
for (int i = 1; i <= Q; i++) {
cout << ans[p[i].a] << endl;
}
}
return 0;
}