如何为Go实现的嵌套数组乘积和ProductSum函数编写测试用例

报错核心原因 #

你之前的写法无法编译，本质是强类型语言下的类型不匹配问题：

• SpecialArray的底层类型是[]interface{}，和你声明的[][]int类型完全不兼容，[][]int要求所有元素都是[]int类型，直接放入整数1、2本身就不符合类型约束
• Go不会自动把未显式声明类型的嵌套切片字面量识别为SpecialArray，如果不手动指定类型，代码内的el.(SpecialArray)类型断言会直接失败，无法进入递归计算逻辑。

可直接运行的测试用例写法 #

所有嵌套层级的数组都需要显式声明为SpecialArray类型，以下是完整可运行代码，包含你提到的预期结果为13的基础测试场景：

package main

import "fmt"

type SpecialArray []interface{}

func ProductSum(array SpecialArray) int {
	return helper(array, 1)
}

func helper(array SpecialArray, multiplier int) int {
	sum := 0
	for _, el := range array {
		if cast, ok := el.(SpecialArray); ok {
			sum += helper(cast, multiplier + 1)
		} else if cast, ok := el.(int); ok {
			sum += cast
		}
	}
	return sum * multiplier
}

func main() {
	// 测试用例1: [1, [2, 3], 2] 预期结果13
	test1 := SpecialArray{
		1,
		SpecialArray{2, 3},
		2,
	}
	res1 := ProductSum(test1)
	fmt.Printf("测试用例1结果: %d, 预期值:13, 是否通过:%v\n", res1, res1 == 13)

	// 测试用例2: 三层嵌套 [5, 2, [7, -1], 3, [6, [-13, 8], 4]] 预期结果12
	test2 := SpecialArray{
		5,
		2,
		SpecialArray{7, -1},
		3,
		SpecialArray{
			6,
			SpecialArray{-13, 8},
			4,
		},
	}
	res2 := ProductSum(test2)
	fmt.Printf("测试用例2结果: %d, 预期值:12, 是否通过:%v\n", res2, res2 == 12)

	// 测试用例3: 空数组 预期结果0
	test3 := SpecialArray{}
	res3 := ProductSum(test3)
	fmt.Printf("测试用例3结果: %d, 预期值:0, 是否通过:%v\n", res3, res3 == 0)
}

运行上述代码输出如下：

测试用例1结果: 13, 预期值:13, 是否通过:true
测试用例2结果: 12, 预期值:12, 是否通过:true
测试用例3结果: 0, 预期值:0, 是否通过:true

编写注意事项 #

• 所有嵌套层级的切片，都必须用SpecialArray{...}的方式初始化，不能直接写{...}，否则类型断言无法识别为嵌套数组
• 整数元素直接写字面量即可，会被自动识别为int类型，命中整数计算分支
• 不要直接用[]interface{}初始化后传入函数，[]interface{}和自定义类型SpecialArray是独立的不同类型，会导致类型断言失败。

内容的提问来源于stack exchange，提问作者user531300