【华为OD统一考试B卷 】座位调整 python

【华为OD统一考试B卷 】座位调整 python

题目描述
疫情期间课堂的座位进行了特殊的调整，不能出现两个同学紧挨着，
必须隔至少一个空位。
给你一个整数数组 desk 表示当前座位的占座情况，
由若干 0 和 1 组成，其中 0 表示没有占位，1 表示占位。
在不改变原有座位秩序情况下，还能安排坐几个人？
输入描述
第一行是一个子数组表示座位占座情况，由若干 0 和 1 组成，
其中 0 表示没有占位，1 表示占位
输出描述
输出数值表示还能坐几个人
备注
1 ≤ desk.length ≤ 2 * 10^4
用例1
输入
1,0,0,0,1
输出
1
说明
只有desk[2]的位置可以坐一个人
用例2
输入
0,0,0
输出
2
用例3
输入
0,0
输出
1
用例4
输入
0
输出
1

代码实现

arr=input().split(',')
n=len(arr)
vec=[]
for i in arr:
    vec.append(int(i))
sum=0
for i in range(n):
    if i==0:
        if vec[i]==0 and (n==1 or vec[i+1]==0):
            vec[i]=1
            sum+=1
    elif i==n-1:
        if vec[i-1]==0 and vec[i]==0:
            vec[i]=1
            sum+=1
    else:
        if vec[i]==0 and vec[i-1]==0 and vec[i+1]==0:
            vec[i]=1
            sum+=1
print(sum)