码迷,mamicode.com
首页 > 其他好文 > 详细

子集生成

时间:2019-10-16 16:18:32      阅读:115      评论:0      收藏:0      [点我收藏+]

标签:mamicode   com   nbsp   图片   class   img   push   子集   显示   

问题:输出$[0,1,2,3,...n)$的所有子集。

1. 增量构造法:一次选出一个元素放到集合中。由于$A$中的元素个数不确定,每次递归调用都要输出当前集合。另外,递归边界也不需要显示确定——如果无法继续添加元素,自然就不会递归了。

void print_subset(int n, int* A, int cur){
    for(int i = 0; i < cur; i++) printf("%d ", A[i]);
    printf("\n");
    int s = cur ? A[cur - 1] + 1 : 0;
    for(int i = s; i < n; ++i){
        A[cur] = i;
        print_subset(n, A, cur + 1);
    }
}

2. 位向量法:构造一个位向量$B[i]$,而不是直接构造子集$A$本身,其中$B[i]=1$,当且仅当$i$在子集$A$中。

void print_subset1(int n, int* B, int cur){
    if(cur == n){
        for(int i = 0; i < cur; ++i)
            if(B[i]) printf("%d ", i);
        printf("\n");
        return;
    }
    B[cur] = 1;
    print_subset1(n, B, cur + 1);
    B[cur] = 0;
    print_subset1(n, B, cur + 1);
}

3. 二进制法:当用二进制表示子集时,位运算中的按位与、或、异或对应集合的交、并和对称差。

技术图片

输出子集$S$对应的各个元素:

void print_subset2(int n, int s){
    for(int i = 0; i < n; ++i)
        if(s & (1 << i)) printf("%d ", i);
    printf("\n");
}

枚举子集:

for(int i = 0; i < (1 << n); ++i)
        print_subset2(n, i);

 Leetcode 78. Subsets

给一个包含不同整数的集合,返回其所有子集。

方法1:(recursive)

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> subs;
        vector<int> sub;
        dfs(nums, 0, sub, subs);
        return subs;
    }
    
    void dfs(vector<int>& nums, int i, vector<int>& sub, vector<vector<int>>& subs){
        subs.push_back(sub);
        for(int j = i; j < nums.size(); ++j){
            sub.push_back(nums[j]);
            dfs(nums, j + 1, sub, subs);
            sub.pop_back();
        }
    }
};

方法2:位运算

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> res;
        for(int i = 0; i < (1 << n); ++i){
            vector<int> cur;
            for(int j = 0; j < n; ++j){
                if(i & (1 << j))
                    cur.push_back(nums[j]);
            }
            res.push_back(cur);
        }
        return res;
    }
};

 方法3:(iterative)

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> subs = {{}};
        for (int num : nums) {
            int n = subs.size();
            for (int i = 0; i < n; i++) {
                subs.push_back(subs[i]); 
                subs.back().push_back(num);
            }
        }
        return subs;
    }
}; 

 Leetcode 90. Subsets II

给定一个集合,包含重复元素,返回其所有子集。

方法1:(iterative)

技术图片

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> res = {{}};
        for(int i = 0; i < nums.size();){
            int cnt = 0, num = nums[i];
            while(i < nums.size() && nums[i] == num){
                i++;
                cnt++;
            }
            int size = res.size();
            for(int j = 0; j < size; ++j){
                vector<int> tmp = res[j];
                for(int k = 0; k < cnt; ++k){
                    tmp.push_back(num);
                    res.push_back(tmp);
                }
            }
        }
        return res;
    }
};

 方法2:(recursive)

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> subs;
        vector<int> sub;
        dfs(subs, sub, nums, 0);
        return subs;
    }
    
    void dfs(vector<vector<int>>& subs, vector<int>& sub, vector<int>& nums, int i){
        subs.push_back(sub);
        for(int j = i; j < nums.size(); ++j){
            if(j == i || nums[j] != nums[j - 1]){
                sub.push_back(nums[j]);
                dfs(subs, sub, nums, j + 1);
                sub.pop_back();
            }
        }
    }
};

 

子集生成

标签:mamicode   com   nbsp   图片   class   img   push   子集   显示   

原文地址:https://www.cnblogs.com/betaa/p/11686111.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!