回溯算法题（7）复原IP地址

目录

复原IP地址

描述

示例 1

示例 2

示例 3

提示

方法一：回溯

方法二：回溯优化

---

复原IP地址

描述

有效 IP 地址 正好由四个整数（每个整数位于 0 到 255 之间组成，且不能含有前导 0），整数之间用 '.' 分隔。

例如："0.1.2.201" 和 "192.168.1.1" 是 有效 IP 地址，但是 "0.011.255.245"、"192.168.1.312" 和 "192.168@1.1" 是 无效 IP 地址。

给定一个只包含数字的字符串 s ，用以表示一个 IP 地址，返回所有可能的有效 IP 地址，这些地址可以通过在 s 中插入 '.' 来形成。你 不能 重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。

示例 1

输入：s = "25525511135"
输出：["255.255.11.135","255.255.111.35"]

示例 2

输入：s = "0000"
输出：["0.0.0.0"]

示例 3

输入：s = "101023"
输出：["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]

提示

• 1 <= s.length <= 20
• s 仅由数字组成

方法一：回溯

class Solution {
    private List<String> res = new ArrayList<>();
    private List<String> cur = new ArrayList<>();

    public List<String> restoreIpAddresses(String s) {
        backTrace(s, 0);
        return res;
    }

    public void backTrace(String str, int index) {
        if (index < 4 && "".equals(str)) return;
        if (index == 4) {
            if ("".equals(str) && cur.size() == 4) {
                res.add(cur.get(0) + "." + cur.get(1) + "." + cur.get(2) + "." + cur.get(3));
            }
            return;
        }
        if ('0' == str.charAt(0)) {
            cur.add("0");
            backTrace(str.substring(1, str.length()), index + 1);
            cur.remove(index);
        }else{
            int size = Math.min(3, str.length());
            for (int i = 0; i < size; i++) {
                String temp = str.substring(0, i + 1);
                if (Integer.parseInt(temp) > 255) return;
                cur.add(temp);
                backTrace(str.substring(i + 1, str.length()), index + 1);
                cur.remove(index);
            }
        }
    }
}

方法二：回溯优化

class Solution {
    static final int SEG_COUNT = 4;
    List<String> ans = new ArrayList<String>();
    int[] segments = new int[SEG_COUNT];

    public List<String> restoreIpAddresses(String s) {
        segments = new int[SEG_COUNT];
        dfs(s, 0, 0);
        return ans;
    }

    public void dfs(String s, int segId, int segStart) {
        // 如果找到了 4 段 IP 地址并且遍历完了字符串，那么就是一种答案
        if (segId == SEG_COUNT) {
            if (segStart == s.length()) {
                StringBuffer ipAddr = new StringBuffer();
                for (int i = 0; i < SEG_COUNT; ++i) {
                    ipAddr.append(segments[i]);
                    if (i != SEG_COUNT - 1) {
                        ipAddr.append('.');
                    }
                }
                ans.add(ipAddr.toString());
            }
            return;
        }

        // 如果还没有找到 4 段 IP 地址就已经遍历完了字符串，那么提前回溯
        if (segStart == s.length()) {
            return;
        }

        // 由于不能有前导零，如果当前数字为 0，那么这一段 IP 地址只能为 0
        if (s.charAt(segStart) == '0') {
            segments[segId] = 0;
            dfs(s, segId + 1, segStart + 1);
        }

        // 一般情况，枚举每一种可能性并递归
        int addr = 0;
        for (int segEnd = segStart; segEnd < s.length(); ++segEnd) {
            addr = addr * 10 + (s.charAt(segEnd) - '0');
            if (addr > 0 && addr <= 0xFF) {
                segments[segId] = addr;
                dfs(s, segId + 1, segEnd + 1);
            } else {
                break;
            }
        }
    }
}