377. Combination Sum IV

是看到discuss里面的解法,因为用backtracking实在太多可能性了

思路是和https://leetcode.com/problems/climbing-stairs/

在climbing stairs里面假如有n个台阶,每次可以跨一个台阶或者两个台阶,那么它的状态转移方程是res[i] = res[i - 1] + res[i - 2],初始化是res[0] = 1; res[1] = 1;

但是在本题中,每次不再只是可以跨一步或者两步了,每次可以跨nums数组里面的任意数字的步,而且终点不是n了,终点是target数。所以状态转移方程是

foreach possible number in nums: res[i] += res[i - each] (i - each >= 0)

而且其中等于0也没有关系我们把res[0]初始化成1就好了。

 

 1     public int combinationSum4(int[] nums, int target) {
 2         if(nums.length == 0) {
 3             return 0;
 4         }
 5         Arrays.sort(nums);
 6         int[] res = new int[target + 1];
 7         res[0] = 1;
 8         for(int i = 1; i <= target; i++) {
 9             for(int num: nums) {
10                 if(i - num >= 0) {
11                     res[i] += res[i - num];
12                 } else {
13                     break;
14                 }
15             }
16         }
17         return res[target];
18     }

 

posted @ 2016-09-30 08:51  warmland  阅读(137)  评论(0编辑  收藏  举报