Leetcode 494. Target Sum 目标和

题目:

给定一个非负整数数组,a1, a2, ..., an, 和一个目标数,S。现在你有两个符号 + 和 -。对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。

返回可以使最终数组和为目标数 S 的所有添加符号的方法数。

示例 1:

输入: nums: [1, 1, 1, 1, 1], S: 3
输出: 5
解释: 

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

一共有5种方法让最终目标和为3。

注意:

  1. 数组的长度不会超过20,并且数组中的值全为正数。
  2. 初始的数组的和不会超过1000。
  3. 保证返回的最终结果为32位整数。

解题思路:

动态规划,j表示的状态是求和的值,里面存的是出现这个和的次数。

代码实现:

class Solution {
    public int findTargetSumWays(int[] nums, int S) {
        if (S > 1000) return 0;

        int[][] map = new int[nums.length][2001];
        map[0][1000 - nums[0]] ++;
        map[0][1000 + nums[0]] ++;
        for (int i = 1; i < nums.length; i ++) {
            for (int j = map[0].length - 1; j >= 0; j --) {
                if (j - nums[i] >= 0) map[i][j] += map[i - 1][j - nums[i]];
                if (j + nums[i] < map[0].length) map[i][j] += map[i - 1][j + nums[i]];
            }
        }

        return map[nums.length - 1][S + 1000];
    }
}

你可能感兴趣的:(Leetcode,(401~500),leetcode,java)