Leetcode 473. 火柴拼正方形

Leetcode 473. 火柴拼正方形

2022.6.1 Leetcode 打卡每日一题

链接：https://leetcode.cn/problems/matchsticks-to-square/

看到n<=15，首选状态压缩dp，st[s] 表示当状态为s的时候当前正在放置的边上还空余的长度，这里需要取模len，是因为当前的边已经放满len的时候，我们要切换到下一条边。最后返回当n位数全部为1（全部用完n个木棍）是否为零即可，为零意味着没有剩余长度。

注意~是按位取反；!是按逻辑取反（非0的数变为0，0变为1）；memset只能用来初始化全0或全-1的数组。

class Solution {
public:
    int sum = 0;
    int st[(1 << 15)];
    bool makesquare(vector<int>& v) {
        int n = v.size();
        bool flag = false;
        for(int i=0; i<n; i++) {
            sum += v[i];
        }
        if(sum % 4 != 0) return false;
        int len = sum / 4;

        // sort(v.begin(), v.end());
        memset(st, -1, sizeof(st));
        st[0] = 0;
        for(int i = 1; i < (1 << n); i++) {
            for(int j = 0; j < n; j++) {
                if((i & (1<<j)) == 0 )  continue;
                int s = i & ~(1<<j);
                if(st[s] >= 0 && st[s] + v[j] <= len) {
                    st[i] = (st[s] + v[j]) % len;
                    break;
                }
            }
        }

        return st[(1 << n) - 1] == 0;

    }
};

大家六一儿童节快乐！！！