LeetCode 解答 #39. Combination Sum プログラミング練習

問題:
f:id:stlisacity:20180611151530p:plain
難易度: medium
入力: int型array, (int) target
目的: 入力arrayの中から和がtargetになる組み合わせを
出力: リストのリスト

例にあるように、[2,3,6,7]、target=7が入力の場合、
答えは
[
[7],
[2,2,3]
]
になります。
入力arrayの中の数字を使い、ターゲットの値に同等な組み合わせをすべて探し出す問題です。
ただ入力の数字は同じものを何度も使うことが可能です。

注意すべき事: 

  • 入力arrayの数字はすべて正数である。
  • 重複した答えがないよう注意しましょう。

以下回答です。

class Solution {
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> l = new ArrayList<Integer>();
        for (int i=0; i<candidates.length; i++) {
            if (candidates[i] > target) break;
            combSum(candidates, target, i, l, res);
            l.clear();
            }
        return res;
    }
    
    public boolean combSum(int[] candidates, int target, int i, List<Integer> l, List<List<Integer>> res) {
        int s = 0;
        for (int j=0; j<l.size(); j++) {
            s += l.get(j);
        }
        if (s == target) {
            res.add(new ArrayList<>(l));
            return false;
        }
        if (s > target) return false;
        for (; i<candidates.length; i++) {
            if (candidates[i] > target) return false;
            l.add(candidates[i]);
            boolean flag = combSum(candidates, target, i, l, res);
            l.remove(l.size() - 1);
            if (l.size() == 0) return false;
            if (!flag) return true;
        }
        return true;
    }
}

最初にリストをソートします。
backtracking法を使い全体探索します。
リストはソートしてあるので一つずつ足して試していきます。
targetと同じ値が出た場合はresに追加し、
超えてしまった場合は現時点のリストを破棄し次の正数から探し直します。

今日は以上。
よいプログラミング生活を!