【leetcode】468 验证IP地址（字符串）

题目链接：https://leetcode-cn.com/problems/validate-ip-address/

题目描述

编写一个函数来验证输入的字符串是否是有效的 IPv4 或 IPv6 地址。

IPv4 地址由十进制数和点来表示，每个地址包含4个十进制数，其范围为 0 - 255， 用(".")分割。比如，172.16.254.1；

同时，IPv4 地址内的数不会以 0 开头。比如，地址 172.16.254.01 是不合法的。

IPv6 地址由8组16进制的数字来表示，每组表示 16 比特。这些组数字通过 (":")分割。比如, 2001:0db8:85a3:0000:0000:8a2e:0370:7334 是一个有效的地址。而且，我们可以加入一些以 0 开头的数字，字母可以使用大写，也可以是小写。所以， 2001:db8:85a3:0:0:8A2E:0370:7334 也是一个有效的 IPv6 address地址 (即，忽略 0 开头，忽略大小写)。

然而，我们不能因为某个组的值为 0，而使用一个空的组，以至于出现 (:? 的情况。 比如， 2001:0db8:85a3::8A2E:0370:7334 是无效的 IPv6 地址。

同时，在 IPv6 地址中，多余的 0 也是不被允许的。比如， 02001:0db8:85a3:0000:0000:8a2e:0370:7334 是无效的。

说明: 你可以认为给定的字符串里没有空格或者其他特殊字符。

示例 1:

输入: "172.16.254.1"

输出: "IPv4"

解释: 这是一个有效的 IPv4 地址, 所以返回 "IPv4"。

示例 2:

输入: "2001:0db8:85a3:0:0:8A2E:0370:7334"

输出: "IPv6"

解释: 这是一个有效的 IPv6 地址, 所以返回 "IPv6"。

示例 3:

输入: "256.256.256.256"

输出: "Neither"

解释: 这个地址既不是 IPv4 也不是 IPv6 地址。

思路

class Solution {
public:
    string validIPAddress(string IP) {
        if(isValidIPV4(IP)) return "IPv4";
        if(isValidIPV6(IP)) return "IPv6";
        return "Neither";
    }

private:
    void split(const string s, vector<string>& vs, const char delim=' '){
        istringstream iss(s);
        string tmp;
        while (getline(iss,tmp, delim)){
            vs.push_back(tmp);
        }
        if (!s.empty() && s.back() == delim) vs.push_back({});
        //加这句的原因是getline不会识别最后一个delim,避免误判"172.16.254.1.","2001:0db8:85a3:0:0:8A2E:0370:7334:"之类的情况
    }

    // 是否为有效IPV4地址
    bool isValidIPV4(string IP){
        vector<string> vs;
        split(IP, vs, '.');
        if(vs.size() != 4)
            return false;

        for (auto &v: vs) {
            if(v.empty() ||(v.size() > 1 && v[0] == '0') || v.size() > 3)
                return false;
            // 每个字符都是数字
            for(auto c:v){
                if(!isdigit(c))
                    return false;
            }
            int n = stoi(v);
            if(n<0 || n > 255)
                return false;
        }
        return true;
    }

    // 是否为有效IPV6地址
    bool isValidIPV6(string IP){
        vector<string> vs;
        split(IP, vs, ':');
        if(vs.size()!=8)
            return false;

        for (auto &v : vs) {
            if(v.empty() || v.size()>4) return false; 
            for (auto c:v) {
                if(!(isdigit(c) || (c>='a' && c<='f') ||(c>='A' && c<='F')))
                    return false;
            }
        }
        return true;
    }
};