牛客-华为机试-较难-字符串-HJ18识别有效的IP地址和掩码并进行分类统计-C++

描述

请解析IP地址和对应的掩码，进行分类识别。要求按照A/B/C/D/E类地址归类，不合法的地址和掩码单独归类。

所有的IP地址划分为 A,B,C,D,E五类

A类地址从1.0.0.0到126.255.255.255;

B类地址从128.0.0.0到191.255.255.255;

C类地址从192.0.0.0到223.255.255.255;

D类地址从224.0.0.0到239.255.255.255；

E类地址从240.0.0.0到255.255.255.255

私网IP范围是：

从10.0.0.0到10.255.255.255

从172.16.0.0到172.31.255.255

从192.168.0.0到192.168.255.255

子网掩码为二进制下前面是连续的1，然后全是0。（例如：255.255.255.32就是一个非法的掩码）

（注意二进制下全是1或者全是0均为非法子网掩码）

注意：

1. 类似于【0.*.*.*】和【127.*.*.*】的IP地址不属于上述输入的任意一类，也不属于不合法ip地址，计数时请忽略

2. 私有IP地址和A,B,C,D,E类地址是不冲突的

输入描述：

多行字符串。每行一个IP地址和掩码，用~隔开。

请参考帖子https://www.nowcoder.com/discuss/276处理循环输入的问题。

输出描述：

统计A、B、C、D、E、错误IP地址或错误掩码、私有IP的个数，之间以空格隔开。

示例1

输入：

10.70.44.68~255.254.255.0
1.0.0.1~255.0.0.0
192.168.0.2~255.255.255.0
19..0.~255.255.255.0

输出：

1 0 1 0 0 2 1

说明：

10.70.44.68~255.254.255.0的子网掩码非法，19..0.~255.255.255.0的IP地址非法，所以错误IP地址或错误掩码的计数为2；
1.0.0.1~255.0.0.0是无误的A类地址；
192.168.0.2~255.255.255.0是无误的C类地址且是私有IP；
所以最终的结果为1 0 1 0 0 2 1        

示例2

输入：

0.201.56.50~255.255.111.255
127.201.56.50~255.255.111.255

输出：

0 0 0 0 0 0 0

说明：

类似于【0.*.*.*】和【127.*.*.*】的IP地址不属于上述输入的任意一类，也不属于不合法ip地址，计数时请忽略       

代码

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

int main() {
	string in;
	string ip_start,ip_end;
	int A = 0, B = 0, C = 0, D = 0, E = 0, ERR = 0, PRI = 0;
	while (getline(cin, in)) {
		if (in.empty())
			break;
		int s = in.find("~");
		ip_start = in.substr(0, s);
		ip_end = in.substr(s + 1, in.length() - s - 1);
		int d1, d2, d3;
		string s1, s2, s3, s4;
		int i1, i2, i3, i4;
		d1 = ip_start.find(".");
		s1 = ip_start.substr(0, d1);
		i1 = stoi(s1);
		if (i1 == 0 || i1 == 127) {
			continue;
		}
		d1 = ip_end.find(".");
		d2 = ip_end.find(".", d1 + 1);
		d3 = ip_end.find(".", d2 + 1);
		s1 = ip_end.substr(0, d1);
		s2 = ip_end.substr(d1 + 1, d2 - d1 - 1);
		s3 = ip_end.substr(d2 + 1, d3 - d2 - 1);
		s4 = ip_end.substr(d3 + 1);
		i1 = stoi(s1);
		i2 = stoi(s2);
		i3 = stoi(s3);
		i4 = stoi(s4);
		long long x = (long long)i1 * 256 * 256 * 256 + (long long)i2 * 256 * 256 + (long long)i3 * 256 + (long long)i4;
		bool one = false;
		int r;
		bool maskErr = false;
		if (i1 == 255 && i2 == 255 && i3 == 255 && i4 == 255) {
			ERR++;
			maskErr = true;
		}
		if (!maskErr) {
			while (x != 0) {
				r = x % 2;
				if (one) {
					if (r == 0) {
						ERR++;
						maskErr = true;
						break;
					}
				}
				if (r == 1 && one == false) {
					one = true;
				}
				//cout << x << " " << r << endl;
				x /= 2;
			}
		}
		if (maskErr) {
			continue;
		}
		d1 = ip_start.find(".");
		d2 = ip_start.find(".", d1+1);
		d3 = ip_start.find(".", d2+1);
		s1 = ip_start.substr(0, d1);
		s2 = ip_start.substr(d1+1, d2-d1-1);
		s3 = ip_start.substr(d2+1, d3-d2-1);
		s4 = ip_start.substr(d3+1);
		if (s1.empty() || s2.empty() || s3.empty() || s4.empty()) {
			ERR++;
			continue;
		}
		 i1 = stoi(s1);
		 i2 = stoi(s2);
		 i3 = stoi(s3);
		 i4 = stoi(s4);
		if (i1 > 255 || i1 < 0 || i2>255 || i2 < 0 || i3>255 || i3 < 0 || i4>255 || i4 < 0) {
			ERR++;
			continue;
		}
		if (i1 > 0 && i1 < 127) {
			if (i1 == 10)
				PRI++;
			A++;
		}else if (i1 > 127 && i1 < 192) {
			if (i1 == 172 && i2 >= 16 && i2 <= 31)
				PRI++;
			B++;	
		}
		else if (i1 >= 192 && i1 <= 223) {
			if (i1 == 192 && i2 == 168)
				PRI++;
			C++;
		}
		else if (i1 >= 224 && i1 <= 239) 
			D++;
		else if (i1 >= 240 && i1 <= 255)
			E++;
	}
	cout << A << " " << B << " " << C << " " << D << " " << E << " " << ERR << " " << PRI << endl;
    return 0;
}