78. 子集

难度：中等

题目

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的子集（幂集）。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

输入：nums = [1,2,3]
输出：[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

• 1 <= nums.length <= 10
• -10 <= nums[i] <= 10
• nums 中的所有元素 互不相同

Reference

题解

已知集合中不存在重复元素，则可以使用整数中的每一位表示集合中元素的选取状态。

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int count1(int x)
{
    int ret = 0;
    while (x)
    {
        ret += x & 1;
        x >>= 1;
    }
    return ret;
}
int** subsets(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){
    *returnSize = 1 << numsSize;
    *returnColumnSizes = (int *)memset(malloc(sizeof(int) * *returnSize), 0, sizeof(int) * *returnSize);
    int i = 0, j = 0, **ret = (int **)malloc(sizeof(int *) * *returnSize), pos;
    for (i = 0; i < *returnSize; i++)
    {
        (*returnColumnSizes)[i] = count1(i);
        ret[i] = (int *)malloc(sizeof(int) * (*returnColumnSizes)[i]);
        pos = 0;
        for (j = 0; j < numsSize; j++)
            if (i & (1 << j))
                ret[i][pos++] = nums[j];
    }
    return ret;
}

评论