夹娃多项式展开

夹娃多项式展开
突然发神经想到写个多项式展开的代码,想了一下发现还有点绕,于是...
首先研究下cn2的形式
private int run97_2(int[] a) {
int s = 0;
for (int i = 0; i < a.length; i++) {
for (int j = i + 1; j < a.length; j++) {
for (int k = j + 1; k < a.length; k++) {
s += a[i] * a[j] * a[k];
}
}
}
return s;
}
再研究下cn3的形式
private int run97_3(int[] a) {
int s = 0;
for (int i = 0; i < a.length; i++) {
for (int j = i + 1; j < a.length; j++) {
for (int k = j + 1; k < a.length; k++) {
for (int m = k + 1; m < a.length; m++) {
s += a[i] * a[j] * a[k] * a[m];
}
}
}
}
return s;
}
如果将这个形式改写成递归的形式,变量会出现在递归方法外,这段代码被我改掉了,也不复杂有兴趣的可以自己写下,
于是将这种形式改写为如下形式
private int run97_5(int[] a) {
int s = 0;
for (int i = 0; i < a.length; i++) {
int si = 0;
for (int j = i + 1; j < a.length; j++) {
int sk = 0;
for (int k = j + 1; k < a.length; k++) {
int sm = 0;
for (int m = k + 1; m < a.length; m++) {
sm += a[m];
}
sk += sm * a[k];
}
si += sk * a[j];
}
s += si * a[i];
}
return s;
}
这样变量就全在这个方法内了,
根据这种形式的代码改写出来的递归形式如下
private int run97_4_1(int[] a, int[] idx, int lvcurrent, int lvall) {
int s = 0;
if (lvcurrent == lvall - 1) {
// System.out.println("a:"+idx[lvcurrent]);
for (int i = idx[lvcurrent-1] + 1; i < a.length; i++) {
s += a[i];
// s+=a[idx[lvcurrent-1]]*a[idx[i]];
}
return s;
} else {
// System.out.println("b:"+idx[lvcurrent]);
for (int i = idx[lvcurrent-1] + 1; i < a.length; i++) {
idx[lvcurrent] = i;
s += a[i] * run97_4_1(a, idx, lvcurrent + 1, lvall);
}
return s;
}
}


在matlab上测试一下结果对比一下
夹娃:
private void run97() {
int[] a = new int[] { 1, 2, 3, 4, 5, 6 };
System.out.println(run97_4(a, 2));
System.out.println(run97_4(a, 3));
System.out.println(run97_4(a, 4));
System.out.println(run97_4(a, 5));
System.out.println(run97_4(a, 6));
System.out.println(run97_4(a, 7));
}

private int run97_4(int[] a, int lvall) {
int[] idx = new int[lvall];
idx[0]=-1;
return run97_4_1(a, idx, 1, lvall);
}

public static void main(String[] args) {
Test1 t = new Test1();
t.run97();
}
matlab
syms x y;
y=sym((x-1)*(x-2)*(x-3)*(x-4)*(x-5)*(x-6));
expand(y)
一致.恩,还马马虎虎
(*..*)