leetcode-algorithms/src/TotalDistanceTraveled.java at master · anishLearnsToCode/leetcode-algorithms · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
// https://leetcode.com/problems/total-distance-traveled
// M = mainTank
// T: O(M)
// S: O(1)

public class TotalDistanceTraveled {
    public int distanceTraveled(int mainTank, int additionalTank) {
        int totalDistance = 0;
        while (mainTank > 0) {
            if (mainTank >= 5) {
                mainTank -= 5;
                totalDistance += 50;
                if (additionalTank > 0) {
                    mainTank++;
                    additionalTank--;
                }
            } else {
                totalDistance += mainTank * 10;
                mainTank = 0;
            }
        }

        return totalDistance;
    }
}