leetcode957. Prison Cells After N Days

文章链接
题目：8 间牢房排成一排，每间牢房不是有人住就是空着。
每天，无论牢房是被占用或空置，都会根据以下规则进行更改：
如果一间牢房的两个相邻的房间都被占用或都是空的，那么该牢房就会被占用。
否则，它就会被空置。
（请注意，由于监狱中的牢房排成一行，所以行中的第一个和最后一个房间无法有两个相邻的房间。）
我们用以下方式描述监狱的当前状态：如果第 i 间牢房被占用，则 cell[i]==1，否则 cell[i]==0。
根据监狱的初始状态，在 N 天后返回监狱的状况（和上述 N 种变化）。

示例 1：

输入：cells = [0,1,0,1,1,0,0,1], N = 7
输出：[0,0,1,1,0,0,0,0]
解释：
下表概述了监狱每天的状况：
Day 0: [0, 1, 0, 1, 1, 0, 0, 1]
Day 1: [0, 1, 1, 0, 0, 0, 0, 0]
Day 2: [0, 0, 0, 0, 1, 1, 1, 0]
Day 3: [0, 1, 1, 0, 0, 1, 0, 0]
Day 4: [0, 0, 0, 0, 0, 1, 0, 0]
Day 5: [0, 1, 1, 1, 0, 1, 0, 0]
Day 6: [0, 0, 1, 0, 1, 1, 0, 0]
Day 7: [0, 0, 1, 1, 0, 0, 0, 0]

示例 2：

输入：cells = [1,0,0,1,0,0,1,0], N = 1000000000
输出：[0,0,1,1,1,1,1,0]

提示：

cells.length == 8
cells[i] 的值为 0 或 1
1 <= N <= 10^9
参考：https://leetcode.com/problems/prison-cells-after-n-days/discuss/205684/JavaPython-Find-the-Loop-or-Mod-14
别人思路：某位高手发现，随着天数的增加是有规律的，每经历14步就会重复一次，所以迎刃而解。

class Solution {
    public int[] prisonAfterNDays(int[] cells, int N) {
        for(N = (N-1) % 14 + 1; N > 0; N--){
            int[] ans = new int[8];
            for(int j = 1; j < 7; j++){
                if(cells[j - 1] == cells[j + 1]){
                    ans[j] = 1;
                }else{
                    ans[j] = 0;
                }
            }
            cells = ans;
        }
        return cells;
    }
}

思路2：或者发现可能是存在某种规律，就是子串一定会重复，2的8次方不也就256，串的种类是有限的，何况开头和结尾还都是0，这意味着也就64种可能，用map保存一下，然后取余一波即可

class Solution {
    public int[] prisonAfterNDays(int[] cells, int N) {
        Map<String, Integer> map = new HashMap<>();
        while(N > 0){
            map.put(Arrays.toString(cells), N--);
            int[] ans = new int[8];
            for(int i = 1; i < 7; i++){
                ans[i] = cells[i - 1] == cells[i + 1] ? 1 : 0;
            }
            cells = ans;
            if(map.containsKey(Arrays.toString(cells))){
                N = N % (map.get(Arrays.toString(cells)) - N);
            }
        }
        return cells;
    }
}

leetcode957. Prison Cells After N Days

猜你喜欢