【

还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。

输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。

示例 1:

输入: [1,1,2,2,2]
输出: true

解释: 能拼成一个边长为2的正方形，每边两根火柴。
示例 2:

输入: [3,3,3,3,4]
输出: false

解释: 不能用所有火柴拼成一个正方形。

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

】

数组先考虑排序。然后dfs，先用大的火柴，省点时间。找到退出条件，四条边都等于target，target = sum/4.

bucket[4]为四条边的长度，选择一根火柴加下长度，不选要记得把刚加的减掉。

int comp(void const *a, void const *b) {
    return *(int *)b - *(int *)a;
}

bool generate(int i, int *nums, int target, int *bucket, int numsSize) {
    int j;

    if (i >= numsSize) {
        return (bucket[0] == target) && (bucket[1] == target) &&
                (bucket[2] == target) && (bucket[3] == target);
    }

    for (j = 0; j < 4; j++) {
        if (bucket[j] + nums[i] > target) {
            continue;
        }
        bucket[j] += nums[i];
        if (generate(i+1, nums, target, bucket, numsSize)) {
            return true;
        }

        bucket[j] -= nums[i];
    }

    return false;
}

bool makesquare(int* nums, int numsSize){
    int flag = 0;
    int bucket[4] = {0};
    int i;
    int sum = 0;

    if (numsSize < 4) {
        return false;
    }

    for (i = 0; i < numsSize; i++) {
        sum += nums[i];
    }

    if (sum % 4) {
        return false;
    }

    qsort(nums, numsSize, sizeof(int), comp);

    return generate(0, nums, sum/4, bucket, numsSize);
}