华为OD机试真题B卷 Python 实现【箱子之形摆放】，附详细解题思路

news2024/11/17 15:26:19

在这里插入图片描述

一、题目描述

要求将一批箱子按从上到下以‘之’字形的顺序摆放在宽度为 n 的空地上，输出箱子的摆放位置，例如：箱子ABCDEFG，空地宽为3。

摆放效果如下图：
在这里插入图片描述
则输出结果为：

AFG
BE
CD

二、输入描述

一行字符串，通过空格分割两部分，前面str部分表示箱子的字符串由数字或字母组成，后面部分是表示宽度的数字n，

如下：

ABCDEFG 3

三、输出描述

输出：

AFG
BE
CD

注：最后一行不得输出额外的空行

str只包含数字和数字，1<=len(str)<=1000，1<=n<=1000。

四、解题思路

将输入的字符串按空格分割为两部分，分别为箱子的字符串和空地的宽度；
创建一个HashMap map，用于存储每行的字符串；Key为行的下标，Value为对应的字符串；
遍历箱子的字符串，根据空地的宽度确定每个字符应该放置的行和列；
- 计算当前字符所在的列，即columnIdx = i / num，其中i为当前字符的索引；
- 如果columnIdx是偶数，则表示从左往右摆放，行索引不变，列索引为i % num；
- 如果columnIdx是奇数，则表示从右往左摆放，行索引不变，列索引为num - 1 - (i % num)，即倒序摆放；
- 将当前字符添加到对应行的字符串中；
遍历HashMap map，按行输出箱子的摆放位置；
输出结果。

五、Python算法源码

import math

text, w = input().split()
w = int(w)

cols = [''] * math.ceil(len(text) / w)

for i in range(len(text)):
    num_c = i // w
    if num_c % 2 == 0:
        cols[num_c] += text[i]
    else:
        cols[num_c] = text[i] + cols[num_c]

if len(cols) % 2 == 0 and len(cols[-1]) != w:
    cols[-1] = (w - len(cols[-1])) * ' ' + cols[-1]

result = [''] * w

for i in range(w):
    for col in cols:
        if i < len(col) and col[i] != ' ':
            result[i] += col[i]
for res in result:
    if res == '':
        continue
    print(res)