求一个集合的排列 LeetCode OJ :Permutations

最新推荐文章于 2020-03-22 18:08:20 发布

niji523

最新推荐文章于 2020-03-22 18:08:20 发布

阅读量1k

点赞数

分类专栏：程序

本文链接：https://blog.csdn.net/niji523/article/details/17004711

版权

程序专栏收录该内容

4 篇文章 0 订阅

订阅专栏

LeetCode的一道题目，题目如下：

Given a collection of numbers, return all possible permutations.
For example,
[1,2,3] have the following permutations:
[1,2,3], [1,3,2], [2,1,3], [2,3,1], [3,1,2], and [3,2,1].

思路：假设集合S[n]的排列为P[n]，集合S[n+1]是在S[n]的基础上添加一个元素a，记S[n+1]的排列为P[n+1]。求P[n+1]的过程就是：取P[n]中的每个集合元素P[i]，在P[i]的第一个元素前面添加a，得到一个新集合，接着在P[i]中每个元素的后面添加a，得到n个新集合，将这n+1个新产生的集合覆盖掉P[i]集合即可。但对元素P[n]的每个元素都做这样的处理后，得到的新集合就是P[n+1]。

比如：S[2]=[1,2]，P[2]=[ [1,2], [2,1] ]，S[3]=[1,2,3]，那么求P[3]的过程为：

取P[2]的集合元素[1,2]，将a=3先放在第一个元素的前面得到[3,1,2]，再依次[1,2]放在每个元素的后面，得到：[1,3,2] [1,2,3]。用上面得到的三个新集合覆盖[1,2]得到第一个元素处理后的结果： [ [2,1][3,1,2] [1,3,2] [1,2,3 ] ]

继续处理P[2]的第二个元素[2,1]，同理将得到的集合[3,2,1] [2,3,1] [2,1,3]覆盖[2,1]，得到： [[3,2,1] [2,3,1] [1,2,3] [3,1,2] [1,3,2] [1,2,3 ] ]，即是要求的P[3]。

C++代码实现如下：

class Solution {
public:
    vector<vector<int> > permute(vector<int> &num) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        vector<vector<int> > result, val;  
        vector<int> elem;  
        int i,j,size; 
        
        if(num.size()==0) return result;
        elem.push_back(num[0]);  
        result.push_back(elem);
        for(i=1;i<num.size();i++) 
        {  
            size=result.size();
            for(j=size-1;j>=0;j--)  
            {  
                val.clear();
                for(int k=0;k <= result[j].size()-1;k++)
                {
                    elem.resize(result[j].size());  
                    memcpy(&elem[0], &result[j][0], elem.size()*sizeof(int));  
                    elem.insert(elem.begin()+k,num[i]);  
                    val.push_back(elem);  
                }
                result[j].insert(result[j].end(),num[i]);
                result.insert(result.end(), val.begin(), val.end()) ;
            }  
        }  
        return result;  
    }
};

本人最近在学Python语言，发现使用Python写出来的代码更易懂：

#!/usr/bin/python
import copy
source = [1,2,3]  #原始的集合
result =[[]]  #结果保存的集合
def add(sum,k):  #当t个元素的处理结果为sum时，计算添加元素k后的结果
    elem=[]
    size=len(sum)
    if size == 1 and sum[0]==[]:  #处理第一个元素
        elem.append(k)
        sum.append(elem)
        sum.remove([])
        return
    list = [i for i in range(size)]
    for i in list[::-1]:  #倒序访问sum的每个元素
        for j in range(len(sum[i])+1):  #将k添加到sum[i]的0-len(sum[i])的前面，其中编号len(sum[i])超出下标，但是表示添加到队列末尾
            elem=copy.deepcopy(sum[i])
            elem.insert(j,k)
            sum.insert(len(sum),elem)  #将结果保存在sum的队列尾部
        sum.remove(sum[i])  #sum[i]处理完成，删除sum[i]

if __name__ == "__main__":
    size=len(source)
    for i in range(size):  #遍历添加元素集合的每一个元素
        add(result,source[i])
    L=sorted(result,key=lambda x:x[0])  #自己定义的排序函数，不重要
    print "set:%s"%(source)
    print "subset:\n%s"%("\n".join([str(L[elem]) for elem in range(len(L))])) #输出结果
    print "totalsize:%s"%(len(result))  #个数

结果截图：