一、介绍
本文我们将了解java.util.Arrays。这是一个自Java1.2以来就存在的Java工具类。
使用Arrays,我们可以创建、比较、排序、搜索、流操作以及其他转换操作。
二、创建
让我们看一下通过Arrays创建数组的方法:copyOf、copyOfRange和fill。
2.1. copyOf and copyOfRange
使用copyOfRange,我们需要一个源数组和开始这索引与结束索引,前闭后开:
String[] intro = new String[]{"once", "upon", "a", "time"};
String[] abridgement = Arrays.copyOfRange(intro, 0, 3);
assertArrayEquals(new String[]{"once", "upon", "a"}, abridgement);
assertFalse(Arrays.equals(intro, abridgement));
使用copyOf,需要传递一个源数组和目标数组长度并且返回一个新数组,新数组长度为所传参数长度:
String[] intro = new String[]{"once", "upon", "a", "time"};
String[] revised = Arrays.copyOf(intro, 3);
String[] expanded = Arrays.copyOf(intro, 5);
assertArrayEquals(Arrays.copyOfRange(intro, 0, 3), revised);
assertNull(expanded[4]);
注意:copyOf后,如果返回数组长度大于源数组长度会用null值来填充剩余空间。
2.2. fill
另一种方式,我们可以使用fill创建一个固定长度的数组,当所有数组元素都一样的时候非常有用:
String[] stutter = new String[3];
Arrays.fill(stutter, "once");
assertTrue(Stream.of(stutter).allMatch(el -> "once".equals(el));
setAll是创建不同元素的数组。注意:我们需要提前初始化数组在调用fill前。
三、比较
现在让我们来看一下使用Arrays进行比较的方法。
3.1. equals and deepEquals
我们可以使用equals对数组进行大小和内容的简单比较。如果我们向数组中添加了一个空元素,两个数组就不相等了。
String[] intro = new String[]{"once", "upon", "a", "time"};
assertTrue(Arrays.equals(new String[]{"once", "upon", "a", "time"}, intro));
assertFalse(Arrays.equals(new String[]{"once", "upon", "a", "null"}, intro));
如果我们比较嵌套或多维数组,我们可以使用deepEquals不仅比较第一层元素,而且还会递归比较对象元素:
String[] end = new String[]{"end"};
String[] intro = new String[]{"once", "upon", "a", "time"};
Object[] story =
new Object[]{intro, new String[]{"chapter one", "chapter two"}, end};
Object[] copy =
new Object[]{intro, new String[]{"chapter one", "chapter two"}, end};
assertTrue(Arrays.deepEquals(story, copy));
assertFalse(Arrays.equals(story, copy));
上述代码,deepEquals返回true, equals返回false。
这是因为deepEquals每次比较时都会调用自己,而equals只会比较子数组的引用。
因此,使用自引用数组的调用非常危险。
3.2. hashCode _and _deepHashCode
我们用hashCode方法来基于对象的内容计算出一个整数:
Object[] looping = new Object[]{intro, intro};
int hashBefore = Arrays.hasCode(looping);
int deepHashBefore = Arrays.deepHashCode(looping);
现在,我们将数组中一个元素置为空并重新获取它的hash值
intro[3] = null;
int hashAfter = Arrays.hashCode(looping);
我们再试试deepHashCode,它检查数组的每个元素的内容。我们重新来调用一次:
int deepHashAfter = Arrays.deepHashCode(looping);
现在,我们来看一些这两个方法有什么不同:
assertEquals(hashAfter, hashBefore);
assertNotEquals(deepHashAfter, deepHashBefore);
deepHashCode是我们在HashMap和HashSet等数据结构做为数组元素时做低层运算的。
四、排序和搜索
下面让我们来看一下数据的排序和搜索
4.1. sort
如果我们的元素是基本元素或实现了Compareable的对象,我们可以用sort来进行排序:
String[] sorted = Arrays.copyOf(intro, 4);
Arrays.sort(sorted);
assertArrayEquals(new String[]{"a", "once", "time", "upon"}, sorted);
sort根据数组不同的元素类型使用了不同的算法。基本类型使用了dual-pivot quicksort,对象类型的数组使用了Timsort。两者的时间复杂度都是O(nlog(n))。
4.2. binarySearch
在未排序的数组中搜索是线性的,但是如果是排序的数组,我们可以用二分查找在O(logn)完成:
int exact = Arrays.binarySearch(sorted, "time");
int caseInsensitive =
Arrays.binarySearch(sorted, "TiMe", String::compareToIgnoreCase);
assertEquals("time", sorted[exact]);
assertEquals(2, exact);
assertEquals(exact, caseInsensitive);
如果我们不提供比较器作为第三个参数,那么binarySearch将依赖于我们元素类型的Comparable。再次注意,使用binarySearch的前提是已排序数组。
五、流操作
之前提到过,Arrays在Java8中进行了更新,提供了Stream API的方法,例如parallelSort、stream和setAll。
5.1. stream
stream方法让我们像流一样调用Stream API来使用我们的数组:
Assert.assertEquals(Arrays.stream(intro).count(), 4);
exception.expect(ArrayIndexOutOfBoundsException.class);
Arrays.stream(intro, 2, 1).count();
我们可以传入在范围内的索引或走出范围的索引,超出范围会抛出ArrayindexOutOfBoundsException。
六、转换
最后,toString、asList和setAll为我们提供了几种不同方法来转换数组。
6.1. toString and deepToString
一个好的可读性很好的将基础类型数组转为字符串的方法:
assertEquals("[once, upon, a time]", Arrays.toString(storyIntro));
同样我们也可以用deepToSTring来打印对象类型的数组:
assertEquals(
"[[once, upon, a, time], [chapter one, chapter tow], [the, end]]",
Arrays.deepToString(story));
6.2. asList
Arrays方法中最方便的一个是asList,它可以将数组转换为列表:
List rets = Arrays.asList(intro);
assertTrue(rets.contains("upon"));
assertTrue(rets.contains("time"));
assertEquals(rets.size(), 4);
但是,返回的List将是固定长度,因此我们无法添加或删除元素。注意:java.util.Arrays的方法asList返回的是它自己的ArrayList子类。
6.3. setAll
使用setAll, 设置一个生成器可以为数组初始化元素:
String[] longAgo = new String[4];
Arrays.setAll(longAgo, i -> intro[i]);
assertArrayEquals(longAgo, new String[]{"a", "long", "time", "ago"});
使用lambda表达式也是极易出异常的。
7. Parallel Prefix
Arrays在Java8引入的另一个新方法是parallelPrefix。使用parallelPrefix ,我们可以以累积的方式对输入数组的每个元素进行操作。
7.1. parallelPrefix
做一个累加操作像[1, 2, 3, 4]输出结果[1, 3, 6, 10]:
int[] arr = new int[]{1, 2, 3, 4};
Arrays.parallelPrefix(arr, (left, right) -> left + right);
assertThat(arr, is(new int[]{1, 3, 6, 10}));
我们也可以指定一个范围:
int[] arr = new int[]{1, 2, 3, 4, 5};
Arrays.parallelPrefix(arr, 1, 4, (left, right) -> left + right);
assertThat(arr, is(new int[]{1, 2, 5, 9, 5}));
注意:因为该方法是并行执行的,因此累积操作应该是无状态的。
对于非关联函数:
int nonassociativeFunc(int left, int right) {
return left + right * left;
}
如下会产生不一致的情况:
@Test
public void whenPrefixNonAssociative_thenError() {
boolean consistent = true;
Random r = new Random();
for (int k = 0; k < 100_000; k++) {
int[] arrA = r.ints(100, 1, 5).toArray();
int[] arrB = Arrays.copyOf(arrA, arrA.length);
Arrays.parallelPrefix(arrA, this::nonassociativeFunc);
for (int i = 1; i < arrB.length; i++) {
arrB[i] = nonassociativeFunc(arrB[i - 1], arrB[i]);
}
consistent = Arrays.equals(arrA, arrB);
if(!consistent) break;
}
assertFalse(consistent);
}
7.2. Performance
并行前缀计算通过比顺序循环更高效,尤其对于大型数组。下面是一个基准测试结果:
Benchmark Mode Cnt Score Error Units
largeArrayLoopSum thrpt 5 9.428 ± 0.075 ops/s
largeArrayParallelPrefixSum thrpt 5 15.235 ± 0.075 ops/s
Benchmark Mode Cnt Score Error Units
largeArrayLoopSum avgt 5 105.825 ± 0.846 ops/s
largeArrayParallelPrefixSum avgt 5 65.676 ± 0.828 ops/s
下面是基准测试代码:
@Benchmark
public void largeArrayLoopSum(BigArray bigArray, Blackhole blackhole) {
for (int i = 0; i < ARRAY_SIZE - 1; i++) {
bigArray.data[i + 1] += bigArray.data[i];
}
blackhole.consume(bigArray.data);
}
@Benchmark
public void largeArrayParallelPrefixSum(BigArray bigArray,
Blackhole blackhole) {
Arrays.parallelPrefix(bigArray.data, (left, right) -> left + right);
blackhole.consume(bigArray.data);
}
八、总结
本文中我们了解了如何使用java.util.Arrays类创建、搜索、排序和转换数组的一些方法。
该类在最新的Java版本中进行了扩展,包括Java8中引入了流和Java9中的加强。