lt-468. 验证IP地址

编写一个函数来验证输入的字符串是否是有效的 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 地址。

该题需要充分考虑各种情况，刚开始也只是通过了63 / 79，然后利用测试用例一点一点完善的（逃。
建议编写多个函数，例如，isValidIPv4Num用于验证ipv4中的数字是否符合题目的要求，若都写在同一个函数内部极易混淆，且代码行数过长。

#include <iostream>
#include <string>
using namespace std;

bool isValidIPv4Num(string str) {
    for (char ch : str) {
        if(!isdigit(ch)) {
            return false;
        }
    }
    if (atoi(str.c_str()) > 255) {
        return false;
    }
    return true;
}

bool isValidIPv4(string IP) {
    if (IP.size() > 15) {
        return false;
    }
    string subStr;
    int pointCnt = 4;
    while (pointCnt) {
        subStr.clear();
        size_t idx= IP.find('.');
        subStr = IP.substr(0, idx);
        if (subStr.empty() || (subStr.size() >= 2 &&subStr[0] == '0')) {
            return false;
        }
        if (!isValidIPv4Num(subStr)) {
            return false;
        }
        IP = IP.substr(idx + 1);
//        cout << "IP sub : " << IP << " pointCnt : " << pointCnt << endl;
        pointCnt--;
        if (pointCnt > 1 && IP.find('.') == -1  || pointCnt == 1 && IP.find('.') != -1) {
            return false;
        }
    }
    return true;
}
bool isValidIPv6Num(string str) {
//    cout << atoi(str.c_str()) << endl;
    if (str.size() > 4) {
        return false;
    }
    for (char ch : str) {
        if (ch > 'F' && ch <= 'Z' || ch > 'f' && ch <= 'z') {
            return false;
        }
        if (!isdigit(ch) && !isalpha(ch)) {
            return false;
        }
    }
    return true;
}
bool isValidIPv6(string IP) {
    if(IP.find("::") != -1 || IP.size() < 15) {
        return false;
    }
    int pointCnt = 8;
    string subStr;
    while (pointCnt) {
        subStr.clear();
        size_t idx= IP.find(':');

        subStr = IP.substr(0, idx);
//        cout << "subStr : " << subStr << endl;
        if (!isValidIPv6Num(subStr)) {
            return false;
        }
        IP = IP.substr(idx + 1);
//        cout << "IP sub : " << IP << " pointCnt : " << pointCnt << endl;
        pointCnt--;
        if (pointCnt > 1 && IP.find(':') == -1  || pointCnt == 1 && IP.find(':') != -1) {
            return false;
        }
    }
    return true;
}

string validIPAddress(string IP) {
    if(IP.empty() || IP.size() < 7) {
        return "Neither";
    }
    bool isIPv4 = isValidIPv4(IP);
    if (isIPv4) {
        return "IPv4";
    }
    bool isIPv6 = isValidIPv6(IP);
    if (isIPv6) {
        return "IPv6";
    }
    return "Neither";
}
int main() {
    string str = "12.12.12.12";
    string res = validIPAddress(str);
    cout << res << endl;
}