码迷,mamicode.com
首页 > 其他好文 > 详细

377. Combination Sum IV

时间:2016-09-30 09:47:26      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:

是看到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     }

注意arrays要sort。因为dp是从小往大走的,我们会希望处理到后面的时候前面的结果已经处理好可以用了。

 

377. Combination Sum IV

标签:

原文地址:http://www.cnblogs.com/warmland/p/5922591.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!