結果

問題 No.1602 With Animals into Institute 2
コンテスト
ユーザー hiikunZ
提出日時 2022-07-18 17:24:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 3,449 bytes
コンパイル時間 2,375 ms
コンパイル使用メモリ 218,048 KB
最終ジャッジ日時 2025-01-30 10:56:03
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 TLE * 11 MLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

//情報科学の達人で履修したので、解説ACだけどACしてみたい
//解説の写経です
#include <bits/stdc++.h>
using namespace std;
//#include<atcoder/all>
//using namespace atcoder;
using ll = long long int;
using ull = unsigned long long int;
using ld = long double;
constexpr ll MAX = 2000000000000000000;
constexpr ld PI = 3.14159265358979;
constexpr ll MOD = 0;//2024948111;
ld dotorad(ld K){return PI * K / 180.0;}
ld radtodo(ld K){return K * 180.0 / PI;}
mt19937 mt;
void randinit(){srand((unsigned)time(NULL));mt = mt19937(rand());}
int main(){
    ll N,M,K;
    cin >> N >> M >> K;
    vector<vector<ll>> G(N),C(N),D(N);
    for(ll i = 0;i < M;i++){
        ll a,b,c;
        string x;
        cin >> a >> b >> c >> x;
        a--;
        b--;
        ll d = 0;
        for(ll j = 0;j < K;j++){
            d *= 2;
            d += (x[j] == '1');
        }
        G[a].emplace_back(b);
        G[b].emplace_back(a);
        C[a].emplace_back(c);
        C[b].emplace_back(c);
        D[a].emplace_back(d);
        D[b].emplace_back(d);
    }
    vector<ll> dis(N,-1),lav(N,-1),par(N,-1),depth(N,-1);
    {
        //dis,lav,idx,par,depth
        priority_queue<tuple<ll,ll,ll,ll,ll>,vector<tuple<ll,ll,ll,ll,ll>>,greater<tuple<ll,ll,ll,ll,ll>>> que;
        que.emplace(0,0,N - 1,-1,0);
        while(!que.empty()){
            ll d = get<0>(que.top());
            ll l = get<1>(que.top());
            ll a = get<2>(que.top());
            ll p = get<3>(que.top());
            ll dep = get<4>(que.top());
            que.pop();
            if(dis[a] == -1){
                dis[a] = d;
                lav[a] = l;
                par[a] = p;
                depth[a] = dep;
                for(ll i = 0;i < (ll)G[a].size();i++){
                    que.emplace(d + C[a][i],l ^ D[a][i],G[a][i],a,dep + 1);
                }
            }
        }
    }
    vector<ll> root(N),ans(N,MAX);
    for(ll i = 0;i < N;i++){
        root[i] = i;
    }
    priority_queue<tuple<ll,ll,ll>,vector<tuple<ll,ll,ll>>,greater<tuple<ll,ll,ll>>> que;
    for(ll i = 0;i < N;i++){
        for(ll j = 0;j < (ll)G[i].size();j++){
            ll u = i,v = G[i][j];
            ll l = lav[u] ^ lav[v] ^ D[i][j];
            if(l != 0){
                ll h = dis[u] + dis[v] + C[i][j];
                que.emplace(h,u,v);
            }
        }
    }
    while(!que.empty()){
        ll h = get<0>(que.top());
        ll u = get<1>(que.top());
        ll v = get<2>(que.top());
        que.pop();
        u = root[u];
        v = root[v];
        vector<ll> B;
        while(u != v){
            if(depth[u] > depth[v]){
                B.emplace_back(u);
                u = root[par[u]];
            }
            else{
                B.emplace_back(v);
                v = root[par[v]];
            }
        }
        for(ll i = 0;i < (ll)B.size();i++){
            ll x = B[i];
            root[x] = u;
            ans[x] = h - dis[x];
            for(ll j = 0;j < (ll)G[x].size();j++){
                ll p = x,q = G[x][j];
                ll l = lav[p] ^ lav[q] ^ D[x][j];
                if(l == 0){
                    ll h = ans[p] + dis[q] + C[x][j];
                    que.emplace(h,p,q);
                }
            }
        }
    }
    for(ll i = 0;i < N;i++) if(lav[i] != 0) ans[i] = dis[i];
    for(ll i = 0;i < N - 1;i++){
        if(ans[i] == MAX) cout << -1 << endl;
        else cout << ans[i] << endl;
    }
}
0