Submission #1179220

#	Time	Username	Problem	Language	Result	Execution time	Memory
1179220		JelalTkm	Cyberland (APIO23_cyberland)	C++20	15 / 100	21 ms	6724 KiB

cyberland

#include <bits/stdc++.h>
#include "cyberland.h"
#pragma GCC optimize ("O3")
#pragma GCC target ("sse4")

using namespace std;

// #define int long long int

// const int N = 2e5 + 10;
// const int md = 1e9 + 7;
// const int INF = 1e18;

double solve(int n, int m, int k, int h, vector<int> x, vector<int> y, vector<int> w, vector<int> a) {
  vector<vector<pair<int, double>>> g(n + 1, vector<pair<int, double>> ());
  for (int i = 0; i < m; i++) {
    g[x[i]].push_back({y[i], w[i]});
    g[y[i]].push_back({x[i], w[i]});
  }

  vector<double> dist(g.size() + 10, -1);
  vector<bool> visited(g.size() + 10);
  set<pair<double, int>> q;
  dist[h] = 0;
  q.insert({0, h});
  double ans = 0;

  while (!q.empty()) {
    int v = (*q.begin()).second;
    if (a[v] == 0) {
      return dist[v];
    }
    if ((*q.begin()).first != dist[v]) {
      q.erase(q.begin());
      continue;
    }
    q.erase(q.begin());
    for (auto i : g[v]) {
      if (dist[i.first] == -1) {
        dist[i.first] = dist[v] + i.second;
        q.insert({dist[i.first], i.first});
      }
      else if (dist[i.first] > (dist[v] + i.second)) {
        dist[i.first] = dist[v] + i.second;
        q.insert({dist[i.first], i.first});
      }
    }
  }

  return dist[0];
}

// int32_t main(int32_t argc, char *argv[]) {
//   ios::sync_with_stdio(false);
//   cin.tie(nullptr);

//   int T = 1;
//   // cin >> T;
//   while (T--) {
//     cout << solve(4, 4, 30, 3, {0, 0, 1, 2}, {1, 2, 3, 3}, {5, 4, 2, 4}, {1, 0, 1, 1}) << '\n';
//   }

//   return 0;
// }

• Subtask 1: 0 / 5
• Subtask 2: 8 / 8
• Subtask 3: 0 / 13
• Subtask 4: 0 / 19
• Subtask 5: 7 / 7
• Subtask 6: 0 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3