需要多少个人探索资源
在一个荒地上探索资源,每一个4连接的资源区只需要一个人探索,问当前总共需要多少人探索资源。
输入:二维数组,如
111101
100100
101110
000001
001001
输出:4
这是一个华为算子开发的计算题目,要求在20分钟内做完。
稍微分析了一下,我准备用图的深度遍历来完成这个题目,中间跟面试官聊了一会浪费了点时间,所以其实我真正编程的时间应该只给了10分钟,所以我没有在给定时间内完成这个题目,有点小遗憾,主要问题在我自己,想要用一个更好点办法来解决这个问题,然鹅。。。没有成功,时间太少了。给自己一点教训吧算是。另外再说一点,笔记本编程真的不顺手。
下面开始解题吧,暂时我还没有去看更简单点图搜索算法,先来一个自己点简单版本吧。
第一步,当然是要遍历整个二维数组,找到一个值为1的点开始进行遍历;
第二步,写一个迭代过程,以当前点为基础向4个方向进行迭代,当为1时,标记当前点并继续,为0推出。
第三部,每次迭代完成,就相当是一个资源区域探索完毕。
简单代码如下:
public class Mytest {
public static void main(String[] args) {
System.out.println("test");
int a[][] = new int[10][10];
a[0][0] = 1;
a[1][1] = 1;
a[1][2] = 1;
a[2][2] = 1;
int x = cal(a);
System.out.println(x);
}
public static int cal(int[][] ver) {
int result = 0;
//标记数组,探索过的标记为-1
int[][] flag = ver;
//寻找第一个1
for (int i = 0; i < ver.length; ++i) {
for (int j = 0; j < ver[i].length; ++j) {
if (ver[i][j] == 1 && flag[i][j] != -1) {
//开始搜索
if (flag[i][j]!= -1){
flag[i][j] = -1;
func(i + 1, j, ver, flag);
func(i, j + 1, ver, flag);
func(i - 1, j, ver, flag);
func(i, j - 1, ver, flag);
result += 1;
}
}
}
}
return result;
}
private static void func(int i, int j, int[][] ver, int[][] flag) {
boolean thisflag = i >= 0 && i < ver.length && j >= 0 && j < ver[0].length && ver[i][j] == 1 && flag[i][j] != -1;
if (thisflag) {
flag[i][j] = -1;
func(i + 1, j, ver, flag);
func(i, j + 1, ver, flag);
func(i - 1, j, ver, flag);
func(i, j - 1, ver, flag);
} else {
return;
}
}
}
迭代的方式差不多就是这样来,不知道哪位有不用迭代的方式,请不吝赐教。谢谢。
下面再聊一下之前有一次面试,面试官让我遍历二叉树,我哗哗的就写完来,然后问题来了,如果不用迭代,怎么遍历二叉树呢?
迭代是我们最先反应到到,也是最简单到实现方法。如果不用迭代到话,就必须要引入一个栈来记录经过到节点,每次走到一个节点,如果不满足退出条件,就把该节点放入栈中,再继续走,走不下去了就pop一个栈元素。当然,这也就增加了更多的空间消耗。
图也是一样,不过图就更加多的分支了。顺便说一句,这种限制时间的场合还是用最简单的方式去实现,哪怕代码看起来丑一点也无所谓。
等再学习学习再来改进这个算法。