华为OD试题一(磁盘容量排序、数大雁、计算疫情扩散时间)

1. 磁盘容量排序

题目描述：磁盘的容量单位常用的有M，G，T这三个等级，它们之间的换算关系为1T = 1024G，1G = 1024M，现在给定n块磁盘的容量，请对它们按从小到大的顺序进行稳定排序，例如给定5块盘的容量，1T，20M，3G，10G6T，3M12G9M排序	后的结果为20M，3G，3M12G9M，1T，10G6T。注意单位可以重复出现，上述3M12G9M表示的容量即为3M+12G+9M，和12M12G相等。

参考代码：

# 测试数据
test_data = ['2G4M','3M2G','1T']
# 核心是将其划归为统一的单位M
# 单位字典
unit_dict = {'T':1024 * 1024,'G':1024,'M':1,
}
# 换算后的结果
conversion_data = []
# 换算后的结果与原结果对应关系
conversion_dict = {}# 该函数用于换算结果
def fun(ELE):# 用于存放结果count = 0# 辅助指针 cur = 0 for index,ele in enumerate(ELE):if ele in ['T','G','M']:count += int(ELE[cur:int(index)]) * unit_dict[ele]# cur 指向下一个要换算的结果cur = index + 1return count # 处理测试数据
for ele in test_data:res = fun(ele)# 添加换算后的数据conversion_data.append(res)# 建立换算后的数据和源数据之间的关系conversion_dict[ele] = res# 处理的数据排序
conversion_data.sort()# 存放最终结果
result = []# 将处理后的数据转化成原数据
for _ in conversion_data:for k,v in conversion_dict.items():if _ == v:result.append(k)conversion_dict.pop(k)break
# 打印最终结果
print(result)

2. 数大雁

题目描述：
一群大雁往南飞，给定一个字符串记录地面上的游客听到的大雁叫声，请给出叫
声最少由几只大雁发出。具体的：
1. 大雁发出的完整叫声为"quack"，因为有多只大雁同一时间嘎嘎作响，所以字符串中可能会混合多个 "quack"。
2. 大雁会依次完整发出 "quack"，即字符串中‘q’, ’u’, ’a’, ’c’, ’k’ 这 5个字母按顺序完整存在才能计数为一只大雁。如果不完整或者没有按顺序则不予计数。
3. 如果字符串不是由‘q’, ’u’, ’a’, ’c’, ’k’字符组合而成，或者没有找到一只大雁，请返回 -1。核心：同过q 和 k 的位置确定大雁数量


# 测试数据
test_str1 = "quackquack"
test_str2 = "quqackuack"
test_str3 = "quackquook"std_str = "quack"def fun(temp):# 统计大雁个数count = 0# 判断非法情况for _ in temp:if _ not in std_str:return -1# 查找q的位置q_pos = []for pos,value in enumerate(temp):if value == 'q':q_pos.append(pos)# 查找k的位置k_pos = []for _ in q_pos:before = _for s in std_str:after = temp.find(s,before)# 如果没找到if after == -1:breakif s == 'k':k_pos_append(after)# 通过分析q 的位置和其对应的k的位置计算大雁数量for i in range(len(q_pos)):if i == 0:count += 1else:if q_pos[i] < k_pos[i - 1]:count += 1return count 		result = fun(str1)
print(result)

3. 计算疫情扩散时间

题目描述：在一个地图中(地图由n*n个区域组成），有部分区域被感染病菌。感
染区域每天都会把周围（上下左右）的4个区域感染。
请根据给定的地图计算，多少天以后，全部区域都会被感染。
如果初始地图上所有区域全部都被感染，或者没有被感染区域，返回-1。核心：计算位置关系

input1 = [1, 0, 1, 0, 0, 0, 1, 0, 1]
input2 = [0, 0, 0, 0]
input3 = [1, 1, 1, 1, 1, 1, 1, 1, 1]
import mathdef fun(temp_list):# 全为0 的情况if all(x == 0 for x in temp_list):return -1# 全为1 的情况if all(x == 1 for x in temp_list):return -1# 正方形边长l = int(math.sqrt(len(temp_list)))# 统计天数count = 0while True:# 如果全为1 即全部感染则 退出if all(x == 1 for x in temp_list):return count# 计算 1 的位置temp_1_pos = [pos for pos,x in enumerate(temp_list) if x == 1]# 计算感染区域for i in temp_1_pos:# 上方区域if 0 <= i - l <= len(temp_list) - 1:if temp_list[i - l] == 0:temp_list[i - l] = 1# 左边区域if 0 <= i - 1 <= len(temp_list) - 1:if temp_list[i - 1] == 0:temp_list[i - 1] = 1# 右边区域if 0 <= i + 1 <= len(temp_list) - 1:if temp_list[i + 1] == 0:temp_list[i + 1] = 1# 下方区域if 0 <= i + l <= len(temp_list) - 1:if temp_list[i + l] == 0:temp_list[i + l] = 1count += 1
res = fun(input3)
print(res)