當前位置: 妍妍網 > 碼農

還在用 list.contain 做去重?太Low了,該換換了!

2024-05-15碼農

大家好,我是鵬磊。

最近又是一輪程式碼review 發現了一些實作去重的程式碼在使用 list.contain ......

我沈思,是不是其實很多初學者也存在這種去重使用問題?

所以我選擇把這個事情整出來,分享一下。

正文

首先是造出一個 List 模擬數據,一共2W條,裏面有一半數據1W條是重復的:

publicstatic List<String> getTestList(){
List<String> list = new ArrayList<>();
for (int i = 1; i <= 10000; i++) {
list.add(String.valueOf(i));
}
for (int i = 10000; i >= 1; i--) {
list.add(String.valueOf(i));
}
return list;
}

先看看我們用 contain 去重的程式碼:

/**
 * 使用 list.contain 去重
 *
 * @param testList
 */

privatestaticvoiduseContain2Distinct(List<String> testList){
System.out.println("contains 開始去重,條數:" + testList.size());
List<String> testListDistinctResult = new ArrayList<>();
for (String str : testList) {
if (!testListDistinctResult.contains(str)) {
testListDistinctResult.add(str);
}
}
System.out.println("contains 去重完畢,條數:" + testListDistinctResult.size());
}

我們呼叫一下看看耗時:

publicstaticvoidmain(String[] args){
List<String> testList = getTestList();
StopWatch stopWatch = new StopWatch();
stopWatch.start();
useContainDistinct(testList);
stopWatch.stop();
System.out.println("去重 最終耗時" + stopWatch.getTotalTimeMillis());
}

耗時:

評價: list.contain 的效率,我的建議是,知道就行,別用。

眾所周知 Set 不存在重復數據, 所以我們來看看使用 HashSet 去重的效能:

如果你近期準備面試跳槽,建議在ddkk.com線上刷題,涵蓋 一萬+ 道 Java 面試題,幾乎覆蓋了所有主流技術面試題,還有市面上最全的技術五百套,精品系列教程,免費提供。

ps:這裏是采取使用 set 的 add 方法做去重

/**
 * 使用set去重
 *
 * @param testList
 */

privatestaticvoiduseSetDistinct(List<String> testList){
System.out.println("HashSet.add 開始去重,條數:" + testList.size());
List<String> testListDistinctResult = new ArrayList<>(new HashSet(testList));
System.out.println("HashSet.add 去重完畢,條數:" + testListDistinctResult.size());
}

我們呼叫一下看看耗時:

publicstaticvoidmain(String[] args){
List<String> testList = getTestList();
StopWatch stopWatch = new StopWatch();
stopWatch.start();
useSetDistinct(testList);
stopWatch.stop();
System.out.println("去重 最終耗時" + stopWatch.getTotalTimeMillis());
}

耗時:

評價: HashSet 的效率,我的建議是推薦。

為什麽耗時差距這麽大?

不多說,我們看源碼:

list.contains(o):

可以看到裏面用到了 index(o) :

時間復雜度 :O(n) n: 元素個數

那麽我們看看 set.add(o) 是怎麽樣的 :

map的add , 老生常談就不談了,hash完 直接塞到某個位置, 時間復雜度 :O(1)

如果你近期準備面試跳槽,建議在ddkk.com線上刷題,涵蓋 一萬+ 道 Java 面試題,幾乎覆蓋了所有主流技術面試題,還有市面上最全的技術五百套,精品系列教程,免費提供。

所以O(n) 和 O(1) 誰快誰慢?顯然。

ps:順嘴說下 hashset 的 contain

時間復雜度也是:O(1)

那麽我們最後再看看別的去重:

雙for迴圈 ,remove去重

/**
 * 使用雙for迴圈去重
 * @param testList
 */

privatestaticvoiduse2ForDistinct(List<String> testList){
System.out.println("list 雙迴圈 開始去重,條數:" + testList.size());
for (int i = 0; i < testList.size(); i++) {
for (int j = i + 1; j < testList.size(); j++) {
if (testList.get(i).equals(testList.get(j))) {
testList.remove(j);
}
}
}
System.out.println("list 雙迴圈 去重完畢,條數:" + testList.size());
}

publicstaticvoidmain(String[] args){
List<String> testList = getTestList();
StopWatch stopWatch = new StopWatch();
stopWatch.start();
use2ForDistinct(testList);
stopWatch.stop();
System.out.println("去重 最終耗時" + stopWatch.getTotalTimeMillis());
}

耗時:

評價: 知道就行,圖個樂,別用,賊慢,而且程式碼看起來亂

stream的distinct去重:

/**
 * 使用Stream 去重
 *
 * @param testList
 */

privatestaticvoiduseStreamDistinct(List<String> testList){
System.out.println("stream 開始去重,條數:" + testList.size());
List<String> testListDistinctResult = testList.stream().distinct().collect(Collectors.toList());
System.out.println("stream 去重完畢,條數:" + testListDistinctResult.size());
}

publicstaticvoidmain(String[] args){
List<String> testList = getTestList();
StopWatch stopWatch = new StopWatch();
stopWatch.start();
useStreamDistinct(testList);
stopWatch.stop();
System.out.println("去重 最終耗時" + stopWatch.getTotalTimeMillis());
}

耗時:

評價: 還不錯,主要是程式碼也蠻簡潔,有一點點動心。

🔥 磊哥私藏精品 熱門推薦 🔥