Combination and Permutation

来源:互联网 发布:淘宝有哪些女装潮店 编辑:程序博客网 时间:2024/06/06 04:29

码了几个题,都是backtrack 的问题变形

主要涉及的还是 combination and permutation 形式基本相同 

稍微写一下

首先是combination, 就是从n个数里面选k个,先挑一个出来,再从剩下的n-1个挑 k-1个或者k个

    public List<List<Integer>> combine(int n, int k) {        List<List<Integer>> list = new ArrayList<>();        backtrace(n,k,list,new ArrayList<>());//helper method         return list;    }            private void backtrace (int n, int k, List<List<Integer>> list, List<Integer> present){        if(k<=0){//已经挑出来k个了结束            list.add(new ArrayList(present));//退出递归的条件        }else{            for(int i = n; i >0; i--){                present.add(i);                backtrace(i-1,k-1,list,present);//包含i的情况                present.remove(present.size() - 1);//删除i的情况            }        }    }


然后是permutation,和combination不同的是,所有的元素都要参与

    public List<List<Integer>> combine(int n, int k) {        List<List<Integer>> list = new ArrayList<>();        backtrace(n,k,list,new ArrayList<>());        return list;    }            private void backtrace (int n, int k, List<List<Integer>> list, List<Integer> present){        if(present.size()==n){//只有到present序列的长度和全部元素的个数相同的时候才结束            list.add(new ArrayList(present));        }else{            for(int i = n; i >0; i--){//每一次都把所有元素遍历一遍 与combination不同的是,combination里面n是递减的                if(present.contains(i)) continue;//这一步在permutation里面就是跳过自己                present.add(i);                backtrace(n,k-1,list,present);//实际上这里不需要n。。只是直接从combination改过来的                present.remove(present.size() - 1);            }        }    }


在combinationsum2的问题里面

是这样的

public class Solution {public List<List<Integer>> combinationSum2(int[] nums, int target) {    List<List<Integer>> list = new ArrayList<>();    Arrays.sort(nums);    backtrack(list, new ArrayList<>(), nums, target, 0);    return list;    }private void backtrack(List<List<Integer>> list, List<Integer> tempList, int [] nums, int remain, int start){    if(remain < 0) return;    else if(remain == 0) list.add(new ArrayList<>(tempList));    else{        for(int i = start; i < nums.length; i++){            if(i > start && nums[i] == nums[i-1]) continue; // ~~!!不一样的点在这            tempList.add(nums[i]);            backtrack(list, tempList, nums, remain - nums[i], i + 1);            tempList.remove(tempList.size() - 1);         }    }} }


首先这个nums是sorted所以如果有相同数字就连在一块了

比如说输入 11223  要求sum是5 那么就需要避免 122 122 这种情况,其实是不一样的1 和2 但是结果是duplicate的



原创粉丝点击