結果

問題 No.3158 Collect Stamps
ユーザー asmin
提出日時 2025-05-23 19:45:14
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 3,642 bytes
コンパイル時間 2,441 ms
コンパイル使用メモリ 223,096 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-05-23 19:45:18
合計ジャッジ時間 3,849 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma region
/*
\(^o^)/
import sys
私は人間です.



吾輩はやれば出来る子である.

    ∩ ∩
   (´・ω・)
   _| ⊃/(___
 / └-(____/
  ̄ ̄ ̄ ̄ ̄ ̄ ̄

 やる気はまだない.

   ⊂⌒/ヽ-、__
 /⊂_/____ /
  ̄ ̄ ̄ ̄ ̄ ̄ ̄
*/
#include<algorithm>
#include<array>
#include<bitset>
#include<cassert>
#include<chrono>
#include<cinttypes>
#include<climits>
#include<cmath>
#include<complex>
#include<cstdio>
#include<cstring>
#include<deque>
#include<functional>
#include<iomanip>
#include<iostream>
#include<iterator>
#include<limits>
#include<map>
#include<numeric>
#include<queue>
#include<random>
#include<set>
#include<sstream>
#include<stack>
#include<string>
#include<tuple>
#include<type_traits>
#include<unordered_map>
#include<unordered_set>
#include<utility>
#include<vector>
using namespace std;
struct Init{Init(){std::cin.tie(0); ios::sync_with_stdio(false); cout << setprecision(20) << fixed;}}init;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
#define all(x) begin((x)), end((x))
#define pb push_back
#define mp make_pair
#define mt make_tuple
#define uq(v) v.erase(unique(begin(v), end(v)), end(v))
#define _overload4(_1,_2,_3,_4,name,...) name
#define _overload3(_1,_2,_3,name,...) name
#define _rep1(n) for(int i=0;i<n;++i)
#define _rep2(i,n) for(int i=0;i<n;++i)
#define _rep3(i,a,b) for(int i=a;i<b;++i)
#define _rep4(i,a,b,c) for(int i=a;i<b;i+=c)
#define rep(...) _overload4(__VA_ARGS__,_rep4,_rep3,_rep2,_rep1)(__VA_ARGS__)
#define _rrep1(n) for(int i=(n)-1;i>=0;i--)
#define _rrep2(i,n) for(int i=(n)-1;i>=0;i--)
#define _rrep3(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define _rrep4(i,a,b,c) for(int i=a+(b-a-1)/c*c;i>=a;i-=c)
#define rrep(...) _overload4(__VA_ARGS__,_rrep4,_rrep3,_rrep2,_rrep1)(__VA_ARGS__)
template<class T> using pq = priority_queue<T>;
template<class T> using pq_g = priority_queue<T, vector<T>, greater<T>>;
template<class T> bool chmax(T &a, const T &b){if(a < b){a = b; return 1; } return 0;}
template<class T> bool chmin(T &a, const T &b){if(a > b){a = b; return 1; } return 0;}
template<class T> auto min(const T& a){ return *min_element(all(a)); }
template<class T> auto max(const T& a){ return *max_element(all(a)); }
constexpr ull INF = (1ULL << 61) + (1ULL << 30);
constexpr int inf = (1 << 30);
constexpr int dx[] = {1, 0, -1, 0, 1, 1, -1, -1};
constexpr int dy[] = {0, 1 ,0, -1, 1, -1, 1, -1};

#pragma endregion



int main(){
    int N, M, K; cin >> N >> M >> K;
    vector<ll> dist1(N, INF);
    pq_g<pair<ll, int>> q1;
    rep(i, K){
        int A; cin >> A;
        --A;
        dist1[A] = 0;
        q1.push(mp(0, A));
    }
    vector<vector<ll>> T(N, vector<ll>(N));
    rep(i, N) rep(j, N) cin >> T[i][j];
    while(!q1.empty()){
        auto [d, v] = q1.top(); q1.pop();
        if(dist1[v] < d) continue;
        rep(i, N){
            if(i == v) continue;
            if(chmin(dist1[i], d + T[i][v])){
                q1.push(mp(dist1[i], i));
            }
        }
    }
    set<int> root;
    ll ans = INF;
    ll sum = 0;
    auto dfs = [&](auto self, int cur, int cnt) -> void {
        if(cnt == M){
            chmin(ans, sum + dist1[cur]);
            return;
        }
        rep(i, N){
            if(root.contains(i)) continue;
            sum += T[cur][i];
            root.insert(i);
            self(self, i, cnt + 1);
            sum -= T[cur][i];
            root.erase(i);
        }
    };
    rep(i, N){
        root.insert(i);
        dfs(dfs, i, 1);
        root.erase(i);
    }
    cout << ans << "\n";

}
0