public class Solution {
    public int FindTargetSumWays(int[] nums, int S) {
        Queue<int> Q = new Queue<int>();

            Q.Enqueue(0);

            var count = 0;
            var dic = new Dictionary<int, int>();
            var dic2 = new Dictionary<int, int>();
            for (int i = 0; i < nums.Length; i++)
            {
                var num = nums[i];
                dic2.Clear();
                foreach (var d in dic)
                {
                    dic2.Add(d.Key, d.Value);
                }
                dic.Clear();
                while (Q.Count > 0)
                {
                    var n = Q.Dequeue();
                    var N = 1;
                    if (dic2.ContainsKey(n))
                    {
                        N = dic2[n];
                    }

                    if (!dic.ContainsKey(n + num))
                    {
                        dic.Add(n + num, N);
                    }
                    else
                    {
                        dic[n + num] += N;
                    }

                    if (!dic.ContainsKey(n - num))
                    {
                        dic.Add(n - num, N);
                    }
                    else
                    {
                        dic[n - num] += N;
                    }
                }

                foreach (var l in dic.Keys)
                {
                    if (l == S && i == nums.Length - 1)
                    {
                        count = dic[l];
                    }
                    else
                    {
                        Q.Enqueue(l);
                    }
                }
            }
            return count;
    }
}

https://leetcode.com/problems/target-sum/#/description

补充一个python的实现:

 1 class Solution:
 2     def findTargetSumWays(self, nums: 'List[int]', S: 'int') -> int:
 3         s = {0:1}
 4         s2 = {}
 5         for n in nums:
 6             for x in s.keys():
 7                 x1 = x + n
 8                 x2 = x - n
 9                 if x1 in s2.keys():
10                     s2[x1]=s2[x1]+s[x]
11                 else:
12                     s2[x1]=s[x]
13                 if x2 in s2.keys():
14                     s2[x2]=s2[x2]+s[x]
15                 else:
16                     s2[x2]=s[x]
17             #s.clear()
18             s = s2.copy()
19             s2.clear()
20         if S in s.keys():
21             return s[S]
22         else:
23             return 0

这道题的主要思想是利用字典存储之前的已经计算过的值,但是做题的时候状态不太好,所以写了很久。

做算法题,头脑不清醒的时候,效率很低,应该休息,把状态调整好再做。

简单介绍一下本题的思路,举例数据nums=[1, 1, 1, 1, 1], S=3。

初始情况下字典s为{0:1},表示值0出现过1次。

遍历nums,取每一个值进行计算,i为遍历的下标:

i=0时,当前值为1,得到0+1和0-1,因此s={1: 1, -1: 1}

i=1时,当前值为1,在之前的基础上分别计算+1和-1,有1+1==2,1-1==0,-1+1==0,-1-1==-2,因此s={2: 1, 0: 2, -2: 1}

i=2时,在之前基础上继续计算每一个值+1和-1,得到s={3: 1, 1: 3, -1: 3, -3: 1}

i=3时,s={4: 1, 2: 4, 0: 6, -2: 4, -4: 1}

i=4时,s={5: 1, 3: 5, 1: 10, -1: 10, -3: 5, -5: 1}

遍历完毕,目标S=3,则s[3]的值是5,表示有5种情况可以得到运算结果为3。

posted on 2017-05-13 18:55  Sempron2800+  阅读(183)  评论(0编辑  收藏  举报