码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode957. N 天后的牢房

时间:2018-12-16 14:43:02      阅读:376      评论:0      收藏:0      [点我收藏+]

标签:except   exce   无法   dstat   lse   思路   情况   循环   第一个   

8 间牢房排成一排,每间牢房不是有人住就是空着。

每天,无论牢房是被占用或空置,都会根据以下规则进行更改:

  • 如果一间牢房的两个相邻的房间都被占用或都是空的,那么该牢房就会被占用。
  • 否则,它就会被空置。

(请注意,由于监狱中的牢房排成一行,所以行中的第一个和最后一个房间无法有两个相邻的房间。)

我们用以下方式描述监狱的当前状态:如果第 i 间牢房被占用,则 cell[i]==1,否则 cell[i]==0

根据监狱的初始状态,在 N 天后返回监狱的状况(和上述 N 种变化)。

--

思路:

  如果是按照一般思路,根据前一天状态,更新第二天状态,逻辑会很清晰,但是会超时或溢出(当N=1000000000时),因为要生成N次数组来保存状态。

  优化办法:

    1、题干是8个房间,没个房间就0或1两种状态,所以8个房间最多一共也就 2的8次方 =256种状态, 

    2、状态是有限的,而 天数N可以是无穷大的,所以规律就在这256种之内。

    3、最多生成256天的状态, 在这之内,不断寻找有无相同状态的一天,如果有,那么规律就出现了

        1. 2. [3].  . . . k .  . . . 256. . N

        加入我们算出第k天的状态,和第3天一致,那么从k之后的未来所有状态,都是处于 3到k 之间循环的。

 

  步骤:

    1. 添加第0天状态到 list

        2. 根据 list中的最后一天的状态pre,再添加新的一天的状态 cur

            3. 查看cur 在list 0~pre中有无重复,如果有,规律出现,返回出现的下标index,如果无 继续上述步骤

    4.  (N-index)%(k-index) + index 即为结果的下标

//957. N 天后的牢房
    public int[] prisonAfterNDays(int[] cells, int N) {
        if (N == 0) {
            return cells;
        }
        ArrayList<int[]> list = new ArrayList<>();
        list.add(cells);
        int k = 1;  //k为第一天
        while (k <= N) {
            int[] cur = prisionStatus(list.get(k - 1));//根据昨天的状态获取今天的状态
            list.add(cur);//将今天的状态添加至list
            int index = findStatus(list);//循环判断list中是否存在已有状态,如果有返回下标,无返回-1
            if (index != -1) { //如果 index != -1 ,说明找到重复 ,   ps:此处存在 /by zero 异常
                try {
                    int res = (N - index) % (k - index); //结果的下标
                    return list.get(index + res);
                } catch (Exception e){
                    return list.get(index);  //存在异常的情况是 index + 1 == cur ,也就是昨天和今天状态一致,也意味着监狱状态稳定,以后不会变化。
                }
            }
            k++;
        }
        return list.get(list.size() - 1);
    }

    //根据昨天获取当天监狱状态
    private int[] prisionStatus(int[] pre) {
        int[] cur = new int[pre.length];
        cur[0] = 0;
        cur[cur.length - 1] = 0;
        for (int i = 1; i < pre.length - 1; i++) {
            if (pre[i - 1] == pre[i + 1]) {
                cur[i] = 1;
            } else {
                cur[i] = 0;
            }
        }
        return cur;
    }

    //循环判断list中是否存在已有状态,如果有返回下标,无 返回-1
    private int findStatus(ArrayList<int[]> list) {
        for (int i = 0; i <= list.size() - 2; i++) {
            if (arrayIsSame(list.get(list.size() - 1), list.get(i))) {
                return i;
            }
        }
        return -1;
    }

    //判断数组是否一致
    private boolean arrayIsSame(int[] a, int[] b) {
        if (a.length != b.length) {
            return false;
        }
        for (int i = 0; i < a.length; i++) {
            if (a[i] != b[i]) {
                return false;
            }
        }
        return true;
    }

 

leetcode957. N 天后的牢房

标签:except   exce   无法   dstat   lse   思路   情况   循环   第一个   

原文地址:https://www.cnblogs.com/magicya/p/10126231.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!