题目链接
思路一:位运算+模拟
分析:规则说的很清楚,当前位置由左边和右边位置的值决定。
首先我们可以先将数组转为int。
将一个长度为8的数组转为int的代码如下:
public int transferArrToNum(int[] cells){
int num = 0;
for(int i=0;i<8;i++){
if(cells[i]==1){
//把1移动到相应位置,然后与num或运算,或运算就是将1加入到num中
num = num | (1<<(7-i));
}
}
return num;
}
我们首先令: int cur = transferArrToNum(cells);
根据当前数字计算下一个数字。
每个位置,只和左右位置的值有关,第一个位置和最后一个位置肯定是0,因为没有左右位置。
举例:01011001
从右边开始计算
这是第一个要计算的位置,这个位置由 ((cur>>0)&1) 与 ((cur>>2)&1)决定,也是就是上面cur右边的第一个格子和右边的第三个格子。
cur>>0:就是cur,完全没动。
cur>>2:就是cur右移2个位置。
&1:表示取最后的一个位。
当前位置就是看上面取值的两个位是否相等,如果相等就变为1.
如下图:
此时next=96,也就是二进制的0110 0000
然后将cur更新为next,再经历上面的过程即可。
一共经历N轮。
但是此时会出现个问题,因为N会很大,9次方,并且一共8位,最多也就是256种,而且最前和最后一定是0,也就是一共6位变化的,最多就是64种情况。那么N轮肯定会出现循环。
所以用一个HashMap来记住此时的cur是第几轮。
当出现循环的时候,处理方法:
1.首先我们计算这个圈多大
2.再计算还剩下多少步,对圈的大小求余。
3.再向前走 【步骤2的结果】步。
图解:
比如此时,发现有循环。
圈的大小为:11-4=7
假设一共100步。
此时剩余100-11=89。
对圈的大小求余:89%7=5,也就是还需要再向前走五步。
这个就是答案。
最后再将这个数字转为数组即可;
数字转数组代码:
public int[] transferNumToArr(int num){
int[] res = new int[8];
for(int i=0; i<8; i++){
res[7-i] = num&1;
num>>=1;
}
return res;
}
完整代码:
class Solution {
HashMap<Integer, Integer> map = new HashMap<>();
public int[] prisonAfterNDays(int[] cells, int n) {
int cur = transferArrToNum(cells);
for(int i=0; i<n; i++){
int next = getNext(cur);
if(map.containsKey(next)) {
//出现了循环
//计算圈的大小
int size = i + 1 - map.get(next);
//还需要向前走的步数
int redu = (n - i - 1) % size;
while (redu != 0) {
next = getNext(next);
redu--;
}
return transferNumToArr(next);
}
map.put(next,i+1);
cur = next;
}
return transferNumToArr(cur);
}
//计算下一轮状态
public int getNext(int cur){
int next = 0;
for(int i=1;i <7;i++){
if(((cur>>(i-1))&1) == ((cur>>(i+1))&1) ){
next = next | (1<<i);
}
}
return next;
}
//将数字转数组
public int[] transferNumToArr(int num){
int[] res = new int[8];
for(int i=0; i<8; i++){
res[7-i] = num&1;
num>>=1;
}
return res;
}
//将数组转数字
public int transferArrToNum(int[] cells){
int num = 0;
for(int i=0;i<8;i++){
if(cells[i]==1){
num = num | (1<<(7-i));
}
}
return num;
}
}
思路二:位运算+模拟(位运算)
这里主要改进的是计算下一个状态的函数getNext
将cur右移2位
进行异或
我们要的只是中间的6位。那么可以和6个1与一下,切下来最后6位。
然后我们发现这里和答案,0和1正好相反,因为异或是相同为0,题目要求是相同为1.
所以将此时的0和1反向一下。
也就是和6个1异或一下。
此时就得到了下一个状态,但是我们是8位,这里只去了6位,最前面一位是0,正好这里也是0,所以就是取了7位,少了最右边的一位,那么将此时的数字左移一位即可。
代码:
private int getNext(int cur){
cur = cur ^ (cur>>2);
cur = cur & 63;
cur = cur ^ 63;
cur = cur<<1;
return cur;
}
这里改进getNext即可。其他不用动。