python 离散数学 判断单射 双射 满射

# 离散数学 判断单射 双射 满射
def panduan(sets):
    # 单射标志
    danshe = True
    # 满射标志
    manshe = True
    # 将传入对象分隔开,存入两个变量
    set1, set2 = sets
# 判断是否单射
    # 遍历set1,set2中所有数据,是否在 set2,set1中存在映射,判断是否满足单射或者满射
    for i in set1:
        # 单射判断
        # 如果set2中存在两个及以上或不存在与set1中元素对应的映射,则不满足单射条件,跳出循环
        if set2.count(i)!=1:
            danshe = False
            break
    for i in set2:
        # 满射判断
        # 如果set1中不存在一个及以上与set2中元素对应的映射,则不满足满射条件,跳出循环
        if set1.count(i) == 0:
            manshe = False
            break
    if danshe and manshe:
        print("双射")
    elif danshe:
        print("单射")
    elif manshe:
        print("满射")
    else:
        print("既不是单射也不是满射")


if __name__ == '__main__':
    a = eval(input('请输入两个集合或者列表或者元组以","或" "隔开').replace(' ', ","))
    panduan(a)