【LeetCode】93. Restore IP Addresses 解题报告（Python）

【LeetCode】93. Restore IP Addresses 解题报告（Python）

标签（空格分隔）： LeetCode

作者： 负雪明烛
id： fuxuemingzhu
个人博客： http://fuxuemingzhu.me/

---

题目地址：https://leetcode.com/problems/restore-ip-addresses/description/

题目描述：

Given a string containing only digits, restore it by returning all possible valid IP address combinations.

Example:

Input: "25525511135"
Output: ["255.255.11.135", "255.255.111.35"]

题目大意

题目中给了一个仅有数字组成的字符串，要求这个字符串能构成的合法IP组合。

解题方法

我是按照Tag刷的，当然知道这个题是回溯法了。。其实只要看到所有的组合，一般都是用回溯。

第一遍超时，原因是没有找到合理的剪枝！！这就是回溯法最难的地方：剪枝！

当然了，看出了测试用例是一个特别长的由1组成的字符串，仅仅这一个测试用例超时，所以我加上了len(s)和12的判断就ok了。所以有了下面的版本：

代码如下：

class Solution(object):
    def restoreIpAddresses(self, s):
        """
        :type s: str
        :rtype: List[str]
        """
        if len(s) > 12:
            return []
        res = []
        self.dfs(s, [], res)
        return res

    def dfs(self, s, path, res):
        if not s and len(path) == 4:
            res.append('.'.join(path))
            return
        for i in range(1, 4):
            if i > len(s):
                continue
            number = int(s[:i])
            if str(number) == s[:i] and number <= 255:
                self.dfs(s[i:], path + [s[:i]], res)

看到了题目中有别的同学的剪枝方法特别好，那就是每次dfs的时候都去检查一下所有的字符串的长度是不是能满足在最多4个3位数字组成，果然速度提升了很多：

class Solution(object):
    def restoreIpAddresses(self, s):
        """
        :type s: str
        :rtype: List[str]
        """
        res = []
        self.dfs(s, [], res)
        return res

    def dfs(self, s, path, res):
        if len(s) > (4 - len(path)) * 3:
            return
        if not s and len(path) == 4:
            res.append('.'.join(path))
            return
        for i in range(min(3, len(s))):
            curr = s[:i+1]
            if (curr[0] == '0' and len(curr) >= 2) or int(curr) > 255:
                continue
            self.dfs(s[i+1:], path + [s[:i+1]], res)

日期

2018 年 6 月 11 日 ———— 今天学科三在路上跑的飞快～