Merge pull request neetcode-gh#789 from MaratKhakim/787-Cheapest-Flights-Go

Ahmad-A0 · web-flow · commit 7be9da6e6f97 · 2022-08-13T15:28:18.000+01:00
GO: 787. Cheapest Flights within K Stops
diff --git a/go/787-Cheapest-Flights-Within-K-Stops.go b/go/787-Cheapest-Flights-Within-K-Stops.go
@@ -0,0 +1,38 @@
+func findCheapestPrice(n int, flights [][]int, src int, dst int, k int) int {
+	cost := make([]int, n)
+
+	for i := 0; i < n; i++ {
+		cost[i] = math.MaxUint32
+	}
+
+	cost[src] = 0
+
+	for i := 0; i <= k; i++ {
+		temp := make([]int, n)
+		copy(temp, cost)
+
+		for j := 0; j < len(flights); j++ {
+			from, to, price := flights[j][0], flights[j][1], flights[j][2]
+
+			if cost[from] != math.MaxUint32 {
+				temp[to] = min(temp[to], cost[from]+price)
+			}
+		}
+
+		cost = temp
+	}
+
+	if cost[dst] == math.MaxUint32 {
+		return -1
+	}
+
+	return cost[dst]
+}
+
+func min(x, y int) int {
+	if x > y {
+		return y
+	}
+
+	return x
+}
