一道Twitter编程面试题

Twitter面试题：水沟积水问题

问题描述：“看下面这个图片”



“在这个图片里我们有不同高度的墙。这个图片由一个整数数组所代表，数组中每个数是墙的高度。上边的图可以表示为数组[2,5,1,2,3,4,7,7,6]”

“假如开始下雨了，那么墙之间的水坑能够装多少水呢？”
思路分析： 一种只需要一次遍历的方法（注：不知道算不算动态规划算法，请知道的高人鉴定一下）

1. 首先，用两个指针（left 和 right）分别指向数组的第一个元素和最后一个元素，左指针从左向右遍历，右指针从右向左遍历；

2. 初始化数组中一个元素（a[0]）为左边遍历得到的最大值（max_left），最后一个元素（a[a.length-1]）为从右边遍历得到的最大值（max_right）；

3. 开始遍历，遍历条件为左指针小于右指针

4. 如果左边遍历的最大值小于右边遍历的最大值，说明只要有水沟（即小于左边最大值max_left的元素）就会有积水，因为右边的最大值可以保证左边水沟的积水不会流失掉；

同样，如果左边遍历的最大值不小于右边遍历的最大值，只要右边有水沟（即小于右边最大值max_right的元素）就会有积水。

解决方案：Java实现代码

public class Twitter_puddle {
public int caculate(int[] a) {
if (a == null) return 0;
int left = 0;
int right = a.length - 1;
int max_left = a[left];
int max_right = a[right];
int volume = 0;

while (left < right) {
if (max_left < max_right) {
left++;
if (max_left < a[left]) {
max_left = a[left];
}
else {
volume += max_left - a[left];
}
}
else {
right--;
if (max_right < a[right]) {
max_right = a[right];
}
else {
volume += max_right - a[right];
}
}
}
return volume;
}
}

测试代码：

import static org.junit.Assert.*;

import org.junit.Before;
import org.junit.Test;
import static org.hamcrest.Matchers.*;

public class Twitter_puddleTest {

private Twitter_puddle puddle;

@Before
public void before() {
puddle = new Twitter_puddle();
}

@Test
public void testNull() {
assertThat(puddle.caculate(null), is(0));
}

@Test
public void testOne() {
int a[] = {1};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testTwoRightHigher() {
int a[] = {1, 2};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testTwoLeftHigher() {
int a[] = {2, 1};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testTwoSameHight() {
int a[] = {1, 1};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testThreeMiddleHigher() {
int a[] = {1, 2, 1};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testThreeMiddleLower() {
int a[] = {2, 1, 2};
assertThat(puddle.caculate(a), is(1));
}

@Test
public void testThreeSameHeight() {
int a[] = {1, 1, 1};
assertThat(puddle.caculate(a), is(0));
}

@Test
public void testRandom1() {
int a[] = {2, 5, 1, 2, 3, 4, 7, 7, 6};
assertThat(puddle.caculate(a), is(10));
}

@Test
public void testRandom2() {
int a[] = {2, 5, 1, 3, 1, 2, 1, 7, 7, 6};
assertThat(puddle.caculate(a), is(17));
}

@Test
public void testRandom3() {
int a[] = {6, 1, 4, 6, 7, 5, 1, 6, 4};
assertThat(puddle.caculate(a), is(13));
}

@Test
public void testRandom4() {
int a[] = {6, 6, 6, 6, 6, 6, 6, 6, 6};
assertThat(puddle.caculate(a), is(0));
}
}

此面试题是前几天在CSDN的极客头条版看到的，在这里把问题以及思路自己整理了一遍，并用Java实现了！面试题分析：我的Twitter面试技术面试失败了点击打开链接