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;
}
全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务