HashMap解決雙層For循環效率問題

原創

2020-02-21 08:16

業務場景中很經常類似的需求，對兩個List相同的數據進行處理。例子如下：

import lombok.Builder;
import lombok.Data;
import lombok.experimental.Tolerate;

/**
 * @author 954L
 * @create 2019/12/31 20:42
 */
@Data
@Builder
public class User {

    private String name;

    @Tolerate
    public User(){}

}

import java.util.Arrays;
import java.util.List;

/**
 * @author 954L
 * @create 2019/12/31 20:42
 */
public class Main {

    public static void main(String[] args) {
        User zs = User.builder().name("張三").build();
        User ls = User.builder().name("李四").build();
        List<User> list1 = Arrays.asList(zs);
        List<User> list2 = Arrays.asList(ls);
        for (User user1: list1){
            for (User user2 : list2){
                if (user1.getName().equals(user2.getName())){
                    //TODO 執行業務邏輯
                    break;
                }
            }
        }
    }
}

假設大數據量情況下上述執行代碼list2則會出現很多不必要的循環，直到equals爲止，假設使用map，直接getKey出相同的list2中的對象，則效率可以大大提升！

測試10w條數據雙層for循環耗時

import java.util.ArrayList;
import java.util.List;

/**
 * @author 954L
 * @create 2019/12/31 20:42
 */
public class Main {

    public static void main(String[] args) {
        List<User> list1 = new ArrayList<>();
        for (int i = 0; i <= 100000; i++){
            list1.add(User.builder().name("張三" + i).build());
        }
        List<User> list2 = new ArrayList<>(list1);
        long startTime = System.currentTimeMillis();
        for (User user1: list1){
            for (User user2 : list2){
                if (user1.getName().equals(user2.getName())){
                    //TODO 執行業務邏輯
                    break;
                }
            }
        }
        System.out.println(System.currentTimeMillis() - startTime);
    }
}

測試HashMap耗時時間

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author 954L
 * @create 2019/12/31 20:42
 */
public class Main {

    public static void main(String[] args) {
        List<User> list1 = new ArrayList<>();
        for (int i = 0; i <= 100000; i++){
            list1.add(User.builder().name("張三" + i).build());
        }
        List<User> list2 = new ArrayList<>(list1);
        Map<String, User> list2Map = list2.stream()
                .collect(Collectors.toMap(User::getName, Function.identity()));
        long startTime = System.currentTimeMillis();
        for (User user1: list1){
            User user = list2Map.get(user1.getName());
            if (user != null){
                //TODO 執行業務邏輯
            }
        }
        System.out.println(System.currentTimeMillis() - startTime);
    }
}