-
Notifications
You must be signed in to change notification settings - Fork 44
/
DerivativePoly_test.go
78 lines (68 loc) · 1.97 KB
/
DerivativePoly_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// DerivativePoly_test
/*
------------------------------------------------------
作者 : Black Ghost
日期 : 2018-12-25
版本 : 0.0.0
------------------------------------------------------
求单变量多项式n阶导数
理论：
------------------------------------------------------
输入 :
A 按幂次连续增加的系数向量,(Nn+1)x1,Nn为最高幂次
n 求导次数
输出 :
sol 解,(Nn+1-n)x1
err 解出标志：false-未解出或达到边界；
true-全部解出
------------------------------------------------------
*/
package goNum_test
import (
"testing"
"github.com/chfenger/goNum"
)
// DerivativePoly 求单变量多项式n阶导数
func DerivativePoly(A goNum.Matrix, n int) (goNum.Matrix, bool) {
/*
求单变量多项式n阶导数
输入 :
A 按幂次连续增加的系数向量,(Nn+1)x1,Nn为最高幂次
n 求导次数
输出 :
sol 解,(Nn+1-n)x1
err 解出标志：false-未解出或达到边界；
true-全部解出
*/
//判断求导次数与最高幂次关系
Nn := A.Rows - 1
if n > Nn+1 {
panic("Error in goNum.DerivativePoly: Derivative number greater than polynomial's order")
}
//Nn+1 = n
if Nn+1 == n {
return goNum.NewMatrix(1, 1, []float64{0.0}), true
}
sol := goNum.ZeroMatrix(Nn+1, 1)
var lenSol int = Nn + 1
var err bool = false
//赋予soltemp初值
for i := 0; i < Nn+1; i++ {
sol.Data[i] = A.Data[i]
}
//求导计算
for i := 1; i < n+1; i++ {
for j := 1; j < lenSol; j++ {
sol.Data[j-1] = float64(j) * sol.Data[j]
}
lenSol--
}
err = true
return goNum.NewMatrix(lenSol, 1, sol.Data[:lenSol]), err
}
func BenchmarkDerivativePoly(b *testing.B) {
A51 := goNum.NewMatrix(4, 1, []float64{5.0, 3.0, 4.0, 2.0})
for i := 0; i < b.N; i++ {
goNum.DerivativePoly(A51, 2) //{8.0,12.0}
}
}