【新员工座位安排系统】

最新推荐文章于 2025-02-22 22:14:49 发布

原创最新推荐文章于 2025-02-22 22:14:49 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#算法

华为OD算法机试真题-python版专栏收录该内容

18 篇文章

订阅专栏

该文讨论了一种计算工位友好度的问题，其中友好度定义为空位两侧连续老员工的数量之和。给定工位序列，目标是找出友好度最高的空位。文中提供了两种不同的算法方法来解决这个问题：一种是分别向前和向后查找连续的1，另一种是从左右两个方向累加计算每个空位的友好度，最后取最大值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【新员工座位】

工位由序列F1,F2…Fn组成，Fi值为0、1或2。其中0代表空置，1代表有人，2代表障碍物。

1、某一空位的友好度为左右连续老员工数之和，
2、为方便新员工学习求助，优先安排友好度高的空位，

给出工位序列，求所有空位中友好度的最大值。

输入描述：

第一行为工位序列：F1，F2…Fn组成，
1<=n<=10000，Fi值为0、1或2。其中0代表空置，1代表有人，2代表障碍物。

输出描述：

所有空位中友好度的最大值。如果没有空位，返回0。

示例1 输入输出示例仅供调试，后台判题数据一般不包含示例

输入

0 1 0

输出

1

说明

第1个位置和第3个位置，友好度均为1。

示例2 输入输出示例仅供调试，后台判题数据一般不包含示例

输入

1 1 0 1 2 1 0

输出

3

方法一：找到每个0出现的位置，然后分别向前后找连续出现的1，出现计数变量就加一，不是1就跳出

s = input().split()#
l = []
for i in range(len(s)):
    if s[i] == '0':
        count = 0
        for j in range(i-1,-1,-1):
            if s[j] == '1':
                count += 1
            else:
                break
        for k in range(i+1,len(s)):
            if s[k] == '1':
                count += 1
            else:
                break
        l.append(count)
if l:
    print(max(l))
else:
    print(0)

方法二：

记录空位左边的友好度：正向遍历，遇到1则计数变量加1，遇到0则记录变量值，并且重置为0，遇到2则直接重置为0

记录空位右边的友好度：逆向遍历，遇到1则计数变量加1，遇到0则记录变量值，并且重置为0，遇到2则直接重置为0

然后累加取最大值

# 输入获取
arr = input().split()
 
 
# 算法入口
def getResult(arr):
    # 记录空位的友好度
    ep = {}
 
    friendShip = 0
    # 从左向右遍历，记录每个空位左边的友好度
    for i in range(len(arr)):
        if arr[i] == "0":
            ep[i] = friendShip
            friendShip = 0
        elif arr[i] == "1":
            friendShip += 1
        else:
            friendShip = 0
 
    friendShip = 0
    ans = 0
    # 从右向左遍历，累加每个空位右边的友好度，这样就得到了每个空位的友好度，取最大值即可
    for i in range(len(arr) - 1, 0, -1):
        if arr[i] == "0":
            ans = max(ans, ep[i] + friendShip)
            friendShip = 0
        elif arr[i] == "1":
            friendShip += 1
        else:
            friendShip = 0
 
    return ans
 
 
# 算法调用
print(getResult(arr))