78. Subsets

Given a set of distinct integers, nums, return all possible subsets (the power set).

Note: The solution set must not contain duplicate subsets.

Example:

Input: nums = [1,2,3]
Output:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

 

解法1:回溯

void backtrack(int *nums, int numsSize, int **ret, int *retIdx, int **returnColumnSizes, int idx, int *output, int *out_idx)
 {
     if(idx == numsSize)
     {
         int i;
         int k = *retIdx;
         for(i = 0; i < *out_idx; ++i)
         {
             ret[k][i] = output[i];
         }
         (*returnColumnSizes)[k] = *out_idx;
        *retIdx += 1;
         return;
     }

     output[*out_idx] = nums[idx];
     *out_idx += 1;
     backtrack(nums, numsSize, ret, retIdx, returnColumnSizes, idx + 1, output, out_idx);
     *out_idx -= 1;
     backtrack(nums, numsSize, ret, retIdx, returnColumnSizes, idx + 1, output, out_idx);

 }
int** subsets(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){

    int totalSize = 1;
    int tmp = numsSize;
    while( tmp-- > 0)
    {
        totalSize *= 2;
    }

    *returnSize = totalSize;

    int **ret = (int **)malloc(sizeof(int *) * totalSize);
    int *buffer = (int *)malloc(sizeof(int) * numsSize * totalSize);
    int i;
    for(i = 0; i < totalSize; ++i)
    {
        ret[i] = buffer + i * numsSize;
    }

    *returnColumnSizes = (int *)malloc(sizeof(int) * totalSize);

    int *output = (int *)malloc(sizeof(int) * numsSize);

    int out_idx = 0;
    
    *returnSize = 0;
    backtrack(nums, numsSize, ret, returnSize, returnColumnSizes, 0, output, &out_idx);

    return ret;
}

回溯2:

/**
 * 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().
 */

 void backtrack(int *nums, int numsSize, int **ret, int *retIdx, int **returnColumnSizes, int idx, int *output, int *output_idx)
 {
     int i;
     int k = *retIdx;
     for(i = 0; i < *output_idx; ++i)
     {
         ret[k][i] = output[i];
	printf("%d ", output[i]);
     }
     printf("\n");
     (*returnColumnSizes)[k] = *output_idx;
     *retIdx += 1;

     for(i = idx; i < numsSize; ++i)
     {
         output[*output_idx] = nums[i];
         *output_idx += 1;
         backtrack(nums, numsSize, ret, retIdx, returnColumnSizes, i + 1, output, output_idx);
         *output_idx -= 1;
     }
 }
int** subsets(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){

    int totalSize = 1;
    int tmp = numsSize;
    while( tmp-- > 0)
    {
        totalSize *= 2;
    }

    *returnSize = totalSize;

    int **ret = (int **)malloc(sizeof(int *) * totalSize);
    int *buffer = (int *)malloc(sizeof(int) * numsSize * totalSize);
    int i;
    for(i = 0; i < totalSize; ++i)
    {
        ret[i] = buffer + i * numsSize;
    }

    *returnColumnSizes = (int *)malloc(sizeof(int) * totalSize);

    int *output = (int *)malloc(sizeof(int) * numsSize);

    *returnSize = 0;
    int output_idx = 0;
    backtrack(nums, numsSize, ret, returnSize, returnColumnSizes, 0, output, &output_idx);
    return ret;
}

3. 位运算解法

/**
 * 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().
 */

 void backtrack(int *nums, int numsSize, int **ret, int *retIdx, int **returnColumnSizes)
 {
     int *output = (int *)malloc(sizeof(int) * numsSize);
     int idx;
     int i, j;
     for(i = 0; i < (1 << numsSize); ++i)
     {
         idx = 0;
         int j = i;
         int k = 0;
         while(j > 0)
         {
             if(j &1 == 1)
             {
                output[idx++] = nums[k];
             }
             ++k;
             j = (j >> 1);
         }

         for(k = 0; k < idx; ++k)
         {
             ret[*retIdx][k] = output[k];
         }

         (*returnColumnSizes)[*retIdx] = idx;
        *retIdx += 1;
     }
 }
int** subsets(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){

    int totalSize = 1;
    int tmp = numsSize;
    while( tmp-- > 0)
    {
        totalSize *= 2;
    }

    *returnSize = totalSize;

    int **ret = (int **)malloc(sizeof(int *) * totalSize);
    int *buffer = (int *)malloc(sizeof(int) * numsSize * totalSize);
    int i;
    for(i = 0; i < totalSize; ++i)
    {
        ret[i] = buffer + i * numsSize;
    }

    *returnColumnSizes = (int *)malloc(sizeof(int) * totalSize);

    int *output = (int *)malloc(sizeof(int) * numsSize);

    *returnSize = 0;
    int output_idx = 0;
    backtrack(nums, numsSize, ret, returnSize, returnColumnSizes);
    return ret;
}

图引用自

https://leetcode-cn.com/problems/subsets/solution/shou-hua-tu-jie-zi-ji-hui-su-fa-xiang-jie-wei-yun-/

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值