LeetCode/validate-ip-address.cc at master · MaskRay/LeetCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
// Validate IP Address
class Solution {
  bool ipv4(string a) {
    int p = 0, t;
    for (int j, i = 0; i < a.size(); i = j+1) {
      for (j = i; j < a.size() && isdigit(a[j]); j++);
      if (i == j || j-i > 3 ||
          j-i > 1 && a[i] == '0' ||
          j < a.size() && a[j] != '.')
        return false;
      t = atoi(a.c_str()+i);
      if (! (0 <= t && t < 256))
        return false;
      p++;
    }
    return a.size() && a.back() != '.' && p == 4;
  }
  bool ipv6(string a) {
    int p = 0, t;
    for (int j, i = 0; i < a.size(); i = j+1) {
      for (j = i; j < a.size() && isxdigit(a[j]); j++);
      if (i == j || j-i > 4 ||
          j < a.size() && a[j] != ':')
        return false;
      p++;
    }
    return a.size() && a.back() != ':' && p == 8;
  }
public:
  string validIPAddress(string IP) {
    if (ipv4(IP)) return "IPv4";
    if (ipv6(IP)) return "IPv6";
    return "Neither";
  }
};