package main
import (
"fmt"
"math"
)
func calculateFormula(e float64, a, b, c, d []float64) ([]float64, error) {
// Ensure all arrays have the same length
if len(a) != len(b) || len(b) != len(c) || len(c) != len(d) {
return nil, fmt.Errorf("arrays must have the same length")
}
// Initialize result array
r := make([]float64, len(a))
// Perform the calculation
for i := 0; i < len(a); i++ {
r[i] = e * (a[i] + b[i] * c[i] + math.Cos(d[i]))
}
return r, nil
}
func main() {
e := 7.0
a := []float64{1.0, 2.0, 3.0}
b := []float64{4.0, 5.0, 6.0}
c := []float64{7.0, 8.0, 9.0}
d := []float64{0.0, math.Pi / 4, math.Pi / 2}
// Calculate r
r, err := calculateFormula(e, a, b, c, d)
if err != nil {
fmt.Println("Error:", err)
return
}
// Print the result
fmt.Println("After calculation: ", r)
}
/*
run:
After calculation: [210 298.9497474683058 399]
*/