集合-向集合中添加元素

Question

我有一個從數據庫中獲取的學生列表。 我想要根據學生的年齡來構造該列表，最后要有一個像這樣的列表：

[
  30 => [
    "id1" => "Name1",
    "id2" => "Name2",
  ],
  31 => [
    "id5" => "Name3",
    "id6" => "Name4",
  ]
]

在我的代碼中，我收到這樣的學生列表：

List<ApplicantModel> applicants = applicantDao.getApplicantsByApplicationStatus(applicantTypeId);

我想在這里在“ for”內創建集合：

for(int i=0;i<=applicants.size()-1;i++){
    //code here to create the collection with the above structure 
}

或者，如果您還有其他更好的建議？ 謝謝！

Answer 1

您可以為此使用TreeMap（根據年齡和ID的順序）：看起來可能像這樣：（我不知道獲取年齡和ID的方法的確切名稱-因此您可能需要調整getAge和getId方法來電）：

import java.util.LinkedList;
import java.util.List;
import java.util.TreeMap;

public class ApplicantStructureTest {

    public static void main(String[] args) {
        List<ApplicantModel> applicants = new LinkedList<>();
        applicants.add(new ApplicantModel("id1", 30, "name1"));
        applicants.add(new ApplicantModel("id2", 30, "name2"));
        applicants.add(new ApplicantModel("id5", 31, "name3"));
        applicants.add(new ApplicantModel("id6", 31, "name4"));

        // here is your for loop
        TreeMap<Integer, TreeMap<String, String>> structured = new TreeMap<Integer, TreeMap<String, String>>();
        for (int i = 0; i <= applicants.size() - 1; i++) {
            ApplicantModel applicant = applicants.get(i);
            Integer age = applicant.getAge();
            TreeMap<String, String> ageMap = structured.get(age);
            if (ageMap == null) {
                ageMap = new TreeMap<String, String>();
                structured.put(age, ageMap);
            }
            ageMap.put(applicant.getId(), applicant.getName());
        }


        System.out.println(structured);
    }

    public static class ApplicantModel {
        private Integer age;
        private String id;
        private String name;

        public ApplicantModel(String id, Integer age, String name) {
            this.id = id;
            this.age = age;
            this.name = name;
        }

        public String getName() {
            return name;
        }

        public Integer getAge() {
            return age;
        }

        public String getId() {
            return id;
        }

    }

}

此代碼將打印：

{30 = {id1 = name1，id2 = name2}，31 = {id5 = name3，id6 = name4}}

Answer 2

您還可以在地圖和列表中創建結構：

Map<Integer,List<ApplicantModel>> structured = new HashMap<Integer, List<ApplicantModel>>();
for(int i = 0; i <= applicants.size() - 1; i++){
  ApplicantModel applicant = applicants.get(i);

  Integer age = applicant.getAge();
  List<ApplicantModel> list = structured.get(age);
  if (list == null) {
    list = new  List<ApplicantModel>();
    structured.put(age, list);
  }
  list.add(applicant);
}

Answer 3

也可以通過流（Java 8或更高版本）獲得與Krzysztof Cichocki相似的結果：

    Map<Integer, Map<String, String>> byAge = studentList.stream()
            .collect(Collectors.groupingBy(ApplicantModel::getAge, 
                                           Collectors.toMap(ApplicantModel::getId, ApplicantModel::getName)));

這樣會產生（我承認，這種格式不太容易閱讀）：

{30={id2=name2, id1=name1}, 31={id6=name4, id5=name3}}

如果您特別想要TreeMap ，請使用三參數groupingBy()和/或四參數toMap() ：

    Map<Integer, Map<String, String>> byAge = studentList.stream()
            .collect(Collectors.groupingBy(ApplicantModel::getAge, 
                                           TreeMap::new, 
                                           Collectors.toMap(ApplicantModel::getId,
                                                            ApplicantModel::getName,
                                                            (u, v) -> { throw new IllegalArgumentException(); },
                                                            TreeMap::new)));

現在元素從toString()排序出來：

{30={id1=name1, id2=name2}, 31={id5=name3, id6=name4}}