HDOJ--2206IP的计算

最近看了正则表达式，发现用这个来约束IP地址比较方便，只不过可读性不高

IPreg="((2[0-4]\d|25[0-5]|[01]?\d\d?)\.){3}(2[0-4]\d|25[0-5]|[01]?\d\d?)"
自己本身在学习TCP/IP协议，对网际IP协议有些了解，在HDOJ上找到了关于这方面的题目，于是小练习一下

ps:这题目WA了三次，最后发现原因居然是我用scanf输入的缘故，在用scanf("%s",s)的情况下，我输入0（空格）0（空格） 0（空格） 0，一下输出了四个NO，如果输入IP格式不正确，题目要求只要输出一个NO就够了，显然直接wrong了，这是由于scanf遇到空格将我输入的数据直接分成了四个部分，于是我换了gets(s)来实现输入，这样就不会出现上面的情况了。

/*
*判断输入IP地址是否正确
*/
#include<iostream>
#include<stdio.h>
using namespace std;
int check(char c)
{
if(c=='.'||(c>='0' && c<='9'))
return 1;
else
return 0;
}
int main()
{
int a[3],b[4];//数组a来记录.的下标，b记录四部分数据
char s[20];
int i,j;
int count,lens,flag;
while(gets(s)){
lens=strlen(s);
count=0,flag=1;
j=0;
for(i=0;i<lens;i++)
{
if(check(s[i])==0)//不满足条件
{
flag=0;
break;
}
}
for(i=0;i<lens;i++)
{
if(s[i]=='.')
{
count++;//记录.的个数
a[j++]=i;//记录.在字符串数组里的下标
}
}
if(count!=3)
flag=0;//no
else
{
for(i=1;i<3;i++){
if(a[0]==0 || (a[i]-a[i-1])<=1)//.的位置不合理
flag=0;
}
b[0]=b[1]=b[2]=b[3]=0;
//点分十进制的四部分数据
for(i=0;i<a[0];i++)
{
b[0]=b[0]*10+s[i]-48;
}
for(i=a[0]+1;i<a[1];i++)
{
b[1]=b[1]*10+s[i]-48;
}
for(i=a[1]+1;i<a[2];i++)
{
b[2]=b[2]*10+s[i]-48;
}
for(i=a[2]+1;i<lens;i++)
{
b[3]=b[3]*10+s[i]-48;
}
for(i=0;i<4;i++)
{
if(b[i]<0 || b[i]>255)
flag=0;
}
}
if(flag)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}

return 0;
}