Find number of subsets with equal sum

最新推荐文章于 2021-12-21 21:09:30 发布

lqcsp

最新推荐文章于 2021-12-21 21:09:30 发布

阅读量521

点赞数

分类专栏： c++ LeetCode 面试题算法文章标签： leetcode 谷歌

本文链接：https://blog.csdn.net/lqcsp/article/details/52290077

版权

c++ 同时被 3 个专栏收录

80 篇文章 0 订阅

订阅专栏

LeetCode 面试题

73 篇文章 1 订阅

订阅专栏

算法

8 篇文章 0 订阅

订阅专栏

@(leetcode)[动态规划|Google]

Problem

Given a number N ( N > 1), gives a set S = [1,2,..., N].
Divide S in two subsets such that S1 ∪ S2 = S and S1 ∩ S2 = Ø and return the number of subsets with equal sum.

Example:
Given N = 7, S = [1, 2, 3, 4, 5, 6, 7]. Return 2, since sum of[1, 6, 7] = sum of [2, 3, 4, 5].

Solution

先求出数组的和sum
如果sum & 1 == 1, 返回0
否则就找出集合中和为 sum / 2的个数就是最终结果，背包问题

Code

int equal_sum_sub_set(vector<int>& num)
{
  int sum = 0;
  for(int n : num) {
    sum += n;
  }
  if((sum & 1) == 1) {
    return 0;
  }
  sum >>= 1;
  vector<int> dp(sum + 1, 0);
  dp[0] = 1;
  int current_sum = 0;
  for (int i = 0; i < num.size(); i++)
  {
      current_sum += num[i];
      for (int j = min(sum, current_sum); j >= num[i]; j--)
          dp[j] += dp[j - num[i]];
  }
  return dp[sum];
}