面试题：Go语言中for和range循环在切片上的应用

Question

在Go语言中，给定一个整数切片 []int{1, 2, 3, 4, 5}，使用for - range循环遍历该切片，并计算所有元素的平方，将结果存入一个新的切片中，最后返回这个新切片。请写出完整的代码。

Answer 1

package main

import "fmt"

func main() {
    nums := []int{1, 2, 3, 4, 5}
    squares := make([]int, len(nums))
    for i, num := range nums {
        squares[i] = num * num
    }
    fmt.Println(squares)
}