每日一题: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;
}
结果:
我还是第一次见到用例全部通过还超时的,我上面这个相当于暴力解法。
不知道如何解决超时,看答案吧。
看了眼答案,答案的思路和我是一样的,那看来是数据结构的使用出现了问题。
答案里,直接将我代码里的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;
}
可以了
标签:map,get,int,每日,list,wall,砖墙,leetcode554,size 来源: https://blog.csdn.net/qq_38041237/article/details/116355891