力扣hot100 分割等和子集变形01背包滚动数组优化

最新推荐文章于 2024-07-21 15:36:00 发布

兑生

最新推荐文章于 2024-07-21 15:36:00 发布

阅读量507

点赞数 6

分类专栏： # 力扣 hot100 文章标签： leetcode 算法职场和发展

兑生：前世大厂架构师、上市公司CEO，被小人陷害，重生为25应届生，三连加关注，看我如何逆风翻盘。

本文链接：https://blog.csdn.net/lt6666678/article/details/135686332

版权

力扣 hot100 专栏收录该内容

100 篇文章 1 订阅

订阅专栏

Problem: 416. 分割等和子集

思路

👨‍🏫 参考地址

在这里插入图片描述

💖 01背包

复杂度

时间复杂度: $O (nm)$ ： $m$ 为数组元素和的一半

空间复杂度: $O (nm)$

Code

class Solution {
	public boolean canPartition(int[] nums)
	{
		int n = nums.length;
		int sum = 0;
		for (int x : nums)
			sum += x;
		if (sum % 2 == 1)
			return false;
		int m = sum / 2;
//		注意：元素只可选一次
		boolean[][] f = new boolean[n][m + 1];// 表示从前 i 个元素中选，使得这些数恰好 = j 的方案是否可行

		if (nums[0] <= m)
			f[0][nums[0]] = true;

		for (int i = 1; i < n; i++)// 先枚举每个数
		{
			for (int j = 0; j <= m; j++)// 再枚举背包容量
			{
				f[i][j] = f[i - 1][j];// 不选当前数
				if (nums[i] <= j)
//							   不选当前数              选当前数
					f[i][j] = f[i - 1][j] || f[i - 1][j - nums[i]];
			}
			if (f[i][m])// 只要出现一次f[i][m] 为真，即返回 true
				return true;
		}
		return false;
//		return f[n - 1][m];//走到这里必返回false
	}
}

💖 滚动数组优化

复杂度

时间复杂度: $O (nm)$ ： $m$ 为数组元素和的一半

空间复杂度: $O (m)$

Code

class Solution {
	public boolean canPartition(int[] nums)
	{
		int n = nums.length;
		int sum = 0;
		for (int x : nums)
			sum += x;
		if (sum % 2 == 1)
			return false;
		int m = sum / 2;
//		注意：元素只可选一次
		boolean[] f = new boolean[m + 1];// 表示从前 i 个元素中选，使得这些数恰好 = j 的方案是否可行

		if (nums[0] <= m)
			f[nums[0]] = true;

		for (int i = 1; i < n; i++)// 先枚举每个数
		{
			for (int j = m; j >= nums[i]; j--)// 再枚举背包容量
			{
//							   不选当前数              选当前数
				f[j] = f[j] || f[j - nums[i]];
			}
			if (f[m])// 只要出现一次f[i][m] 为真，即返回 true
				return true;
		}
		return false;
	}
}