需求:
Given a number n
, we need to find the sum of all the elements from all possible subsets of a set formed by first n natural numbers.
样例
Given n = 2, return 6
Possible subsets are {{1}, {2}, {1, 2}}. Sum of elements in subsets
is 1 + 2 + 1 + 2 = 6
Given n = 3, return 24
Possible subsets are {{1}, {2}, {3}, {1, 2}, {1, 3}, {2, 3}, {1, 2, 3}}
Sum of subsets is :
1 + 2 + 3 + (1 + 2) + (1 + 3) +
(2 + 3) + (1 + 2 + 3)
分析:
对于求所有子集的和、子序列最大和、子序列最大乘积,都是使用动态规划求解,可以将大问题转化成小问题,然后迭代求解。
设n对应的所有子集的和是S(n),我们看一下规律。
n=1,子集是{1}
n=2,子集是{1},{2},{1,2}
n=3,子集是{1},{2},{1,2},{3},{1,3},{2,3},{1,2,3}
n=4,子集是{1},{2},{1,2},{3},{1,3},{2,3},{1,2,3},{4},{1,4},{2,4},{3,4},{1,2,4},{1,3,4},{2,3,4},{1,2,3,4}
n=...
可以发现,n对应的子集包括(n-1)的子集,并且多了2^(n-1)个子集,n和(n-1)对应子集和的关系为:
因此,S(n)的求解依赖于S(1)到S(n-1),使用动态规划算法,将大问题转换成小问题,创建一个长度(n+1)的数组int[] S,S[i]表示i对应的所有子集的和,S[1]=1,从i=2到n逐个求解对应的值,最后返回S[n]。
代码:
public class Solution {
/*
* @param : the given number
* @return: Sum of elements in subsets
*/
public int subSum(int n) {
// write your code here
//如果n<=0,那么这个子集是无意义的,所以返回0
if(n <= 0){
return 0;
}
//创建长度n+1的整型数组,递推求解
int[] S = new int[n+1];
S[1] = 1;//初始化角标1的元素,表示数字1所有子集的和
//递推求解
for(int i = 2; i < n+1; i++){
S[i] = 2*S[i-1] + i*(int)(Math.pow(2, i-1));
}
return S[n];
}
}