刷题使我快乐，满脸开心.jpg

• 来源：力扣（LeetCode）
• 链接：https://leetcode.cn/problems/validate-ip-address/
• 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目

给定一个字符串 queryIP。如果是有效的 IPv4 地址，返回 "IPv4" ；如果是有效的 IPv6 地址，返回 "IPv6" ；如果不是上述类型的 IP 地址，返回 "Neither" 。

有效的IPv4地址 是 “x1.x2.x3.x4” 形式的IP地址。 其中 0 <= xi <= 255 且 xi 不能包含 前导零。例如: “192.168.1.1” 、 “192.168.1.0” 为有效IPv4地址， “192.168.01.1” 为无效IPv4地址; “192.168.1.00” 、 “192.168@1.1” 为无效IPv4地址。

一个有效的IPv6地址 是一个格式为“x1:x2:x3:x4:x5:x6:x7:x8” 的IP地址，其中:

1 <= xi.length <= 4 xi 是一个 十六进制字符串 ，可以包含数字、小写英文字母( 'a' 到 'f' )和大写英文字母( 'A' 到 'F' )。 在 xi 中允许前导零。 例如 "2001:0db8:85a3:0000:0000:8a2e:0370:7334" 和 "2001:db8:85a3:0:0:8A2E:0370:7334" 是有效的 IPv6 地址，而 "2001:0db8:85a3::8A2E:037j:7334" 和 "02001:0db8:85a3:0000:0000:8a2e:0370:7334" 是无效的 IPv6 地址。

示例 1：

输入：queryIP = "172.16.254.1" 
输出："IPv4" 
解释：有效的 IPv4 地址，返回 "IPv4" 

示例 2：

输入：queryIP = "2001:0db8:85a3:0:0:8A2E:0370:7334" 
输出："IPv6" 
解释：有效的 IPv6 地址，返回 "IPv6" 

示例 3：

输入：queryIP = "256.256.256.256" 
输出："Neither" 
解释：既不是 IPv4 地址，又不是 IPv6 地址

提示：

• queryIP 仅由英文字母，数字，字符 '.' 和 ':' 组成。

思路

属于是基础知识了，没啥太多需要说的，就一个 fail fast 感觉是一个比较小的优化

代码

func validIPAddress(queryIP string) string {
    hasPot := false
    hasColon := false
    hasAF := false
    for _, b := range queryIP {
        if b == '.' {
            hasPot = true
        } else if b == ':' {
            hasColon = true
        } else if (b-'0' >= 'a'-'0' && b-'0' <= 'f'-'0') || (b-'0' >= 'A'-'0' && b-'0' <= 'F'-'0') {
            hasAF = true
        } else if (b-'0' >= 'g'-'0' && b-'0' <= 'z'-'0') || (b-'0' >= 'G'-'0' && b-'0' <= 'Z'-'0') {
            return "Neither"
        }
    }

    if hasPot && !hasAF && validIPv4(queryIP) {
        return "IPv4"
    }
    if hasColon && hasAF && validIPv6(queryIP) {
        return "IPv6"
    }
    return "Neither"
}

func validIPv4(queryIP string) bool {
    numStrs := strings.Split(queryIP, ".")
    if len(numStrs) != 4 {
        return false
    }

    for _, numStr := range numStrs {
        if len(numStr) == 0 || (len(numStr) > 1 && numStr[0] == '0') {
            return false
        }
        num, _ := strconv.Atoi(numStr)
        if num > 255 {
            return false
        }
    }
    return true
}

func validIPv6(queryIP string) bool {
    numStrs := strings.Split(queryIP, ":")
    if len(numStrs) != 8 {
        return false
    }
    for _, numStr := range numStrs {

        if len(numStr) > 4 || len(numStr) == 0 {
            return false
        }
    }
    return true
}