mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-10 12:15:58 +08:00
添加二维数组解法 + 优化
This commit is contained in:
@ -291,18 +291,63 @@ false true false false false true true false false false true true
|
||||
|
||||
### Python:
|
||||
```python
|
||||
# 一维度数组解法
|
||||
class Solution:
|
||||
def canPartition(self, nums: List[int]) -> bool:
|
||||
target = sum(nums)
|
||||
if target % 2 == 1: return False
|
||||
target //= 2
|
||||
dp = [0] * 10001
|
||||
dp = [0] * (len(nums) + 1)
|
||||
for i in range(len(nums)):
|
||||
for j in range(target, nums[i] - 1, -1):
|
||||
dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])
|
||||
return target == dp[target]
|
||||
```
|
||||
|
||||
```python
|
||||
# 二维度数组解法
|
||||
class Solution:
|
||||
def canPartition(self, nums: List[int]) -> bool:
|
||||
target = sum(nums)
|
||||
nums = sorted(nums)
|
||||
|
||||
# 做最初的判断
|
||||
if target % 2 != 0:
|
||||
return False
|
||||
|
||||
# 找到 target value 可以认为这个是背包的体积
|
||||
target = target // 2
|
||||
|
||||
row = len(nums)
|
||||
col = target + 1
|
||||
|
||||
# 定义 dp table
|
||||
dp = [[0 for _ in range(col)] for _ in range(row)]
|
||||
|
||||
# 初始 dp value
|
||||
for i in range(row):
|
||||
dp[i][0] = 0
|
||||
|
||||
for j in range(1, target):
|
||||
if nums[0] <= j:
|
||||
dp[0][j] = nums[0]
|
||||
|
||||
# 遍历 先遍历物品再遍历背包
|
||||
for i in range(1, row):
|
||||
|
||||
cur_weight = nums[i]
|
||||
cur_value = nums[i]
|
||||
|
||||
for j in range(1, col):
|
||||
if cur_weight > j:
|
||||
dp[i][j] = dp[i - 1][j]
|
||||
else:
|
||||
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - cur_weight] + cur_value)
|
||||
|
||||
# 输出结果
|
||||
return dp[-1][col - 1] == target
|
||||
```
|
||||
|
||||
### Go:
|
||||
```go
|
||||
// 分割等和子集 动态规划
|
||||
|
Reference in New Issue
Block a user