代码随想录Day21 回溯 LeetCodeT216 组合总和III LeetCode T17电话号码的字母总和

LeetCode T216 组合总和III

题目链接 216. 组合总和 III - 力扣（LeetCode）

 

题目思路

经过昨天组合问题的思考,这道题的难度也就降低了,这道题其实相较于组合问题就是多了一个限制,要求我们元素的和是n,元素个数是k,这里我们仍然是使用回溯三部曲来完成任务

这里的path和result集合我们定义为全局变量,避免调用参数过多.

回溯三部曲

1.确定参数和返回值

我们肯定需要一个sum来统计目前path中的和,一个目标和targetSum,一个开始的位置startIndex,以及元素个数k

public void backtracking(int targetSum, int k, int startIndex, int sum)

2.确定终止条件

如果sum>targetSum,肯定是不符合条件的,直接返回,当两者相等且path的元素个数与限制的元素个数相等时,就是我们收割结果的时候,此时我们将path添加到result中

        if(sum>targetSum)
        {
            return;
        }
        if(path.size() == k)
        {
            if(targetSum == sum)
            {
                result.add(new ArrayList(path));
                return;
            }
        }

3.确定for循环

这里包含昨天的剪枝,不过多赘述

for(int i = startIndex;i<=9-(k-path.size())+1;i++)
        {
            
            path.add(i);
            sum+=i;
            backtracking(targetSum,k,i+1,sum);
            path.remove(path.size()-1);
            sum-=i;
        }

题目代码

class Solution {
    List path = new LinkedList<>();
    List> result = new ArrayList<>();
    public List> combinationSum3(int k, int n) {
        backtracking(n,k,1,0);
        return result;
 
    }
    public void backtracking(int targetSum, int k, int startIndex, int sum)
    {
        if(sum>targetSum)
        {
            return;
        }
        if(path.size() == k)
        {
            if(targetSum == sum)
            {
                result.add(new ArrayList(path));
                return;
            }
        }
        for(int i = startIndex;i<=9-(k-path.size())+1;i++)
        {
            
            path.add(i);
            sum+=i;
            backtracking(targetSum,k,i+1,sum);
            path.remove(path.size()-1);
            sum-=i;
        }
        
    }
}

 

LeetCode T17电话号码的字母总和

题目链接17. 电话号码的字母组合 - 力扣（LeetCode）

 

题目思路

回溯法--解决n层循环问题

这题我们首先想的肯定是digits字符串的长度决定我们for循环的次数,但是我们控制不了它的取值,这就又是一个回溯算法的经典题目,但是这里我们就得知道怎么把2-9的数字和这些字符串来形成一个映射呢?

我们可以使用map来完成,但是这里显然使用一个二维数组的形式更为简单

            "", // 0
            "", // 1
            "abc", // 2
            "def", // 3
            "ghi", // 4
            "jkl", // 5
            "mno", // 6
            "pqrs", // 7
            "tuv", // 8
            "wxyz", // 9

这里我们将0和1都定义为空,这样方便映射九宫格按键的字符串

主函数中我们只需要一次判空,再创建这个字符串数组,然后直接调用回溯函数即可

下面涉及了多次的字符串拼接操作,我们可以直接创建一个StringBuilder来操作

下面我们讨论回溯函数的写法,依然是经典的回溯三部曲

1.确定回溯函数

这里我们需要的参数,首先我们需要digits作为题目传入的参数来确定我们回溯的深度,需要二维数组Stringnum来作为每次查找的参数,最后需要有一个下标num指向digits知道你遍历到哪里了.

public void backtracking(String digits,String[] numString,int num)

2.确定终止条件

这里的终止条件就是当digits被遍历结束了,就结束了,也就到了我们收割结果的时候了

        if(num == digits.length())
         {
             list.add(sb.toString());
             return;
         }

3.确定for循环回溯过程

在回溯之前,我们需要获取num对应的映射字符串

String s = numString[digits.charAt(num) - '0'];

最后for循环,这里不需要前面文章中的startIndex是因为本次递归迭代是在两个字符串中,前文是因为去重所以每次需要控制startIndex的位置.

    for(int i = 0;i
         {
             sb.append(s.charAt(i));
             backtracking(digits,numString,num+1);
             sb.deleteCharAt(sb.length()-1);
 
         }

题目代码

class Solution {
    List list = new ArrayList<>();
    public List letterCombinations(String digits) {
        if(digits == null || digits.length() == 0)
        {
            return list;
        }
        String[] numString = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
        backtracking(digits,numString,0);
        return list;
 
    }
     StringBuilder sb = new StringBuilder();
     public void backtracking(String digits,String[] numString,int num)
     {
         if(num == digits.length())
         {
             list.add(sb.toString());
             return;
         }
         String s = numString[digits.charAt(num) - '0'];
         for(int i = 0;i
         {
             sb.append(s.charAt(i));
             backtracking(digits,numString,num+1);
             sb.deleteCharAt(sb.length()-1);
 
         }
 
     }
}