我已经编写了一段代码来查找数组的所有可能子集的乘积之和。我得到了预期的输出,但我无法使它快速到足以通过与时间相关的测试用例。
有人可以帮我优化我的代码以提高速度吗?
第一个输入(testCases)是测试用例的数量。根据测试用例的数量,我们将拥有数组的大小(size)和数组元素(set)。
例如,有效的输入如下:
例如,有效的输入如下:
1
3
2 3 5
其中:
1
是测试用例的数量。3
是测试集的大小,2 3 5
是输入集的元素。
期望输出为:
71
以上输出的计算方法如下:
{2}, {3}, {5}, {2, 3}, {3, 5}, {2, 5}, {2, 3, 5}
=> 2 3 5 6 15 10 30
=> 2 + 3 + 5 + 6 + 15 + 10 + 30
=> 71
import java.util.Scanner;
public class Test {
static int printSubsets(int set[]) {
int n = set.length;
int b = 0;
for (int i = 0; i < (1 << n); i++) {
int a = 1;
for (int j = 0; j < n; j++){
if ((i & (1 << j)) > 0) {
a *= set[j];
}}
b += a;
}
return b;
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int testCases = scanner.nextInt();
for (int i = 0; i < testCases; i++) {
int size = scanner.nextInt();
int set[] = new int[size];
for (int j = 0; j < set.length; j++) {
set[j] = scanner.nextInt();
}
int c = printSubsets(set);
System.out.println((c - 1));
}
scanner.close();
}
}
(2 * 3) + (2 * 5) + (2 * 3 * 5) + (3 * 5) = 61
。 - Turing852 + 3 + 5
加到61
上。这样你就得到了71
。 - anacron