【华为OD题库-008】座位调整-Java

题目

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

思路

遍历，找出当前为0的位置，如果其前后均不为1，那么此位置可以坐人，将此位置标记为1，结果cnt++；遍历完成后返回cnt即可。
假设当前遍历位置为i，i的范围为[0,len-1],len为数组arr总长度。要让i的前后均不为1，即arr[i-1]!=1&&arr[i+1]!=1。考虑数组越界情况：

1. 当i-1<0时（i=0），只用考虑arr[i+1]!=1
2. 当i+1>=len时(i=len-1)，只用考虑arr[i-1]!=1

综上，利用逻辑与短路规则，可以将上述表达式归纳为：
(i == 0 || arr[i - 1] != 1) && (i == n - 1 || arr[i + 1] != 1)

题解

package hwod;

import java.util.Arrays;
import java.util.Scanner;

public class SeatAjust {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String[] inputs = sc.nextLine().split(",");
        int[] seats = Arrays.stream(inputs).mapToInt(Integer::parseInt).toArray();
        System.out.println(getSeatCnt(seats));
    }

    private static int getSeatCnt(int[] seats) {
        int n = seats.length;
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            if (seats[i] == 0) {
                if ((i == 0 || seats[i - 1] != 1) && (i == n - 1 || seats[i + 1] != 1)) {
                    seats[i] = 1;
                    cnt++;
                }
            }
        }
        return cnt;
    }
}

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。