华为oj：判断两个IP是否属于同一个子网

/*
描述
子网掩码是用来判断任意两台计算机的IP地址是否属于同一子网络的根据。
子网掩码与IP地址结构相同，是32位二进制数，其中网络号部分全为“1”和主机号部分全为“0”。利用子网掩码可以判断两台主机是否中同一子网中。若两台主机的IP地址分别与它们的子网掩码相“与”后的结果相同，则说明这两台主机在同一子网中。
示例：
I P 地址　 192.168.0.1
子网掩码　 255.255.255.0
转化为二进制进行运算：
I P 地址　11010000.10101000.00000000.00000001
子网掩码　11111111.11111111.11111111.00000000
AND运算
　　　　11000000.10101000.00000000.00000000
转化为十进制后为：
　　　　192.168.0.0

I P 地址　 192.168.0.254
子网掩码　 255.255.255.0

转化为二进制进行运算：
I P 地址　11010000.10101000.00000000.11111110
子网掩码　11111111.11111111.11111111.00000000
AND运算
　　　　　11000000.10101000.00000000.00000000
转化为十进制后为：
　　　　　192.168.0.0
通过以上对两台计算机IP地址与子网掩码的AND运算后，我们可以看到它运算结果是一样的。均为192.168.0.0，所以这二台计算机可视为是同一子网络。

• 功能: 判断两台计算机IP地址是同一子网络。
• 输入参数： String Mask: 子网掩码，格式：“255.255.255.0”；
• String ip1: 计算机1的IP地址，格式：“192.168.0.254”；
• String ip2: 计算机2的IP地址，格式：“192.168.0.1”；
• 
  
  • 返回值： 0：IP1与IP2属于同一子网络； 1：IP地址或子网掩码格式非法； 2：IP1与IP2不属于同一子网络
    *
    public int checkNetSegment(String mask, String ip1, String ip2)
    {
    //在这里实现功能
    return 0;
    }
    输入
    输入子网掩码、两个ip地址
    输出
    得到计算结果
    样例输入 255.255.255.0 192.168.224.256 192.168.10.4
    样例输出 1 */
    //采用while循环的时候，用！=判断条件时候，编译不通过

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

int getPointNum1(string str){
    int count = 0;
    for (int i = 0; i < str.size(); i++)  //掩码、IP的点个数(不连续的点等于3个)  
    if (str[i] == '.' && str[i + 1] != '.')
        count++;
    return count;
}
int getPointNum2(string str){
    int count = 0;
    for (int i = 0; i < str.size(); i++)  //掩码、IP的点个数(所有的点等于3个)  
    if (str[i] == '.')
        count++;
    return count;
}
//分成四段
vector<string> get4segment(string str){
    int i = 0;
    vector<string> v;
    while (i < str.size()){
        int  temp = i;
        while (isdigit(str[temp]))
            temp++;
     v.push_back(str.substr(i,temp-i));
        i = temp + 1;
    }
    return v;
}
//判断值是否在0-255之间
int getFlag(vector<string> str){
    int flag = 0;
    for (int i = 0; i < 4; i++)
        if ((atoi(str[i].c_str()) < 256 )&& (atoi(str[i].c_str()) >= 0))
            flag++;
return flag;
}
void checkNetSegment(string mask, string ip1, string ip2){
    vector<string> v1,v2,v3;
    if (getPointNum1(mask) == getPointNum1(ip1) && getPointNum1(ip1) == getPointNum1(ip2) && getPointNum1(mask) == 3
        && getPointNum2(mask) == getPointNum2(ip1) && getPointNum2(ip1) == getPointNum2(ip2) && getPointNum2(mask) == 3){//判断格式是否正确
            v1=get4segment(mask);
            v2 = get4segment(ip1);
            v3 = get4segment(ip2);
            if (getFlag(v1) == 4 && getFlag(v2) == 4 && getFlag(v3) == 4){//判断格式是否正确
                for (int i = 0; i < 4; i++){
                    if ((atoi(v1[i].c_str())&atoi(v2[i].c_str())) != (atoi(v1[i].c_str())& atoi(v3[i].c_str()))){
                        cout << '2' << endl;
                        return ;
                    }
                }
                cout << '0' << endl;
            }
            else 
                cout << '1' << endl;//格式不正确
    }
    else
        cout << '1' << endl;//格式不正确

}       
int main(){
    string mask1, ip11, ip22;
    cin >> mask1 >> ip11 >> ip22;
    checkNetSegment(mask1,ip11,ip22);
    return 0;
}