文章链接
题目: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;
}
}