【探索-字节跳动】复原IP地址

【探索-字节跳动】复原IP地址_第1张图片

可以使用深搜,看每一次的组合是不是符合条件。

public List<String> restoreIpAddresses(String s) {
    int[] segsStart = new int[4];
    dfs(s, segsStart, 0);
    return result;
}

public void dfs(String s, int[] segsStart, int segmentIndex) {
    if (segmentIndex >= 3) {
        String str4 = s.substring(segsStart[3]);
        if (isLessOrEqual255(str4)) {
            String builder = s.substring(segsStart[0], segsStart[1]) + "." +
                    s.substring(segsStart[1], segsStart[2]) + "." +
                    s.substring(segsStart[2], segsStart[3]) + "." +
                    str4;
            result.add(builder);
        }
        return;
    }
    int curStart = segsStart[segmentIndex];
    int curEnd = curStart + 3;
    for (int i = curStart; i < curEnd && i < s.length(); i++) {
        if (isLessOrEqual255(s.substring(curStart, i + 1))) {
            segsStart[segmentIndex + 1] = i + 1;
            dfs(s, segsStart, segmentIndex + 1);
        }
    }
}

// 判断字符串代表的 IP 段是不是符合要求
private boolean isLessOrEqual255(String s) {
    // s 的长度大于 1 小于 4
    // 以 0 开头使只能是 "0"
    // 长度小于 3 且不为 0 开头
    // 长度为 3 时要小于 255 且不为 0 开头
    if (s.length() < 1) return false;
    else if (s.startsWith("0") && s.length() == 1) return true;
    else if (s.length() < 3 && !s.startsWith("0")) return true;
    else if (s.length() == 3 && !s.startsWith("0") && s.compareTo("255") <= 0) return true;
    else return false;
}

关键就是判断字符串代表的 IP 段是不是符合要求,在判断的时候要全面。

你可能感兴趣的:(LeetCode,渣渣的算法历练之路)