题目
(https://leetcode-cn.com/problems/prison-cells-after-n-days/submissions/)
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
分析
想想数组一共就8位,逆天也就2222222*2=256种可能。所以肯定有循环存在
findCircle()代码是用来寻找周期的情况
代码
class Solution {
public int findCircle(int[] cells, int N){
//用来保存历史牢房情况
List temList = new ArrayList<>();
temList.add(cells);
//每次变化后牢房的情况
int[] temp;
//记录变化的次数
int count = 0;
//是否发现重复的情况
boolean found = false;
//发现重复的时候,是第几个牢房重复的
int someIndex = 0;
while (!found){
temp = new int[cells.length];
for (int i = 1; i < cells.length-1; i++) {
if (cells[i+1]==cells[i-1]){
temp[i]=1;
}
}
cells = temp;
for (int i = 0; i < temList.size(); i++) {
if (Arrays.equals(temList.get(i), temp)){
someIndex = i;
System.out.println("跟第几个一样:"+someIndex);
found = true;
break;
}
}
temList.add(temp);
count++;
}
return count-someIndex;
}
public int[] prisonAfterNDays(int[] cells, int N) {
//用来替换第N天后的牢房情况
int[] temp;
//循环中期为14
int circle = findCircle(cells,N);
N = N %circle ;
if(N==0){
N=circle ;
}
while (N>0){
temp = new int[cells.length];
for (int i = 1; i < cells.length-1; i++) {
//如果两边的值一样,该牢房就会被占用
if (cells[i+1]==cells[i-1]){
temp[i]=1;
}
}
cells = temp;
N--;
}
return cells;
}
}