Tokitsukaze and Slash Draw - 最短路

题面

分析

每一种操作可以认为当前位置移动 a a a 个位置到达之后的位置,花费为 b b b,也就是可以理解为从 i i i ( i + a ) m o d n (i + a) mod n (i+a)modn存在一条边,边权为 b b b,那么就可以进行最短路来计算最小权值。

代码
#include 

using namespace std;
using ll = long long;

void solve() {
    int n, m, k;
    cin >> n >> m >> k;
    vector<vector<pair<int, int>>> adj(n);
    while(m --) {
        int a, b;
        cin >> a >> b;
        for(int i = 0; i < n; i ++) adj[i].push_back({(i + a) % n, b});
    }
    if(k == n) {
        cout << "0\n";
        return ;
    }
    vector<ll> dis(n, 1e17);
    vector<int> st(n, 0);
    auto dijkstra = [&](int k) {
        priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> heap;
        heap.push({0, k});
        dis[k] = 0;
        while(heap.size()) {
            auto t = heap.top();
            heap.pop();
            if(st[t.second]) continue;
            st[t.second] = 1;
            for(auto j: adj[t.second]) {
                if(dis[j.first] > dis[t.second] + j.second) {
                    dis[j.first] = dis[t.second] + j.second;
                    heap.push({dis[j.first], j.first});
                }
            }
        }
    };
    dijkstra(k);
    if(dis[0] == 1e17) cout << "-1\n";
    else cout << dis[0] << "\n";
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int T;
    cin >> T;
    while(T --) {
        solve();
    }
}

你可能感兴趣的:(算法,图论,c++)