您的位置:首页 > 职场人生

【程序员面试金典】 集合的子集

2017-06-27 15:41 127 查看
题目:请编写一个方法,返回某集合的所有非空子集。

给定一个int数组A和数组的大小int n,请返回A的所有非空子集。保证A的元素个数小于等于20,且元素互异。各子集内部从大到小排序,子集之间字典逆序排序,见样例。

[123,456,789]

返回:{[789,456,123],[789,456],[789,123],[789],[456 123],[456],[123]}


题解:观察测试样例,会发现每个子集的选择规律与二进制((2^n) - 1)到 1 的顺序生成的规律是一致的,样例中n=3,2^n-1=7,用二进制表示为111,其中每一位的1表示数组中的三个数都选择。

代码:

class Subset {

public:

    vector<vector<int> > getSubsets(vector<int> A, int n) {

        sort(A.begin(),A.end());

        vector<vector<int>> res;

        int size=1<<n;

        for(int i=size-1;i>0;i--){

            vector<int> temp;

            for(int j=n-1;j>=0;j--){

                if((i>>j)&1)

                    temp.push_back(A[j]);

            }

            res.push_back(temp);

        }

        return res;

    }

};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: