从Java列表中获取前一项、当前项和下一项的最佳方法

Question

我有一个对象用户

User
 id: Integer
 nickname : String
 score: Integer

以及一份按分数排序的用户列表：

id  name    score
3   John    90
2   Mike    80
4   Davi    30
5   Ana     20
6   Jane    10

我需要一个函数，该函数返回Id由params、前一个值和next传递的用户

例:婴儿车id = 4：

2   Mike    80
4   Davi    30
5   Ana     20

但是当id有最大分数返回时

例:婴儿车id = 2：

3   John    90
2   Mike    80
4   Davi    30

我需要总是返回一个包含3个值的列表

做这个功能的最佳方法是什么？

Answer 1

你不需要什么特别的东西，只是一个简单的循环。

public static List<User> getThree(List<User> input, int id) {
    int i = 0;
    while (i < input.size() && input.get(i).id != id) i++;
    List<User> res = new ArrayList<>();
    res.add(input.get(i - 1));
    res.add(input.get(i));
    res.add(input.get(i + 1));

    return res;
}

当然，这假定不存在非法案例，i - 1或i + 1可能超出范围，但您可以添加检查，例如

if (i - 1 >= 0))

或

if (i + 1 < input.size()) 

在相应的行之前。

Answer 2

可能有多种方法，但这是实现它的方法之一：

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class User{
    int id;
    String nickName;
    int score;
    User(int id, String nickName, int score){
        this.id = id;
        this.nickName = nickName;
        this.score = score;
    }
    
    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append(id).append(" ").append( nickName ).append(" ").append(score);
        return sb.toString();
    }
}

public class JavaList {
    
    Map<Integer, Integer> idxMap;
    List<User> users;
    
    public JavaList() {
        idxMap = new HashMap<>();
        users = new ArrayList<>();
        users.add(new User(3, "John", 90));
        idxMap.put(3, 0);
        users.add(new User(2, "Mike", 80));
        idxMap.put(2, 1);
        users.add(new User(4, "Davi", 30));
        idxMap.put(4, 2);
        users.add(new User(5, "Ana",  20));
        idxMap.put(5, 3);
        users.add(new User(6, "Jane", 10));
        idxMap.put(6, 4);
    }
    
    public List<User> findThreeRecords(int id){
        if (! idxMap.containsKey(id)) throw new IllegalArgumentException("Invalid User id");
        int index = idxMap.get(id);
        List<User> resultList = new ArrayList<>();
        if (index-1 >= 0) resultList.add(users.get(index-1));
        resultList.add(users.get(index));
        if (index < users.size() ) resultList.add(users.get(index+1));
        
        return resultList;
    }

    public static void main(String[] args) {
        JavaList javaList = new JavaList();
        List<User> res = javaList.findThreeRecords(2);
        System.out.println(res);
        //This will print [3 John 90, 2 Mike 80, 4 Davi 30]
    }

}