1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
| // @Title: 分割等和子集 (Partition Equal Subset Sum)
// @Author: 15816537946@163.com
// @Date: 2019-09-20 18:15:27
// @Runtime: 12 ms
// @Memory: 3.2 MB
/*
* @lc app=leetcode.cn id=416 lang=golang
*
* [416] 分割等和子集
*
* https://leetcode-cn.com/problems/partition-equal-subset-sum/description/
*
* algorithms
* Medium (43.99%)
* Likes: 121
* Dislikes: 0
* Total Accepted: 9.3K
* Total Submissions: 21K
* Testcase Example: '[1,5,11,5]'
*
* 给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
*
* 注意:
*
*
* 每个数组中的元素不会超过 100
* 数组的大小不会超过 200
*
*
* 示例 1:
*
* 输入: [1, 5, 11, 5]
*
* 输出: true
*
* 解释: 数组可以分割成 [1, 5, 5] 和 [11].
*
*
*
*
* 示例 2:
*
* 输入: [1, 2, 3, 5]
*
* 输出: false
*
* 解释: 数组不能分割成两个元素和相等的子集.
*
*
*
*
*/
// 01背包问题
func canPartition(nums []int) bool {
sum := 0
for _, n := range nums {
sum+=n
}
if sum &1 == 1 {
return false
}
sum = sum >> 1
dp := make([]int, sum+1)
for _, v := range nums {
for i := sum;i>=v;i-- {
dp[i] = max(dp[i], dp[i-v]+v) // 使用了滚动数组
}
}
return dp[sum] == sum
}
func max(a, b int) int {
if a > b {
return a
}
return b
}
|