oj.uz

Submission #975963

# Submission time Handle Problem Language Result Execution time Memory
975963 2024-05-06T04:03:49 Z josanneo22 Swapping Cities (APIO20_swap) C++17
0 / 100
 2000 ms 6900 KB 
#include "swap.h"

#include <vector>
#include <bits/stdc++.h>
using namespace std;
struct dsu {
   vector<int> fa, sz, mx_deg;
   dsu(int n) {
      fa.resize(n + 10);
      iota(fa.begin(), fa.end(), 0);
      sz.assign(n + 10, 1);
      mx_deg.assign(n + 10, 0);
   }
   int find(int x) {
      while (x != fa[x]) x = fa[x] = fa[fa[x]];
      return x;
   }
   bool merge(int x, int y) {
      int px = find(x), py = find(y);
      if (px == py) return false;
      if (sz[py] > sz[px]) swap(px, py);
      sz[px] += sz[py]; fa[py] = px;
      mx_deg[px] = max(mx_deg[px], mx_deg[py]);
      return true;
   }
   bool same(int x, int y) {
      int px = find(x), py = find(y);
      return px == py;
   }
};
vector<pair<int, pair<int, int>>> edges;
int n;
void init(int _N, int M, std::vector<int> U, std::vector<int> V, std::vector<int> W) {
    n = _N;
    for (int i = 0; i < M; i++) 
      edges.push_back(make_pair(W[i], make_pair(U[i], V[i])));
    sort(edges.begin(), edges.end());
}
int getMinimumFuelCapacity(int X, int Y) {
   // cout << "++++++++++++++++\n";
   bool flag = true;
   dsu D(n);
   vector<int> deg(n);
   for (int i = 0; i < edges.size(); i++) {
      int w = edges[i].first;
      int u = edges[i].second.first, v = edges[i].second.second;
      // cout << w << ' ' << u << ' ' << v << '\n';
      for (int t = 0; t < 2; t++) {
         if (!flag && X != u && D.same(X, u) && D.same(X, Y) && D.same(u, Y)) return w; 
         swap(u, v);
      }
      D.merge(u, v);
      deg[u]++; deg[v]++;
      if (deg[u] >= 3) {
         D.mx_deg[D.find(u)] = 3;
      }
      if (deg[v] >= 3) {
         D.mx_deg[D.find(v)] = 3;
      }
      if (D.mx_deg[D.find(X)] >= 3 && D.same(X, Y)) return w;
      if (D.same(X, Y)) {
         flag = false;
         continue;
      }
   }
   if (D.mx_deg[D.find(X)] >= 3 && D.same(X, Y)) return edges.back().first;
   return -1;
}

Compilation message

swap.cpp: In function 'int getMinimumFuelCapacity(int, int)':
swap.cpp:44:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |    for (int i = 0; i < edges.size(); i++) {
      |                    ~~^~~~~~~~~~~~~~

Subtask #1
0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Execution timed out 2083 ms 6900 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #3
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #6
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -