completed 2090 - k radius subarray averages in go

Author: jgndev

go/kRadiusSubarrayAverges/README.md

@@ -0,0 +1 @@
+# 2090. K Radius Subarray Averages

go/kRadiusSubarrayAverges/go.mod

@@ -0,0 +1,3 @@
+module kRadiusSubarrayAverages
+
+go 1.23.1

go/kRadiusSubarrayAverges/solution.go

@@ -0,0 +1,57 @@
+package solution
+
+func getAverages(nums []int, k int) []int {
+	//if k == 0 {
+	//	return nums
+	//}
+	//
+	//windowSize := 2*k + 1
+	//n := len(nums)
+	//averages := make([]int, n)
+	//
+	//for i := range averages {
+	//	averages[i] = -1
+	//}
+	//
+	//if windowSize > n {
+	//	return averages
+	//}
+	//
+	//var windowSum int64 = 0
+	//
+	//for i := 0; i < windowSize; i++ {
+	//	windowSum += int64(nums[i])
+	//}
+	//
+	//averages[k] = int(windowSum / int64(windowSize))
+	//
+	//for i := windowSize; i < n; i++ {
+	//	windowSum = windowSum - int64(nums[i-windowSize]) + int64(nums[i])
+	//	averages[i-k] = int(windowSum / int64(windowSize))
+	//}
+	//
+	//return averages
+
+	if k == 0 {
+		return nums
+	}
+	res := make([]int, len(nums))
+
+	left, right, sum, division := 0, 0, 0, 2*k+1
+
+	for ; right < len(nums); right++ {
+		sum += nums[right]
+
+		if (right - left + 1) < division {
+			res[right] = -1
+		} else {
+			// shrink window
+			res[(left+right)/2] = sum / division
+			res[right] = -1
+			sum -= nums[left]
+			left++
+		}
+	}
+
+	return res
+}

go/kRadiusSubarrayAverges/solution_test.go

@@ -0,0 +1,56 @@
+package solution
+
+import (
+	"slices"
+	"testing"
+)
+
+func TestGetAverages(t *testing.T) {
+	testCases := []struct {
+		name string
+		nums []int
+		k    int
+		want []int
+	}{
+		{
+			name: "test case one",
+			nums: []int{1, 3, 2, 6, 2},
+			k:    1,
+			want: []int{-1, 2, 3, 3, -1},
+		},
+		{
+			name: "test case two",
+			nums: []int{7, 4, 3, 9, 1, 8, 5, 2, 6},
+			k:    3,
+			want: []int{-1, -1, -1, 5, 4, 4, -1, -1, -1},
+		},
+		{
+			name: "test case three",
+			nums: []int{5, 2, 8, 1, 9},
+			k:    0,
+			want: []int{5, 2, 8, 1, 9},
+		},
+		{
+			name: "test case four",
+			nums: []int{1, 2, 3, 4, 5},
+			k:    3,
+			want: []int{-1, -1, -1, -1, -1},
+		},
+		{
+			name: "test case five",
+			nums: []int{100000, 200000, 300000, 400000, 500000},
+			k:    1,
+			want: []int{-1, 200000, 300000, 400000, -1},
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			got := getAverages(tc.nums, tc.k)
+
+			if !slices.Equal(got, tc.want) {
+				t.Errorf("got %v, want %v", got, tc.want)
+			}
+		})
+	}
+}