用python实现了一下:甲乙两人互猜数字(数理逻辑)

今天在园子里看到博客:超难面试题:甲乙两人互猜数字(数理逻辑)。然后琢磨了半天,写了个Python程序实现算法,我得出来的结果是1,6或1,8或35,42的组合,

不知道是否正确,请高人指点?

下面列出实现算法代码(代码按照题目文字一步步进行的,没有优化):

# -*- coding: utf-8  -*-
#!/usr/local/bin/python

def is_prime2(number):
    '''
    判断数字是否是特殊质数(数学上质数和分解因子相同的数)。分解因子相同的数,如:4,9等。
    '''
    _is_primenum = True
    for item in range(2,number,1):#不考虑1乘以number
        if number % item == 0:
            if (number/item) == item: #两个因子不能相同
                continue
            _is_primenum = False
            break
    return _is_primenum

def combination(data, t, vl):
    '''
    计算一组数的组合
    '''
    _kk = len(data)
    _ret_list = []
    for _i in range(_kk-t+1):
        vl.append(data[_i])
        new_data = data[_i+1:]
        if t-1 == 1:
            for _j in range(len(new_data)):
                _data_str = ','.join(map(str,vl))+','+str(new_data[_j])
                _ret_list.append(_data_str)
        else:
            _ret_list.extend(combination(new_data, t-1, vl))
        vl.pop()
    return _ret_list
    
def generate_combine(num_list, pick_num):
    _v1 = [] # temp variable.
    return combination(num_list, pick_num, _v1)
    
def filter_1_a(_combination_list):
    '''
    甲:“我不知道这两个数是什么”
    
    遍历所有组合,过滤掉能判断出结果的组合
    '''
    _result_dict = {} # 存储key为乘积的字典,value是组合的数组。
    _new_combination_list = [] #存储还有效的组合
    
    #由于甲不知道这两个数,可以肯定乘积不是质数,排除这类组合。
    for item in _combination_list: 
        a, b = item.split(',',1)
        _product = int(a) * int(b) #甲得到的数字
        if not is_prime2(_product): 
            if _product in _result_dict:
                _result_dict[_product] +=[item]
            else:
                _result_dict[_product] =[item]
    
    _new_result_dict = {} #便于理解,使用一个新变量存字典值
    #由于甲不知道,表示value肯定要大于1种组合。
    #比如[min=1,max=10], 乘积是15的组合:3*5一种,这样的组合应该排除掉。
    for key,value in _result_dict.iteritems():
        if len(value) > 1: 
            _new_result_dict[key] = value
            _new_combination_list.extend(value)
                
    return _new_combination_list, _new_result_dict

def filter_1_b(_combination_list):
    '''
    乙:“我也不知道”
    
    遍历所有组合,过滤掉能判断出结果的组合
    '''
    _new_combination_list = [] #存储还有效的组合
    _result_dict = {}  # 存储key为两值相加的字典,value是组合的数组。

    for item in _combination_list:
        a, b = item.split(',',1)
        _sum = int(a) + int(b) #乙得到的数字
        if _sum in _result_dict:
            _result_dict[_sum] +=[item]
        else:
            _result_dict[_sum] =[item]
            
    _new_result_dict = {}
    #由于乙不知道,表示value肯定要大于1种组合。
    for key,value in _result_dict.iteritems():
        if len(value) > 1: 
            _new_result_dict[key] = value
            _new_combination_list.extend(value)
    return _new_combination_list, _new_result_dict

def filter_2_a(_combination_list, product_dict, sum_dict):
    '''
    甲:“那我知道了”
    
    遍历所有组合,过滤掉能判断出结果的组合
    '''
    _new_prodcut_dict = {}
    #在乘积字典中,去除多余的组合
    for key,value in product_dict.iteritems():
        _new_value = []
        for item in value:
            if item in _combination_list:
                _new_value.append(item)
        if _new_value:
            _new_prodcut_dict[key] = _new_value
    
    # 去除已经不在相加字典中的组合,
    for key,value in _new_prodcut_dict.items():
        _new_value = []
        for item in value:
            a, b = item.split(',',1)
            _sum = int(a) + int(b) #乙得到的数字
            if _sum in sum_dict:
                _new_value.append(item)
        _new_prodcut_dict[key] = _new_value
        
    # 由于甲知道,表示肯定等于1种组合。
    _new_combination_list = [] #存储还有效的组合
    _result_dict = {}
    for key,value in _new_prodcut_dict.iteritems():
        if len(value) == 1: 
            _result_dict[key] = value
            _new_combination_list.extend(value)
    return _new_combination_list, _result_dict

def filter_2_b(_combination_list, sum_dict):
    '''
    乙:“那我也知道了”
    
    '''
    _new_sum_dict = {}
    #在相加字典中,去除多余的组合
    for key,value in sum_dict.iteritems():
        _new_value = []
        for item in value:
            if item in _combination_list:
                _new_value.append(item)
        if _new_value:
            _new_sum_dict[key] = _new_value
    return _new_sum_dict
    
if __name__ == '__main__':
    print "start..."
    _min_num=1
    _max_num=50
    
    #产生数组
    _num_list = range(_min_num,_max_num+1,1) #[1,2,3...,49,50]
    
    #生成Cn2的所有组合列表,如:['1,6', '2,3', '2,6', '3,4'...]
    _combination_list = generate_combine(_num_list, 2)
    
    # 甲:“我不知道这两个数是什么”
    _combination_list, _result_product_dict = filter_1_a(_combination_list)
    
    # 乙:“我也不知道”
    _combination_list, _result_sum_dict = filter_1_b(_combination_list)
    
    # 甲:“那我知道了”
    _new_combination_list, _result_product_dict = filter_2_a(_combination_list, _result_product_dict, _result_sum_dict)
    print u"最终的组合:", _new_combination_list
    print u"最终的乘积字典:", _result_product_dict
    
    # 乙:“那我也知道了”
    _result_sum_dict = filter_2_b(_new_combination_list, _result_sum_dict)
    print u"最终的相加字典:", _result_sum_dict
    
    
    print "end..."
    
原文地址:https://www.cnblogs.com/lxg226/p/3331736.html