含有重复元素集合的全排列
给定一个可包含重复数字的整数集合 nums ,按任意顺序 返回它所有不重复的全排列。
示例 1:
输入:nums = [1,1,2]
输出:
[[1,1,2],
[1,2,1],
[2,1,1]]
示例 2:
输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
这题倒不是难,主要是要掌握方法,我这里给出标准答案,我也不是很熟练
我们将这个问题看作有 nnn 个排列成一行的空格,我们需要从左往右依次填入题目给定的 nnn 个数,每个数只能使用一次。那么很直接的可以想到一种穷举的算法,即从左往右每一个位置都依此尝试填入一个数,看能不能填完这 nnn 个空格,在程序中我们可以用「回溯法」来模拟这个过程。
我们定义递归函数 backtrack(idx,perm)\textit{backtrack}(\textit{idx}, \textit{perm})backtrack(idx,perm) 表示当前排列为 perm\textit{perm}perm,下一个待填入的位置是第 idx\textit{idx}idx 个位置(下标从 000 开始)。那么整个递归函数分为两个情况:
如果 idx=n\textit{idx} = nidx=n,说明我们已经填完了 nnn 个位置,找到了一个可行的解,我们将 perm\textit{perm}perm 放入答案数组中,递归结束。
如果 idx<n\textit{idx} < nidx<n,我们要考虑第 idx\textit{idx}idx 个位置填哪个数。根据题目要求我们肯定不能填已经填过的数,因此很容易想到的一个处理手段是我们定义一个标记数组 vis\textit{vis}vis 来标记已经填过的数,那么在填第 idx\textit{idx}idx 个数的时候我们遍历题目给定的 nnn 个数,如果这个数没有被标记过,我们就尝试填入,并将其标记,继续尝试填下一个位置,即调用函数 backtrack(idx+1,perm)\textit{backtrack}(\textit{idx} + 1, \textit{perm})backtrack(idx+1,perm)。搜索回溯的时候要撤销该个位置填的数以及标记,并继续尝试其他没被标记过的数。
但题目解到这里并没有满足「全排列不重复」 的要求,在上述的递归函数中我们会生成大量重复的排列,因为对于第 idx\textit{idx}idx 的位置,如果存在重复的数字 iii,我们每次会将重复的数字都重新填上去并继续尝试导致最后答案的重复,因此我们需要处理这个情况。
要解决重复问题,我们只要设定一个规则,保证在填第 idx\textit{idx}idx 个数的时候重复数字只会被填入一次即可。而在本题解中,我们选择对原数组排序,保证相同的数字都相邻,然后每次填入的数一定是这个数所在重复数集合中「从左往右第一个未被填过的数字」,即如下的判断条件:
int* vis;
void backtrack(int* nums, int numSize, int** ans, int* ansSize, int idx, int* perm) {
if (idx == numSize) {
int* tmp = malloc(sizeof(int) * numSize);
memcpy(tmp, perm, sizeof(int) * numSize);
ans[(*ansSize)++] = tmp;
return;
}
for (int i = 0; i < numSize; ++i) {
if (vis[i] || (i > 0 && nums[i] == nums[i - 1] && !vis[i - 1])) {
continue;
}
perm[idx] = nums[i];
vis[i] = 1;
backtrack(nums, numSize, ans, ansSize, idx + 1, perm);
vis[i] = 0;
}
}
int cmp(void* a, void* b) {
return *(int*)a - *(int*)b;
}
int** permuteUnique(int* nums, int numsSize, int* returnSize, int** returnColumnSizes) {
int** ans = malloc(sizeof(int*) * 2001);
int* perm = malloc(sizeof(int) * 2001);
vis = malloc(sizeof(int) * numsSize);
memset(vis, 0, sizeof(int) * numsSize);
qsort(nums, numsSize, sizeof(int), cmp);
*returnSize = 0;
backtrack(nums, numsSize, ans, returnSize, 0, perm);
*returnColumnSizes = malloc(sizeof(int) * (*returnSize));
for (int i = 0; i < *returnSize; i++) {
(*returnColumnSizes)[i] = numsSize;
}
return ans;
}