每日一题:leetcode554. 砖墙

题:
每日一题:leetcode554. 砖墙
每日一题:leetcode554. 砖墙


思:
对于每一行,将砖块转换为数字 [1,2,2,1] -> [1,3,5,6]
把每一行都转换完以后,遍历[1,5]每一列,找出一条竖线使穿过的砖头最小
对于每一列,如果有一行存在对应列数的数字,则表示经过此行没有穿过砖头
代码超时
当wall == [[100000000],[100000000],[100000000]]时,还从1遍历到100000000,显然是浪费时间
优化方法:
不从1开始遍历,只遍历出现过的数字,因为如果数字没出现,代表根本不可能在这一列减去1


码:

public int leastBricks(List<List<Integer>> wall) {
        Set repeat = new HashSet();
//        int col;
        for (List list : wall) {
            for (int i = 1; i < list.size(); i++) {
                list.set(i, (int) list.get(i - 1) + (int) list.get(i));
                repeat.add((int)list.get(i));
            }
            repeat.add((int)list.get(0));
        }
//        col = wall.get(0).get(wall.get(0).size() - 1);
        int min = wall.size();
        // 在repeat里去掉最大的数(右边界)
        repeat.remove(wall.get(0).get(wall.get(0).size()-1));
        Object[] objects = repeat.toArray();
        for (int i = 0; i < repeat.size(); i++) {
            int cur = wall.size();
            for(int j = 0 ; j < wall.size() ; j++){
                if(wall.get(j).contains((int)objects[i]))
                    cur--;
            }
            if(cur < min)
                min = cur;
        }
        return min;
    }

结果:
每日一题:leetcode554. 砖墙


我还是第一次见到用例全部通过还超时的,我上面这个相当于暴力解法。
不知道如何解决超时,看答案吧。


看了眼答案,答案的思路和我是一样的,那看来是数据结构的使用出现了问题。
答案里,直接将我代码里的set部分,用一个map来存储,当遇到重复数字的时候,map里那个key对应的value+1.最后找出最大的value,然后用wall.size减这个最大的value就是答案,这样做可以避免我代码里,还得再遍历一遍set,才能找到最大的value。


修改后的代码,把set换成map:

public static int leastBricks(List<List<Integer>> wall) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
//        int col;
        for (List<Integer> list : wall) {
            for (int i = 0; i < list.size(); i++) {
                if (i != 0)
                    list.set(i, list.get(i - 1) + list.get(i));
                // 最后一个数不加入
                if (i != list.size() - 1) {
                    if (map.containsKey(list.get(i))) {
                        map.replace(list.get(i), map.get(list.get(i)) + 1);
                    } else {
                        map.put(list.get(i), 1);
                    }
                }
            }
        }
        // 寻找最大的value
        int max = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            max = Math.max(max, entry.getValue());
        }
        return wall.size()-max;
    }

可以了

上一篇:生成 -Wall 选项不包括的警告


下一篇:砖墙