結果

問題 No.3561 Collect KCPC
コンテスト
ユーザー GOTKAKO
提出日時 2026-05-29 22:36:15
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 2,643 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,845 ms
コンパイル使用メモリ 231,208 KB
実行使用メモリ 47,364 KB
最終ジャッジ日時 2026-05-29 22:37:21
合計ジャッジ時間 63,709 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
サブタスク 配点 結果
部分点1 10 % AC * 6 WA * 9
部分点2 20 % AC * 9 WA * 6
部分点3 20 % AC * 6 WA * 7
部分点4 50 % AC * 18 WA * 32 TLE * 1
合計 0 点
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

vector<long long> dijkstra(vector<vector<pair<int,long long>>> &Graph,int start){
int N = Graph.size();
//O((V+E)logV) 一般最短経路魔法.
long long inf = 3e18;
    vector<bool> used(N);
    vector<long long> ret(N,inf);
    priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<pair<long long,int>>> Q;
    ret.at(start) = 0; Q.push({0,start});
    while(Q.size()){
        auto[nowd,pos] = Q.top(); Q.pop();
        if(used.at(pos)) continue;
        used.at(pos) = true;
        for(auto [to,w] : Graph.at(pos)){
            if(ret.at(to) > nowd+w){
                ret.at(to) = nowd+w;
                Q.push({ret.at(to),to});
            }
        }
    }
    return ret;
}
vector<long long> dijkstra2(vector<vector<pair<int,long long>>> &Graph,int start){
    int N = Graph.size();
    //O(V^2) 密グラフ専用.
    long long inf = 3e18;
    vector<bool> used(N);
    vector<long long> ret(N,inf);
    ret.at(start) = 0;
    while(true){
        long long nowd = inf,pos = -1;
        for(int i=0; i<N; i++){
            if(used.at(i)) continue;
            if(nowd > ret.at(i)) nowd = ret.at(i),pos = i;
        }
        if(pos == -1) break;
        used.at(pos) = true;
 
        for(auto [to,w] : Graph.at(pos)){if(ret.at(to) > nowd+w) ret.at(to) = nowd+w;}
    }
    return ret;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int N,M; cin >> N >> M;
    vector<tuple<int,int,int>> edge(M);
    for(auto &[u,v,c] : edge) cin >> u >> v >> c,u--,v--;
    string s; cin >> s;
    long long answer = 1e18;
    vector<vector<pair<int,long long>>> Graph(5*N);
    for(auto [u,v,c] : edge) for(int i=0; i<5; i++) Graph.at(u*5+i).push_back({v*5+i,c});
    for(int i=0; i<N; i++){
        if(s.at(i) == 'K') Graph.at(i*5).push_back({i*5+1,0});
        if(s.at(i) == 'P') Graph.at(i*5+2).push_back({i*5+3,0});
    }
    for(int t=0; t<17; t++){
        vector<bool> one(N);
        for(int i=0; i<N; i++) one.at(i) = (i>>t)&1;
        for(int i=0; i<N; i++) if(s.at(i) == 'C') Graph.at(i*5+(one.at(i)?3:1)).push_back({i*5+(one.at(i)?4:2),0});    
        answer = min(answer,dijkstra(Graph,0).back());
        for(int i=0; i<N; i++) if(s.at(i) == 'C') Graph.at(i*5+(one.at(i)?3:1)).pop_back();

        for(int i=0; i<N; i++) if(s.at(i) == 'C') Graph.at(i*5+(one.at(i)?1:3)).push_back({i*5+(one.at(i)?2:4),0});    
        answer = min(answer,dijkstra(Graph,0).back());
        for(int i=0; i<N; i++) if(s.at(i) == 'C') Graph.at(i*5+(one.at(i)?1:3)).pop_back();
    }
    if(answer > 1e17) answer = -1; 
    cout << answer << endl;
}
0