結果

問題 No.1690 Power Grid
ユーザー deuteridayodeuteridayo
提出日時 2021-09-24 23:19:26
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,801 ms / 3,000 ms
コード長 3,165 bytes
コンパイル時間 3,741 ms
コンパイル使用メモリ 237,984 KB
実行使用メモリ 81,296 KB
最終ジャッジ日時 2024-07-05 11:32:42
合計ジャッジ時間 17,303 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 27 ms
42,112 KB
testcase_01 AC 28 ms
42,112 KB
testcase_02 AC 30 ms
42,112 KB
testcase_03 AC 29 ms
42,112 KB
testcase_04 AC 30 ms
42,112 KB
testcase_05 AC 29 ms
42,112 KB
testcase_06 AC 258 ms
62,748 KB
testcase_07 AC 255 ms
62,860 KB
testcase_08 AC 250 ms
62,868 KB
testcase_09 AC 261 ms
62,824 KB
testcase_10 AC 170 ms
48,396 KB
testcase_11 AC 1,586 ms
79,248 KB
testcase_12 AC 29 ms
42,112 KB
testcase_13 AC 28 ms
42,112 KB
testcase_14 AC 51 ms
43,108 KB
testcase_15 AC 1,801 ms
81,296 KB
testcase_16 AC 260 ms
62,864 KB
testcase_17 AC 134 ms
51,588 KB
testcase_18 AC 69 ms
45,456 KB
testcase_19 AC 254 ms
62,860 KB
testcase_20 AC 259 ms
62,864 KB
testcase_21 AC 167 ms
46,480 KB
testcase_22 AC 793 ms
71,060 KB
testcase_23 AC 1,772 ms
81,296 KB
testcase_24 AC 1,178 ms
75,148 KB
testcase_25 AC 1,371 ms
77,204 KB
testcase_26 AC 1,591 ms
79,244 KB
testcase_27 AC 28 ms
42,112 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using lint = long long;
using ulint = unsigned long long;
#define endl '\n'
int const INF = 1<<30;
lint const INF64 = 1LL<<61;
lint const mod = 1e9+7;
//long const mod = 998244353;
lint ceilDiv(lint x, lint y){if(x >= 0){return (x+y-1)/y;}else{return x/y;}}
lint floorDiv(lint x, lint y){if(x >= 0){return x/y;}else{return (x-y+1)/y;}}
lint Sqrt(lint x){lint upper = 1e9;lint lower = 0;while(upper - lower > 0){lint mid = (1+upper + lower)/2;if(mid * mid > x){upper = mid-1;}else{lower = mid;}}return upper;}
lint gcd(lint a,lint b){if(a<b)swap(a,b);if(a%b==0)return b;else return gcd(b,a%b);}
lint lcm(lint a,lint b){return (a / gcd(a,b)) * b;}
lint chmin(vector<lint>&v){lint ans = INF64;for(lint i:v){ans = min(ans, i);}return ans;}
lint chmax(vector<lint>&v){lint ans = -INF64;for(lint i:v){ans = max(ans, i);}return ans;}
double dist(double x1, double y1, double x2, double y2){return sqrt(pow(x1-x2, 2) + pow(y1-y2,2));}
string toString(lint n){string ans = "";if(n == 0){ans += "0";}else{while(n > 0){int a = n%10;char b = '0' + a;string c = "";c += b;n /= 10;ans = c + ans;}}return ans;}
string toString(lint n, lint k){string ans = toString(n);string tmp = "";while(ans.length() + tmp.length() < k){tmp += "0";}return tmp + ans;}
vector<lint>prime;void makePrime(lint n){prime.push_back(2);for(lint i=3;i<=n;i+=2){bool chk = true;for(lint j=0;j<prime.size() && prime[j]*prime[j] <= i;j++){if(i % prime[j]==0){chk=false;break;}}if(chk)prime.push_back(i);}}
struct edge{
    int to;
    lint len;
};
using graph = vector<vector<edge>>;
vector<vector<lint>>cost(18, vector<lint>(1<<18, INF64));
vector<vector<lint>>d(18,vector<lint>(18, INF64));
int n;
lint f(int k, lint bit, vector<vector<lint>>&dp) {
    // 
    if(k == 0)return 0;
    if(dp[k][bit] != -1)return dp[k][bit];
    lint ans = INF64;
    for(int i  =0;i<n;i++) {
        if(bit & (1<<n))continue;
        ans = min(ans, cost[i][bit] + f(k-1, bit | (1<<i), dp));
    }
    // cerr << k <<  " " << bit <<" " << ans<< endl;
    return dp[k][bit] = ans;
}

int main(){ 
    int m,k;
    cin >> n >> m >> k;
    graph g(n);
    lint a[n];
    for(int i = 0;i<n;i++)cin >> a[i];
    for(int i = 0;i<m;i++) {
        int x,y;
        lint z;
        cin >> x >> y >> z;
        x--;y--;
        d[x][y] = z;
        d[y][x] = z;
    }    
    for(int k=0;k<n;k++){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++) {
                d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
            }
        }
    }
    for(int i= 0;i<(1<<n);i++){
        for(int j = 0;j<n;j++) {
            if(i & (1<<j)){
                cost[j][i] = INF64;
                continue;
            }
            if(i == 0){
                cost[j][i] = a[j];
                continue;
            }
            lint tmp = INF64;
            for(int k = 0;k<n;k++ ) {
                
                if(i &(1<<k))tmp = min(tmp, d[j][k]);
            }
            cost[j][i] = tmp + a[j];
        }
    }


    vector<vector<lint>>dp(k+1,vector<lint>(1<<n, -1));
    dp[0][0] = 0;

    cout << f(k,0, dp) << endl;


}

0