oj.uz

답안 #453283

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
453283 2021-08-04T10:31:12 Z Vladth11 Cities (BOI16_cities) C++14
100 / 100
 2739 ms 45484 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define debug(x) cerr << #x << " " << x << "\n"
#pragma GCC optimize("Ofast,unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define debugs(x) cerr << #x << " " << x << " "
//#include"holiday.h"
 
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair <ll, ll> pii;
typedef pair <long double, pii> muchie;
typedef tree <ll, null_type, less_equal <ll>, rb_tree_tag, tree_order_statistics_node_update> OST;
 
const ll NMAX = 100001;
const ll INF = (1LL << 60);
const ll HALF = (1LL << 59);
const ll MOD = 666013;
const ll BLOCK = 225;
const ll base = 31;
const ll nr_of_bits = 21;
 
vector <pii> v[NMAX];
ll n, k, m;
struct ura {
    ll a, b, c;
};
vector <ura> edges;
 
ll scor = 0;
ll viz[NMAX];
 
bool cmp(ura a, ura b) {
    return a.c < b.c;
}
 
void add_Edge(ll a, ll b, ll c) {
    edges.push_back({a, b, c});
}
 
ll d[NMAX][6];
ll dp[(1 << 6)][NMAX];
ll imp[NMAX];
ll special[NMAX];
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k >> m;
    for(ll i = 1; i <= k; i++) {
        ll x;
        cin >> x;
        special[x] = i;
        imp[i - 1] = x;
    }
    for(ll i = 1; i <= m; i++) {
        ll a, b, c;
        cin >> a >> b >> c;
        v[a].push_back({b, c});
        v[b].push_back({a, c});
    }
    for(ll mask = 0; mask < (1 << k); mask++) {
        for(ll i = 0; i <= n; i++)
            dp[mask][i] = HALF;
    }
    for(ll i = 1; i <= k; i++) {
        dp[(1 << (i - 1))][imp[i - 1]] = 0;
    }
    for(ll i = 1; i <= n; i++)
        dp[0][i] = 0;
    for(ll mask = 0; mask < (1 << k); mask++) {
        for(ll i = 1; i <= n; i++) {
            for(ll submask = mask; submask > 0; submask = (submask - 1) & mask) {
                if(special[i]) {
                    if(!(mask & (1 << (special[i] - 1))))
                        continue;
                }
                ll newmask = (mask ^ submask);
                if(special[i]) {
                    newmask ^= (1 << (special[i] - 1));
                }
                dp[mask][i] = min(dp[mask][i], dp[submask][i] + dp[newmask][i]);
            }
 
        }
        priority_queue <pii> pq;
        for(ll i = 0; i <= n; i++) {
            viz[i] = 0;
           
            pq.push({-dp[mask][i], i});
        }
        while(pq.size()) {
            pii node = pq.top();
            ll city = node.second;
            pq.pop();
            if(viz[city])
                continue;
            viz[city] = 1;
            for(auto x : v[city]) {
                int i = x.first;
            
                if(dp[mask][city] + x.second < dp[mask][x.first]) {
                    dp[mask][x.first] = dp[mask][city] + x.second;
                    pq.push({-dp[mask][x.first], x.first});
                }
            }
        }
    }
    ll sol = INF;
    for(ll i = 1; i <= n; i++) {
        sol = min(sol, dp[(1 << k) - 1][i]);
    }
    cout << sol;
    return 0;
}

Compilation message

cities.cpp: In function 'int main()':
cities.cpp:102:21: warning: unused variable 'i' [-Wunused-variable]
  102 |                 int i = x.first;
      |                     ^

Subtask #1
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 2636 KB Output is correct
2 Correct 2 ms 2636 KB Output is correct
3 Correct 2 ms 2636 KB Output is correct
4 Correct 2 ms 2764 KB Output is correct
5 Correct 2 ms 2764 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 764 ms 26572 KB Output is correct
2 Correct 782 ms 26120 KB Output is correct
3 Correct 452 ms 18716 KB Output is correct
4 Correct 83 ms 11972 KB Output is correct
5 Correct 418 ms 23456 KB Output is correct
6 Correct 90 ms 11672 KB Output is correct
7 Correct 6 ms 2892 KB Output is correct
8 Correct 4 ms 2892 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 3020 KB Output is correct
2 Correct 9 ms 3144 KB Output is correct
3 Correct 5 ms 2892 KB Output is correct
4 Correct 5 ms 2892 KB Output is correct

Subtask #4
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1316 ms 32940 KB Output is correct
2 Correct 1280 ms 32536 KB Output is correct
3 Correct 902 ms 24928 KB Output is correct
4 Correct 691 ms 23616 KB Output is correct
5 Correct 190 ms 14544 KB Output is correct
6 Correct 86 ms 14120 KB Output is correct

Subtask #5
26.0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2739 ms 45484 KB Output is correct
2 Correct 2601 ms 45472 KB Output is correct
3 Correct 2594 ms 44972 KB Output is correct
4 Correct 1694 ms 37392 KB Output is correct
5 Correct 1367 ms 30144 KB Output is correct
6 Correct 291 ms 16020 KB Output is correct
7 Correct 104 ms 14596 KB Output is correct