leetcode 957. N 天后的牢房

题目链接
思路一:位运算+模拟
分析:规则说的很清楚,当前位置由左边和右边位置的值决定。
首先我们可以先将数组转为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即可。其他不用动。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值