给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这个题目没啥好说的,想到回溯算法基本上就可以解决了,毕竟子集搜索问题,就是用这个方法来处理的
主要的关键在于去重问题 , 如何去重是关键问题,参考 卡尔 大神的思路,进行层级去重就很容易解决,不过要先预处理,进行排序
代码:
/**
* 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 backtracking(int start, int* nums, int numsSize, int** ans, int* returnSize, int* col, int* path, int pathTop)
{
// 收集子集结果
int j;
ans[*returnSize] = (int*)malloc(sizeof(int) * pathTop);
for (j = 0; j < pathTop; j++) {
ans[*returnSize][j] = path[j];
}
col[(*returnSize)++] = pathTop;
// 终止条件
if (start >= numsSize) {
return;
}
// 进行同层广度搜索
int i;
for (i = start; i < numsSize; i++) {
// 进行去重
if (i > start && nums[i] == nums[i - 1]) continue; // 注意这个去重的起始位置,应该是start 和后面剩下的元素进行的去重操作,不应该包含start本身自己,start和自己比较没有什么意义
// 进行下一个节点的尝试
path[pathTop++] = nums[i];
// 进行深度递归搜索
backtracking(i + 1, nums, numsSize, ans, returnSize, col, path, pathTop);
// 恢复现场
pathTop--;
}
}
void QuickSort(int* nums, int start, int end)
{
if (start >= end) return;
int flag = start;
int left = start;
int right = end;
int tmp;
while (left < right) {
// 先从右边开始找第一个小于flag位置的数
while(left < right && nums[right] > nums[flag])right--;
// 从左边找第一个大于flag位置的数
while(left < right && nums[left] <= nums[flag])left++;
// 两个数交换
if (left < right) {
tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
}
}
tmp = nums[left];
nums[left] = nums[flag];
nums[flag] = tmp;
QuickSort(nums, start, right - 1);
QuickSort(nums, right+1, end);
}
int** subsetsWithDup(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){
// 对数组进行预处理,从高到低进行排序
QuickSort(nums, 0, numsSize - 1);
// 保存一个临时子集的空间
int* path = (int*)malloc(sizeof(int) * numsSize);
// 临时子集 的空间里面保存的元素数量
int pathTop = 0;
// 收集结果
int** ans = (int**)malloc(sizeof(int*) * 10086); // 10086 是对中国移动永远的怨念
// 子集个个数
*returnSize = 0;
// 每个子集元素的数量
int* col = (int*)malloc(sizeof(int) * 10086);
memset(col, 0x00,sizeof(int) * 10086);
*returnColumnSizes = col;
// 进行回溯计算
backtracking(0, nums, numsSize, ans, returnSize, col, path, pathTop);
return ans;
}