Submission #977577

#	Time	Username	Problem	Language	Result	Execution time	Memory
977577		Ning07	Swapping Cities (APIO20_swap)	C++17	0 / 100	69 ms	6596 KiB

swap

#include "swap.h"

#include <vector>
#include<bits/stdc++.h>
using namespace std;

#define L(i, j, k) for (int i = (j); i <= k; i++) 
#define R(i, j, k) for (int i = (j); i >= k; i--) 

const int nax = 100050;
int n, m, to[nax];
pair<int, int> T[nax];
void init(int N, int M, std::vector<int> U, std::vector<int> V, std::vector<int> W) {
    n = N, m = M;
    L(i, 1, M){
      T[i] = make_pair(W[i], V[i]);
      to[V[i]] = W[i];
    }
    sort(T + 1, T + 1 + N);
}

int getMinimumFuelCapacity(int X, int Y) {
   if (n <= 3) return -1;
   if (X == 0) {
      int count = 0, ans = to[Y];
      L(i, 0, 2) {
         // 1: we want to find 0 -> A (that is not Y) 
         // 2: Y -> 0 -> B, 
         // 1: A -> 0 -> Y
         // 2: B -> 0
         // we need at most 2 nodes that is not Y
         if (T[i].second == Y) continue;
         count++;
         ans = max(ans, T[i].first);
         if (count >= 2) break;
      }
      return ans;
   }
   else {
      // 1: X -> 0 -> A
      // 2: Y -> 0 -> X
      // 1: A -> 0 -> Y
      // we need at most one node that is not X and not Y
      L(i, 0, 2) {
         if (T[i].second != X && T[i].second != Y) {
            return max({T[i].first, to[X], to[Y]});
         }
      }
   }
}

Compilation message (stderr)

swap.cpp: In function 'int getMinimumFuelCapacity(int, int)':
swap.cpp:50:1: warning: control reaches end of non-void function [-Wreturn-type]
   50 | }
      | ^

• Subtask 1: 0 / 6
• Subtask 2: 0 / 7
• Subtask 3: 0 / 17
• Subtask 4: 0 / 20
• Subtask 5: 0 / 23
• Subtask 6: 0 / 27