FastPrep
Problem · Array

Calculate Total Distance Travelled

Learn this problem
● MediumAmazonOA
See Amazon hiring insights

Problem statement

Giving position of trucks and they are always moving towards end station as it is a gas station. Given a second array indicating extra gas station position in 1-indexing. Calculate the total distance travelled by all the trucks.

Example: Given trucks at positions [0,2,5,9,12,18] and extra gas stations at positions [[2,5],[3]], the answer to be returned is [12,18]. The solution can be solved in O(n^2) time complexity with one for loop for traversing gas station array and a second loop to calculate total distance travelled. However, there are test cases where this approach may not be optimal.

Function

calculateTotalDistanceTravelled(trucks: int[], extraGasStations: int[][]) → int[]

Complete the function calculateTotalDistanceTravelled in the editor.

calculateTotalDistanceTravelled has the following parameters:

  1. 1. int[] trucks: an array of integers indicating the initial positions of the trucks
  2. 2. int[][] extraGasStations: a 2D array of integers indicating the positions of extra gas stations

Returns

int[]: an array of integers indicating the total distance travelled by all the trucks

Examples

Example 1

trucks = [0,2,5,9,12,18]extraGasStations = [[2,5],[1, 3]]return = [12,18]
Heads up ~~> explanation is not from the official The trucks at positions 0, 2, and 5 will travel to the gas station at position 5. The trucks at positions 9 and 12 will travel to the gas station at position 12. The truck at position 18 does not need to travel as it is already at the end station. Therefore, the total distance travelled by all the trucks is 5 + 3 + 0 + 3 + 0 + 0 = 12 for the first gas station and 0 + 0 + 0 + 0 + 0 + 0 = 0 for the second gas station, resulting in the answer [12, 18].

Constraints

🍇🍇

More Amazon problems

drafts saved locally
public int[] calculateTotalDistanceTravelled(int[] trucks, int[][] extraGasStations) {
  // write your code here
}
trucks[0,2,5,9,12,18]
extraGasStations[[2,5],[1, 3]]
expected[12,18]
checking account