✨2739.总行驶距离

Author: techoc

README.md

@@ -132,6 +132,7 @@
 - [2609.最长平衡子字符串](leetcode/easy/t2609/solution.go)
 - [2656.K 个元素的最大和](leetcode/easy/t2656/solution.go)
 - [2706.购买两块巧克力](leetcode/easy/t2706/solution.go)
+- [2739.总行驶距离](leetcode/easy/t2739/solution.go)
 - [2760.最长奇偶子数组](leetcode/easy/t2760/solution.go)
 - [2765.最长交替子数组](leetcode/easy/t2765/solution.go)
 - [2798.满足目标工作时长的员工数目](leetcode/easy/t2798/solution.go)

leetcode/easy/t2739/solution.go

@@ -0,0 +1,38 @@
+package t2739
+
+// 2739. 总行驶距离
+// https://leetcode.cn/problems/total-distance-traveled
+func distanceTraveled(mainTank int, additionalTank int) int {
+	ans := 0
+	for mainTank >= 5 { // 如果主油箱有 5 升油，则消耗 50 升油
+		ans += 50
+		mainTank -= 5
+		if additionalTank > 0 { // 如果副油箱有油，则补充 1 升油
+			mainTank++
+			additionalTank--
+		}
+	}
+	ans += mainTank * 10
+	return ans
+}
+
+func distanceTraveledTwo(mainTank, additionalTank int) int {
+	res := 0 // 初始化总行进距离为0
+
+	// 当主油箱中的油量大于等于5时，进行循环计算
+	for mainTank >= 5 {
+		// 每次至少能行进5单位距离，累加到结果中
+		res += mainTank / 5 * 5
+		// 计算本次能从主油箱中取出的油量，但不超过附加油箱的剩余油量
+		addNum := (mainTank / 5)
+		if addNum > additionalTank {
+			addNum = additionalTank
+		}
+		// 更新主油箱和附加油箱的油量
+		mainTank = mainTank%5 + addNum // 主油箱剩余油量加上取出的油量
+		additionalTank -= addNum       // 附加油箱减去取出的油量
+	}
+
+	// 最后，将剩余的油量转换为行进距离，并累加到结果中，乘以10表示单位转换
+	return (res + mainTank) * 10
+}

leetcode/easy/t2739/solution_test.go

@@ -0,0 +1,45 @@
+package t2739
+
+import "testing"
+
+func TestDistanceTraveled(t *testing.T) {
+	tests := []struct {
+		name           string
+		mainTank       int
+		additionalTank int
+		want           int
+	}{
+		{name: "Case 1", mainTank: 1, additionalTank: 2, want: 10},
+		{name: "Case 2", mainTank: 5, additionalTank: 10, want: 60},
+		{name: "Case 3", mainTank: 9, additionalTank: 2, want: 110},
+		{name: "Case 4", mainTank: 10, additionalTank: 2, want: 120},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := distanceTraveled(tt.mainTank, tt.additionalTank); got != tt.want {
+				t.Errorf("distanceTraveled() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestDistanceTraveledTwo(t *testing.T) {
+	tests := []struct {
+		name           string
+		mainTank       int
+		additionalTank int
+		want           int
+	}{
+		{name: "Case 1", mainTank: 1, additionalTank: 2, want: 10},
+		{name: "Case 2", mainTank: 5, additionalTank: 10, want: 60},
+		{name: "Case 3", mainTank: 9, additionalTank: 2, want: 110},
+		{name: "Case 4", mainTank: 10, additionalTank: 2, want: 120},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := distanceTraveledTwo(tt.mainTank, tt.additionalTank); got != tt.want {
+				t.Errorf("distanceTraveled() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}