入门力扣自学笔记49 C++ （题目编号473）

473. 火柴拼正方形

题目：

你将得到一个整数数组 matchsticks ，其中 matchsticks[i] 是第 i 个火柴棒的长度。你要用 所有的火柴棍 拼成一个正方形。你 不能折断 任何一根火柴棒，但你可以把它们连在一起，而且每根火柴棒必须 使用一次 。

如果你能使这个正方形，则返回 true ，否则返回 false 。

---

示例 1:

输入: matchsticks = [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形，每边两根火柴。

---

示例 2:

输入: matchsticks = [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。

---

提示:

1 <= matchsticks.length <= 15
1 <= matchsticks[i] <= 108

---

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/matchsticks-to-square
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：

首先，我们将所有的数字加起来和4取余，来确定是否能构成正方形，如果能，确定没条边的边长。

其次，我们构建一个cache的容器，来确定每个元素是否被选过，再利用DFS来进行遍历。

最后，在DFS中进行循环，只要找到一个点的状态不是1而且小于等于边长，那么就对他进行递归，只要找到最后结果，就返回TRUE即可。

---

代码：

#define getBit(x, n) ((x >> n) & 1)
#define setBit(x, n) (x | (1 << n))
class Solution {
public:
    int side; // 边长
    vector<int8_t> cache;  
    bool makesquare(vector<int>& matchsticks) {
        // 求出边长
        int sum = accumulate(matchsticks.begin(), matchsticks.end(), 0);
        if(sum % 4) return false;
        side = sum / 4;
        // dfs
        int n = matchsticks.size();
        cache = vector<int8_t>((1 << n), -1);
        cache[(1 << n) - 1] = 1;
        return dfs(0, 0, matchsticks);        
    }

    // state - 当前状态，第 i 位为 1 表示 matchsticks[i] 已选择
    // sum - 表示当前累加边长
    bool dfs(int state, int sum, vector<int>& matchsticks){
        if(cache[state] != -1) return cache[state];
        int n = matchsticks.size();
        bool ans = 0;
        for(int i=0; i<n; ++i){
            if(getBit(state, i) == 0 && sum + matchsticks[i] <= side) { 
                int new_sum = ( sum + matchsticks[i] ) % side;
                if( dfs(setBit(state, i), new_sum, matchsticks) ) {
                    ans = 1;
                    break;
                };
            }
        }
        cache[state] = ans;
        return ans;
    }
};

---

注 ：

1.getBit()函数

geiBit函数是⽤来获取每个位的1或者是0。

这⾥要注意的是⼀般的计算机采⽤bool来表⽰⼆进制变量。但是这个⼆进制变量也要占⽤8个位，即⼀个字节。⽤getBit函数取出每个位的0或者是1，然后⽤bool类型的变量来表⽰这个变量。