oj.uz

답안 #1059606

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1059606 2024-08-15T06:07:31 Z vjudge1 Cities (BOI16_cities) C++17
74 / 100
 6000 ms 172856 KB 
#include <bits/stdc++.h>

using namespace std;

//#define int long long
#define ll long long
#define ii pair<int, int>
#define tp tuple<ll, int, int>
#define fi first
#define se second
#define all(x) x.begin(), x.end()

const int N = 2e5 + 5,
          mod = 1e9 + 7;
const ll INF = 2e16;

// code

int n, k, m, a[6];
ll dp[N][33];
vector<ii> ad[N];
priority_queue<tp, vector<tp>, greater<>> pq;

int id(int i){
  for(int j = 0; j < k; j++) if(a[j] == i) return j;
  return -1;
}

bool check(int msk1, int msk2){
  for(int i = 0; i < (1 << k); i++) if((msk1 >> i & 1) && (msk2 >> i & 1)) return false;
  return true;
}

int32_t main(){
  cin.tie(0)->sync_with_stdio(0);
  #define task "test"
  if(fopen(task ".inp", "r")){
    freopen(task ".inp", "r", stdin);
    freopen(task ".out", "w", stdout);
  }

  cin >> n >> k >> m;

  for(int i = 0; i < k; i++) cin >> a[i];
  for(int i = 0; i < m; i++){
    int a, b, c; cin >> a >> b >> c;
    ad[a].push_back({b, c});
    ad[b].push_back({a, c});
  }

  for(int i = 1; i <= n; i++)
    for(int j = 1; j < (1 << k); j++) dp[i][j] = INF;

  for(int i = 0; i < k; i++){
    dp[a[i]][1 << i] = 0;
    pq.push({0, 1 << i, a[i]});
  }

  while(pq.size()){
    auto [d, msk, u] = pq.top(); pq.pop();

    if(d != dp[u][msk]) continue;

//    cerr << u << " " << d << " " << msk << endl;

    for(auto [v, c] : ad[u]){
      int bit = id(v);

      if(bit != -1 && !(msk >> bit & 1))
        if(d + c < dp[v][msk|(1 << bit)])
      pq.push({dp[v][msk|(1 << bit)] = d + c, msk|(1 << bit), v});

      for(int _msk = 0; _msk < (1 << k); _msk++)
        if(check(msk, _msk) && d + c + dp[v][_msk] < dp[v][msk|_msk])
          pq.push({dp[v][msk|_msk] = d + c + dp[v][_msk], msk|_msk, v});
    }
  }

  ll res = INF;
  for(int i = 1; i <= n; i++){
    res = min(res, dp[i][(1 << k)-1]);
//    cerr << dp[i][(1 << k)-1] << endl;
  }
  cout << res << endl;

  return 0;
}

Compilation message

cities.cpp: In function 'int32_t main()':
cities.cpp:38:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   38 |     freopen(task ".inp", "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
cities.cpp:39:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   39 |     freopen(task ".out", "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
22.0 / 22.0

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

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 722 ms 57760 KB Output is correct
2 Correct 640 ms 57380 KB Output is correct
3 Correct 276 ms 40380 KB Output is correct
4 Correct 120 ms 14292 KB Output is correct
5 Correct 227 ms 45516 KB Output is correct
6 Correct 53 ms 13752 KB Output is correct
7 Correct 5 ms 5724 KB Output is correct
8 Correct 3 ms 5468 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 14 ms 6120 KB Output is correct
2 Correct 12 ms 6060 KB Output is correct
3 Correct 8 ms 5724 KB Output is correct
4 Correct 13 ms 5848 KB Output is correct

Subtask #4
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2568 ms 107208 KB Output is correct
2 Correct 2273 ms 106880 KB Output is correct
3 Correct 1095 ms 52664 KB Output is correct
4 Correct 1524 ms 60824 KB Output is correct
5 Correct 668 ms 32692 KB Output is correct
6 Correct 518 ms 17112 KB Output is correct

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6070 ms 172856 KB Time limit exceeded
2 Halted 0 ms 0 KB -