LeetCode 494: Target Sum 解题与思考

LeetCode 494: Target Sum 解题与思考

[原题链接]

题目描述

给出n个数字，在其间添加加号或者减号，获得某个目标和，求不同的符号添加方法的数目。

思路

显然的n-1个数字能够得到哪些数字，决定了n个数字能够得到哪些数字。
所以我们从第一个数字开始，求所有能够得到的数字，一直递推到第n个即可。
所幸题目状态空间不大，可以直接用数组保存所有状态。

算法

准备两个数组，下标表示目标和，值表示有多少种得到方法；
一个数组$a_{sum}$保存未计算当前数字时，能够得到方法数，初始化为0；
一个数组$b_{sum}$保存计算当前数字后，能够得到方法数，初始化为0；

假设当前数字为nums[k]，
初始时$b_{±nums[0]} += 1；$
之后按照如下方式更新：

$$b_{sum ± nums[k]} = a_{sum}$$
在下次循环时，让 $a_{sum} = b_{sum}$， $b_{sum}$变为0
最终返回 $b_{S}$

代码

#include <iostream>
#include <vector>
#include <stdlib.h>
#include<string.h>
using namespace std;





class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int S) {
        if(S > 1000 || S < -1000) return false;
        int *different_solution = (int*)malloc(2200 * sizeof(int));
        int *different_solution_next = (int*)malloc(2200 * sizeof(int));
        memset(different_solution, 0, 2200 * sizeof(int));
        memset(different_solution_next, 0, 2200 * sizeof(int));


        for ( auto i = nums.begin(); i != nums.end(); i++ ) {
            if ( i == nums.begin() ) {
                different_solution_next[*i + 1000] += 1;
                different_solution_next[-*i + 1000] += 1;
                continue;
            }
            int *temp = different_solution;
            different_solution = different_solution_next;
            different_solution_next = temp;
            memset(different_solution_next, 0, 2200 * sizeof(int));
            for ( int j = *i; j < 2100 - *i; j++ ) {
                different_solution_next[j - *i] += different_solution[j];
                different_solution_next[j + *i] += different_solution[j];
            }
        }

        int solution = different_solution_next[S + 1000];
        free(different_solution);
        free(different_solution_next);
        return solution;
    }
};

思考

一开始想用队列来维护更新，但是发现状态空间比较小之后，干脆就写循环了。