题解 | 合法IP

import sys
a=input().split('.')

def func(a):
    flag=0
    if len(a)==4:
        for i in a:
            if i :
                if int(i)>255 or int(i)<0:
                    flag+=1
                if str(int(i))!=i:
                    return ('NO')
            else:
                return ('NO')
    else:
        return ('NO')
    if(flag==0):
        return( 'YES')
    
    
    return ('NO')
print(func(a))

这也是中等难度吗？这个没难度吧。