【新员工座位安排系统】
【新员工座位】
工位由序列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))