[Mdfs] lc473. 火柴拼正方形(剪枝优化+经典题+好题)

1. 题目来源

链接：473. 火柴拼正方形

拔高、证明：

• [dfs] aw167. 木棒(dfs剪枝与优化+分类讨论+思维+好题)

2. 题目解析

水一篇。和之前的一个问题一模一样，在此不再赘述，写出来方便搜索。

• [Mdfs] lc698. 划分为k个相等的子集(剪枝优化+经典题+好题)

---

• 时间复杂度：此类 dfs 和剪枝，不必进行分析
• 空间复杂度：此类 dfs 和剪枝，不必进行分析

---

class Solution {
public:bool makesquare(vector<int>& matchsticks) {return canPartitionKSubsets(matchsticks, 4);}bool canPartitionKSubsets(vector<int>& nums, int k) {int n = nums.size();int sum = 0;for (int x : nums) sum += x;int x = sum / k;if (x * k != sum) return false;sort(nums.begin(), nums.end(), greater<int>()); // 剪枝1：从大到小进行搜索vector<bool> st(n, false);function<bool(int, int, int)> dfs = [&](int u, int cur, int cnt) {if (cnt == k) return true;if (cur == x) return dfs(0, 0, cnt + 1);for (int i = u; i < n; i ++ ) { // 从大到小搜索if (st[i]) continue;if (cur + nums[i] > x) continue;// 尝试搜索当前位置st[i] = true;if (dfs(i + 1, cur + nums[i], cnt)) return true;// 搜索当前位置失败的话st[i] = false;while (i + 1 < n && nums[i] == nums[i + 1]) i ++ ; // 剪枝2：搜索的相同元素失败if (!cur || cur + nums[i] == x) return false; // 剪枝3、4：如果搜索的第一个、最后一个失败}return false;};return dfs(0, 0, 0);}
};