【无标题】

来看下今天的算法题，这题是LeetCode的第46题：全排列。

问题描述

来源：LeetCode第46题

难度：中等

给定一个不含重复数字的数组 nums ，返回其所有可能的全排列 。你可以 按任意顺序 返回答案。

示例1：

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

输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

示例2：

输入：nums = [0,1]

输出：[[0,1],[1,0]]

• 1 <= nums.length <= 6

• -10 <= nums[i] <= 10

• nums 中的所有整数互不相同

问题分析

这是一道经典的回溯算法问题，之前在讲什么叫回溯算法，一看就会，一写就废的时候，也提到过使用回溯算法来解决。

假设数组长度是n，每次只能从数组中取出一个元素，我们可以把它看作是一颗n叉树的dfs遍历，使用递归，当往下走的时候选择当前元素，往回走的时候撤销选择，可以看下动画演示。

C语言

void dfs(int **ans, int *returnSize, int *path, int index, int *nums, int numsSize, int *visited);

int **permute(int *nums, int numsSize, int *returnSize, int **returnColumnSizes) {
    int **ans = malloc(720 * sizeof(int *));
    *returnColumnSizes = malloc(720 * sizeof(int));
    int *path = malloc(6 * sizeof(int));
    int *visited = calloc(1, sizeof(int));
    *returnSize = 0;
    dfs(ans, returnSize, path, 0, nums, numsSize, visited);
    for (int i = 0; i < *returnSize; ++i)
        (*returnColumnSizes)[i] = numsSize;
    return ans;
}

void dfs(int **ans, int *returnSize, int *path, int index, int *nums, int numsSize, int *visited) {
    // 终止条件，如果数字都被使用完了，说明找到了一个排列
    if (index == numsSize) {
        ans[(*returnSize)] = malloc(index * sizeof(int));
        memcpy(ans[(*returnSize)++], path, index * sizeof(int));
        return;
    }
    // 可以把它看做是遍历n叉树每个节点的子节点
    for (int i = 0; i < numsSize; i++) {
        // 因为不能有重复的，所以有重复的就跳过
        if (((*visited) & (1 << i)) != 0)
            continue;
        // 选择当前元素
        (*visited) |= 1 << i;
        path[index++] = nums[i];
        //递归（可以把它看做遍历子节点的子节点）
        dfs(ans, returnSize, path, index, nums, numsSize, visited);
        // 撤销选择，把最后一次添加的元素给移除
        (*visited) &= ~(1 << i);
        index--;
    }
}