将兔子均匀的放入笼子;保证每个笼子中的兔子重量一致
作者:互联网
算法题;
题目介绍:
某朋友去面试,面试官出的面试题;
题目: 现在需要将总数和各自重量都已知的兔子放进 n 个笼子中,将所有兔子全部放进去后,最终每个笼子的重量需要相同 请实现一个函数,输入 包含了所有兔子重量的数组 和 笼子的数量,输出 最终每个笼子的重量是否能够相同 提示:兔子重量为正整数,且不许杀兔子 数组长度不超过15 测试用例1: 输入:[2, 2, 3, 3, 5], 3 输出:true 解释:5只兔子按以下方式放入3个笼子后,笼子重量相同:[2, 3]、[2, 3]、[5] 测试用例2: 输入:[3, 5, 5, 6, 7], 4 输出:false 解释:5只兔子无论怎样分配到4个笼子中,笼子重量都无法相同
思路一
利用队列的方式;从大到小相加数据;循环判断。PYTHON版本
def func(arr,n): if n == 1: return True if n > 14 or sum(arr) % n != 0: return False avg = sum(arr) // n if max(arr) > avg:return False arr = sorted(arr)[::-1] for i in range(n): s = 0 test_arr = [] flag = True while flag: for k,v in enumerate(arr): if (s + v) <= avg : s += arr.pop(k) test_arr.append(v) if s == avg:flag = False break if flag and s + arr[-1] > avg :return False print(test_arr) if i == n - 1 :return True return False
PHP版本
<?php function weight($a,$count){ if(count($a) > 15 || count($a) < 1 || count($a) < $count || $count < 1){//边界 return false; } if($count < 2){ return true;//一个笼子直接true } $total = array_sum($a);//求和 if( $total%$count != 0 ){//总重量除不尽笼子数量 return false; } $avg = $total/$count; //var_dump($avg);die; rsort($a);//数组取降序 从大数字加小数字 //var_dump($a);die; $tmp = 0; for( $i=0; $i<=$count-1; $i++ ){ $flag = true; $tmp = 0; $tmp_arr = []; while ($flag){ $a = array_values($a); foreach ($a as $key=>$value){ if($tmp + $value <= $avg){ $tmp += $value; unset($a[$key]);//去除加的这个数 array_push($tmp_arr,$value); if($avg == $tmp){ $flag = false; break; } } } if($flag && $tmp + $a[$key] > $avg){ //临时加最后一个 return false; } } print_r($tmp_arr); } return true; } $re = weight([1,2,6,7,7,9], 2); var_dump($re);
思路二
创建一个 list; 循环所有兔子搭配的可能;放入list;然后 求和;算出结果;出现次数最多的那个结果;且这个结果 不大于 总的兔子重量除以笼子数; 如果该求和的所有种结果中少于 笼子的数量;说明无法均匀分配到每一个笼子中。
from itertools import combinations def combine(temp_list, n): '''根据n获得列表中的所有可能组合(n个元素为一组)''' temp_list2 = [] for c in combinations(temp_list, n): temp_list2.append(c) return temp_list2 def rabbit_arr(list1,jieguo): cage_list =[] end_list = [] for i in range(len(list1)): if combine(list1, i) != [()]: end_list.extend(combine(list1, i)) for i in set(end_list): if sum(i)<=jieguo: # 不可能大于 这个数字 否则就肯定会分不均 cage_list.append(sum(i)) # print(cage_list) na = max(cage_list, key=cage_list.count) inum = cage_list.count(na) dictns = {} for i in cage_list: if cage_list.count(i) == inum: dictns[i] = inum res =[] for k in dictns.keys(): res.append(k) return res
标签:count,arr,return,笼子,list,兔子,放入 来源: https://www.cnblogs.com/rianley/p/15565715.html