华为oj初级 合法IP

描述
现在IPV4下用一个32位无符号整数来表示，一般用点分方式来显示，点将IP地址分成4个部分，每个部分为8位，表示成一个无符号整数（因此不需要用正号出现），如10.137.17.1，是我们非常熟悉的IP地址，一个IP地址串中没有空格出现（因为要表示成一个32数字）。
现在需要你用程序来判断IP是否合法。

知识点 字符串,循环,链表,队列,栈,查找,搜索,排序,树,图,数组,函数,指针,枚举,位运算,结构体,联合体,文件操作,递归
运行时间限制 10M
内存限制 128
输入
输入一个ip地址
输出
返回判断的结果YES or NO

样例输入 10.138.15.1
样例输出 YES

#include <iostream>  
#include<vector>
#include<algorithm>
#include<string>
using namespace std;
int main(){
    string s;
    cin >> s;
    int cnt = count(s.begin(), s.end(), '.');
    if (cnt != 3){
        cout << "NO" << endl;
        return 0;
    }
    else{
        vector<int> v;
        string str;
        int pos;
        for (int i = 0; i < 4; i++){
            pos = s.find('.');
            str = s.substr(0, pos);
            v.push_back(stoi(str));
            s = s.substr(pos + 1);
        }
        for (int i = 0; i < 4; i++){
            if (v[i] < 0 || v[i]>255){
                cout << "NO" << endl;
                return 0;
            }
        }
        cout << "YES" << endl;
    }
    return 0;
}
#include <stdio.h>  
#include <iostream>
using namespace std;
int main()
{
    int a[4];
    scanf("%d.%d.%d.%d", a, a + 1, a + 2, a + 3);
    for (int i = 0; i < 4; i++){
        if (a[i] > 255 || a[i] < 0){
            cout << "NO";
            return 0;
        }
    }
    cout << "YES";
    return 0;
}