【LeetCode】494. 目标和

【LeetCode】494. 目标和_第1张图片

第一种思路:全部遍历。复杂度为O(2^n)

【LeetCode】494. 目标和_第2张图片

/**
 * @Auther: Mason
 * @Date: 2020/07/15/11:14
 * @Description:
 */
public class Solution {
    private int res = 0;

    public int findTargetSumWays(int[] nums, int S) {
        // 给定一个非负整数数组
        // 对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。
        // 返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
        // 直观的算法是 进行枚举。每一个元素都可以加两种符号。 复杂度为:O(2^n)
        calculate(nums, 0, 0, S);
        return res;
    }

    private void calculate(int[] nums, int i, int sum, int target) {
        if (i == nums.length) {
            if (sum == target) {
                res++;
            }
        } else {
            calculate(nums, i + 1, sum + nums[i], target);
            calculate(nums, i + 1, sum - nums[i], target);
        }
    }


}

第二种思路:

转化成0-1背包客问题

【LeetCode】494. 目标和_第3张图片

/**
 * @Auther: Mason
 * @Date: 2020/07/15/15:34
 * @Description:
 */
public class Solution {
    // 转换成了 背包客问题
    // 集合P-集合N=target
    // 集合P=(target+集合N+结合P)/2
    // 等式右边可以计算出来 数值。
    // 所以就是从数组中找到若干元素,总和为某值的可能数。转换成了0-1背包问题。
    // 数组的序号为背包的容量。数组的数值为 正好充满此容量的可能数。
    public int findTargetSumWays(int[] nums, int S) {
        int sum = 0;
        for (int num : nums) {
            sum += num;
        }
        if (sum < S || sum < (-1 * S)) return 0;
        if ((S + sum) % 2 == 1) return 0;
        int target = (S + sum) / 2;
        int[] arr = new int[target + 1];
        arr[0] = 1;
        // 遍历金块。
        for (int i = 0; i < nums.length; i++) {
            for (int j = target; j >= nums[i]; j--) { // 遍历容量。
                arr[j] = arr[j] + arr[j - nums[i]];
            }
        }
        return arr[target];
    }
}

你可能感兴趣的:(算法)