1. 程式人生 > >LeetCode--40. Combination Sum II

LeetCode--40. Combination Sum II

題目連結:https://leetcode.com/problems/combination-sum-ii/

這個題目是在39. Combination Sum加了點變化,就是說被選數組裡的數可以重複的,但是同一個陣列位置上的數不能重複選,不併且不同陣列位置上選擇的相同的數導致的相同的組合結果應該被剔除,比如最終結果[1,2,1]和[1,1,2]應該只保留一個。

這個題目我首先那想到的是在39拿到題目上做些微調,然後用HashSet將已經選擇的組合轉成字元存進去,然後利用它判重,如果重複就丟棄,否則加入到集合中並新增到最終返回的list中。這裡需要先把原數排序使得得到的每個答案有序然後轉為數字字串,這個方法是有點蠢笨的,但是容易想到。

class Solution {
    public static List<List<Integer>> ret;

    public static int[] record=new int[100000];

    public static HashSet<String> set;

    public static List<List<Integer>> combinationSum2(int[] candidates, int target) {

        ret=new LinkedList<>();
        set=new HashSet<>();
        Arrays.sort(candidates);
        recursive(candidates,-1,target,0);
        return ret;
    }

    public static void recursive(int[] candidates,int idx,int target,int length)
    {
        if(target==0)
        {
            StringBuilder sb=new StringBuilder();
            for(int i=0;i<length;i++)
                sb.append(record[i]);
            String st=sb.toString();

            if(set.contains(st))
                return;
            else
            {
                set.add(st);
                LinkedList<Integer> tmp=new LinkedList();
                for(int i=0;i<length;i++)
                    tmp.add(record[i]);
                ret.add(tmp);
                return;
            }
        }
        if(target<0 || idx>=candidates.length)
            return;
        for(int i=idx+1;i<candidates.length;i++)
        {
            record[length]=candidates[i];
            recursive(candidates,i,target-candidates[i],length+1);
        }
    }
}

效率當然較低,beat 14.26%!有時間再來看看高效的解法: