結果

問題 No.2914 正閉路検出
ユーザー GOTKAKO
提出日時 2024-10-04 22:46:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,277 bytes
コンパイル時間 2,782 ms
コンパイル使用メモリ 220,516 KB
最終ジャッジ日時 2025-02-24 15:31:04
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N,M; cin >> N >> M;
    vector<vector<tuple<int,int,long long>>> Graph(N);
    for(int i=0; i<M; i++){
        int u,v; cin >> u >> v;
        u--; v--;
        int w; cin >> w;
        Graph.at(u).push_back({v,i,-w});
        Graph.at(v).push_back({u,i,w});
    }

    vector<bool> already(N);
    vector<long long> dist(N,1e18);
    priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<>> Q;
    dist.at(0) = 0; Q.push({0,0});
    
    int start = -1;
    while(Q.size()){
        auto [nowd,pos] = Q.top(); Q.pop();
        if(nowd != dist.at(pos)) continue;
        if(already.at(pos)){start = pos; break;}
        already.at(pos) = true;
        
        for(auto [to,epos,w] : Graph.at(pos)){
            if(dist.at(to) > dist.at(pos)+w){
                dist.at(to) = dist.at(pos)+w;
                Q.push({dist.at(to),to});
            }
        }
    }
    if(start == -1){cout << "-1\n"; return 0;}

    dist.assign(N,1e18); already.assign(N,false);
    dist.at(start) = 0;
    while(Q.size()) Q.pop();
    Q.push({0,start});
    vector<pair<int,int>> back(N);
    int start2 = -1;
    vector<int> route;
    while(Q.size()){
        auto [nowd,pos] = Q.top(); Q.pop();
        if(dist.at(pos) != nowd) continue;
        already.at(pos) = true;
        for(auto [to,epos,w] : Graph.at(pos)){
            if(dist.at(to) > nowd+w){
                if(to == start && nowd+w < 0){
                    route.push_back({epos});
                    start2 = pos; break;
                }
                else if(dist.at(to) > nowd+w && already.at(to) == false){
                    dist.at(to) = nowd+w;
                    Q.push({dist.at(to),to});
                    back.at(to) = {pos,epos};
                }
            }
        }
        if(route.size()) break;
    }
    while(start2 != start){
        route.push_back(back.at(start2).second);
        start2 = back.at(start2).first;
    }

    reverse(route.begin(),route.end());

    cout << route.size() << endl;
    cout << start+1 << endl;
    for(int i=0; i<route.size(); i++){
        if(i) cout << " ";
        cout << route.at(i)+1;
    }
    cout << endl;
}
0