LeetCode554 砖墙

Posted 2021-06-21 Zephyr丶J

LeetCode554 砖墙

2021.5.2日的每日一题，今天看到的时候，第一感觉就是前缀和，然后就写出了下面的代码

class Solution {
    public int leastBricks(List<List<Integer>> wall) {
        //按我上午想的思路，就是先求每一行的前缀和，然后每一行弄个指针移动，这些指针用一个数组存储起来
        //然后遍历，从1到长度和减1，看看哪个穿过的最少
        int col = wall.size();  //有多少行
        List<Integer> list = wall.get(0);   //第一行
        int length = 0;     //一行多长，也就是列数
        for(int i = 0; i < list.size(); i++){
            length += list.get(i);
        }
        //计算前缀和
        int[][] presum = new int[col][20001];
        for(int i = 0; i < col; i++){
            List<Integer> temp = wall.get(i);
            for(int j = 0; j < temp.size(); j++){
                presum[i][j + 1] = presum[i][j] + temp.get(j);
            }
        }

        //每一行目前的下标
        int[] index = new int[col];
        Arrays.fill(index, 1);
        //遍历每一个长度，看看穿过了几个墙，开头结尾不算
        int res = col;
        for(int i = 1; i <= length - 1; i++){
            //遍历每一行
            int cross = 0;
            for(int j = 0; j < col; j++){
                int tempdis = presum[j][index[j]]; //当前行遍历到哪了
                if(tempdis < i) //如果距离小了，就移动下标，等于不管
                    index[j]++;
                tempdis = presum[j][index[j]]; //更新
                if(tempdis > i)     //如果大了，就说明穿过了，等于不算
                    cross++;        
            }
            res = Math.min(res, cross);
        }
        return res;
    }
}

写的有点繁琐，但是我自认为思路没问题，先处理每一行的前缀和，然后从1遍历到length - 1，如果有一行在当前位置的前缀和小于该数，就移动下标，等于该数，不处理，大于该数了，说明就穿过砖块了；最后取最小就行了。但是过不了，明天去实验室处理Debug一下再来改。
然后去看了题解，发现自己麻烦了，其实还是前缀和，只不过处理前缀和的时候，就可以把每个数字穿过的边缘数统计下来了，学习了

改正：第一次过不了的示例是[[1,1][2][1,1]]。然后发现跳不进最后的循环中，看了一下，是判断条件出错了，小于变成小于等于就没问题了
然后马上第二次过不了，示例是[100000000][100000000][100000000]，显示是超出内存限制，把创建的数组大小改成了题目条件所给的最大限制，这个过了
然后第三次过不了，超出时间限制了，没办法了，但是说明这个思路还是正确的，就是缺少了点灵性，哈哈！