Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
matchsticks-to-square.cpp 1.30 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2016-12-20 20:36 +08:00 . Update matchsticks-to-square.cpp
// Time: O(n * s * 2^n), s is the number of subset of which sum equals to side length.
// Space: O(n * (2^n + s))
class Solution {
public:
bool makesquare(vector<int>& nums) {
int sum = accumulate(nums.begin(), nums.end(), 0);
if (sum % 4) {
return false;
}
const auto side_len = sum / 4;
const auto all = (1 << nums.size()) - 1;
vector<int> used_subsets;
vector<bool> valid_half_subsets(1 << nums.size());
for (int subset = 0; subset <= all; ++subset) {
int subset_sum = 0;
for (int i = 0; i < nums.size(); ++i) {
if (subset & (1 << i)) {
subset_sum += nums[i];
}
}
if (subset_sum == side_len) {
for (const auto& used_subset : used_subsets) {
if ((used_subset & subset) == 0) {
int valid_half_subset = used_subset | subset;
valid_half_subsets[valid_half_subset] = true;
if (valid_half_subsets[all ^ valid_half_subset]) {
return true;
}
}
}
used_subsets.emplace_back(subset);
}
}
return false;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助