LeetCode:93 复原IP地址 回溯(递归)

题目描述

给定一个只包含数字的字符串,复原它并返回所有可能的 IP 地址格式。

示例:
输入: “25525511135”
输出: [“255.255.11.135”, “255.255.111.35”]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/restore-ip-addresses
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

因为是IPV4的地址,固定加三个点,划分为4个部分,可以递归地去填写这四个部分

第一部分 . 第二部分 . 第三部分 . 第四部分 

对第一个部分,给定数字长度为n,我们有三种填法:

  • 用一个数字:则用剩下的n-1个数字去填剩下三个部分
  • 用两个数字:则用剩下的n-2个数字去填剩下三个部分
  • 用三个数字:则用剩下的n-3个数字去填剩下三个部分

每一次都如此地递归去填写即可
注意数字的个数n不能超过部分数的三倍,比如不能用7个数字填写两部分,因为IP地址最长就是百位整数

此外还要留意前导0,不能出现开头为0的情况

还要保证三位数的范围在0-255之间

代码

class Solution {
public:
    #define c2i(c) ((int)(c-'0'))
    vector<string> ans;
    void dfs(string s, int l, int num, string Ans)
    {
        if(num==0 && l>=s.length())
        {
            Ans.pop_back();
            ans.push_back(Ans);
        }
        if(l<s.length() && s.length()-l<=num*3)
        {
        	// 用一个数字
            if(l<s.length())
            {
                Ans += s[l], Ans += '.';
                dfs(s, l+1, num-1, Ans);
                Ans.pop_back(), Ans.pop_back();
            }
          	// 用两个数字
            if(l+1<s.length() && s[l]!='0')
            {
                Ans += s[l], Ans += s[l+1], Ans += '.';
                dfs(s, l+2, num-1, Ans);
                Ans.pop_back(), Ans.pop_back(), Ans.pop_back();
            }
            // 用三个数字
            if(l+2<s.length() && c2i(s[l])*100+c2i(s[l+1])*10+c2i(s[l+2])<=255 && s[l]!='0')
            {
                Ans += s[l], Ans += s[l+1], Ans += s[l+2], Ans += '.';
                dfs(s, l+3, num-1, Ans);
                Ans.pop_back(), Ans.pop_back(), Ans.pop_back(), Ans.pop_back();
            }
        } 
    }
    vector<string> restoreIpAddresses(string s)
    {
        dfs(s, 0, 4, "");
        return ans;
    }
};

你可能感兴趣的:(LeetCode)