首页 > 代码库 > leetcode Subsets

leetcode Subsets

2024-07-29 03:45:34   220人阅读

题目：

Given a set of distinct integers, S, return all possible subsets.

Note:

  • Elements in a subset must be in non-descending order.
  • The solution set must not contain duplicate subsets.

For example,
If S = [1,2,3], a solution is:

[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

那么有几种情况呢，如果给你的数字长度为n的话，那么首先是空数组1；选一个呢，为n个；如此下去，一个的个数为2^n个子数组。


这里给出一个java代码：

import java.util.*;

public class Solution {
    public List<List<Integer>> subsets(int[] S) {
        int totalLength  = 1<<S.length; //总共的子数组的个数
        List<List<Integer>>  l = new ArrayList<List<Integer>>();
        Arrays.sort(S);//数组进行排序

       
        for (int i=0;i<totalLength;i++){
             ArrayList<Integer> tmp = new ArrayList<Integer>();
            for(int j=0;j<S.length;j++){
                if(((i >> j) & 1 )!= 0){ //将s[n]分别用二进制的n-1位代替
                   tmp.add(S[j]); 
                }
               }
            l.add(tmp);
        }
    return l;
	}
	//测试主函数
    public static void main(String[] args) {
		int[] array = {1,2,3};
		Solution  s = new Solution();
		System.out.println(s.subsets(array));
	}
}


可能一下在看不懂上面的代码，特别是下面这段代码的意思。
 if(((i >> j) & 1 )!= 0)

看一个图，如下



上图中以S=[1 2 3 4]为例子，可以看到一个规律，就是在前面所有的subset的最后再添加一个位数，就成了新生成的subset。

或者，我们简单的一理解为，s[0]代码二级制数字最后一位为1，s[1]代码二进制数字倒数第二位为1，如此下去。

如上图中，倒数第二个subset为[2 3 4]，那么这代表着其用二级制表示为1110 = 14 这好是其外围数组的下标，这样不能理解上面列出的那个段代码了（或者如下）

 if(((i >> j) & 1 )!= 0)

由上面图中的原理，这样就比较好的理解下面的python代码了：

def subset1(S):
    A=[[]]
    S.sort()

    for n in S:
        for i in range(len(A)):
            ss = A[i][:] 
            ss.append(n) 
            A.append(ss)
    return A

下面的python在原理上再效率上会有所提高：

def subset2(S):
    res = [[]]
    S.sort()

    for n in S:
        res = res + [x + [n] for x in res]

    return res


leetcode Subsets


联系
我们