leetcode957. N 天后的牢房

8 间牢房排成一排，每间牢房不是有人住就是空着。

每天，无论牢房是被占用或空置，都会根据以下规则进行更改：

如果一间牢房的两个相邻的房间都被占用或都是空的，那么该牢房就会被占用。
否则，它就会被空置。

（请注意，由于监狱中的牢房排成一行，所以行中的第一个和最后一个房间无法有两个相邻的房间。）

我们用以下方式描述监狱的当前状态：如果第 i 间牢房被占用，则 cell[i]==1，否则 cell[i]==0。

根据监狱的初始状态，在 N 天后返回监狱的状况（和上述 N 种变化）。

思路：

　　如果是按照一般思路，根据前一天状态，更新第二天状态，逻辑会很清晰，但是会超时或溢出（当N=1000000000时），因为要生成N次数组来保存状态。

　　优化办法：

　　　　1、题干是8个房间，没个房间就0或1两种状态，所以8个房间最多一共也就 2的8次方 =256种状态，

　　　　2、状态是有限的，而天数N可以是无穷大的，所以规律就在这256种之内。

扫描二维码关注公众号，回复： 4528596 查看本文章

　　　　3、最多生成256天的状态，在这之内，不断寻找有无相同状态的一天，如果有，那么规律就出现了

　　　　　　　　1. 2. [3]. . . . k . . . . 256. . N

　　　　　　　　加入我们算出第k天的状态，和第3天一致，那么从k之后的未来所有状态，都是处于 3到k 之间循环的。

　　步骤：

　　　　1. 添加第0天状态到 list

　　 2. 根据 list中的最后一天的状态pre，再添加新的一天的状态 cur

3. 查看cur 在list 0～pre中有无重复，如果有，规律出现，返回出现的下标index，如果无继续上述步骤

　　　　4. (N-index)%(k-index) + index 即为结果的下标

//957. N 天后的牢房
    public int[] prisonAfterNDays(int[] cells, int N) {
        if (N == 0) {
            return cells;
        }
        ArrayList<int[]> list = new ArrayList<>();
        list.add(cells);
        int k = 1;  //k为第一天
        while (k <= N) {
            int[] cur = prisionStatus(list.get(k - 1));//根据昨天的状态获取今天的状态
            list.add(cur);//将今天的状态添加至list
            int index = findStatus(list);//循环判断list中是否存在已有状态,如果有返回下标，无返回-1
            if (index != -1) { //如果 index != -1 ，说明找到重复 ,   ps:此处存在 ／by zero 异常
                try {
                    int res = (N - index) % (k - index); //结果的下标
                    return list.get(index + res);
                } catch (Exception e){
                    return list.get(index);  //存在异常的情况是 index + 1 == cur ，也就是昨天和今天状态一致，也意味着监狱状态稳定，以后不会变化。
                }
            }
            k++;
        }
        return list.get(list.size() - 1);
    }

    //根据昨天获取当天监狱状态
    private int[] prisionStatus(int[] pre) {
        int[] cur = new int[pre.length];
        cur[0] = 0;
        cur[cur.length - 1] = 0;
        for (int i = 1; i < pre.length - 1; i++) {
            if (pre[i - 1] == pre[i + 1]) {
                cur[i] = 1;
            } else {
                cur[i] = 0;
            }
        }
        return cur;
    }

    //循环判断list中是否存在已有状态,如果有返回下标，无 返回-1
    private int findStatus(ArrayList<int[]> list) {
        for (int i = 0; i <= list.size() - 2; i++) {
            if (arrayIsSame(list.get(list.size() - 1), list.get(i))) {
                return i;
            }
        }
        return -1;
    }

    //判断数组是否一致
    private boolean arrayIsSame(int[] a, int[] b) {
        if (a.length != b.length) {
            return false;
        }
        for (int i = 0; i < a.length; i++) {
            if (a[i] != b[i]) {
                return false;
            }
        }
        return true;
    }

leetcode957. N 天后的牢房

猜你喜欢