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