結果

問題 No.3561 Collect KCPC
コンテスト
ユーザー GOTKAKO
提出日時 2026-05-29 22:49:50
言語 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
結果
AC  
実行時間 5,019 ms / 6,000 ms
コード長 2,742 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,301 ms
コンパイル使用メモリ 232,036 KB
実行使用メモリ 46,660 KB
最終ジャッジ日時 2026-05-29 22:50:49
合計ジャッジ時間 54,443 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge4_0
純コード判定待ち
このコードへのチャレンジ
（要ログイン）
サブタスク 配点 結果
部分点1 10 % AC * 15
部分点2 20 % AC * 15
部分点3 20 % AC * 13
部分点4 50 % AC * 51
合計 100 点
権限があれば一括ダウンロードができます

ソースコード

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<4; 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});    
        auto dist = dijkstra(Graph,0);
        for(int i=0; i<N; i++) answer = min(answer,dist.at(i*5+4));
        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});    
        dist = dijkstra(Graph,0);
        for(int i=0; i<N; i++) answer = min(answer,dist.at(i*5+4));
        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