如何利用TreeSet进行数据统计

TreeSet 是 Java 集合框架中的一种实现 SortedSet 接口的类，它基于红黑树（一种自平衡二叉查找树）实现。TreeSet 中的元素会自动按照它们的自然顺序或者通过提供的 Comparator 进行排序。

虽然 TreeSet 本身并不直接提供数据统计功能，但你可以通过遍历 TreeSet 并结合其他数据结构或逻辑来实现一些基本的数据统计任务。以下是一些示例：

示例 1：计算元素出现的次数

假设你有一个 TreeSet，其中包含一些重复的元素，你想计算每个元素出现的次数。由于 TreeSet 不允许重复元素，你需要使用其他数据结构（如 HashMap）来辅助统计。

import java.util.*;

public class TreeSetStatistics {
    public static void main(String[] args) {
        TreeSet<Integer> treeSet = new TreeSet<>(Arrays.asList(1, 2, 2, 3, 3, 3, 4, 4, 4, 4));
        Map<Integer, Integer> frequencyMap = new HashMap<>();

        for (Integer element : treeSet) {
            frequencyMap.put(element, frequencyMap.getOrDefault(element, 0) + 1);
        }

        for (Map.Entry<Integer, Integer> entry : frequencyMap.entrySet()) {
            System.out.println("Element: " + entry.getKey() + ", Frequency: " + entry.getValue());
        }
    }
}

示例 2：计算某个范围内的元素数量

如果你想计算 TreeSet 中在某个范围内的元素数量，可以使用 subSet 方法。

import java.util.*;

public class TreeSetRangeStatistics {
    public static void main(String[] args) {
        TreeSet<Integer> treeSet = new TreeSet<>(Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10));
        int lowerBound = 3;
        int upperBound = 7;

        NavigableSet<Integer> subSet = treeSet.subSet(lowerBound, true, upperBound, true);
        System.out.println("Number of elements in range [" + lowerBound + ", " + upperBound + "]: " + subSet.size());
    }
}

示例 3：计算元素的总和或平均值

如果你想计算 TreeSet 中所有元素的总和或平均值，可以使用增强的 for 循环遍历集合并进行累加。

import java.util.*;

public class TreeSetSumAndAverage {
    public static void main(String[] args) {
        TreeSet<Integer> treeSet = new TreeSet<>(Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10));
        int sum = 0;
        for (Integer element : treeSet) {
            sum += element;
        }
        double average = (double) sum / treeSet.size();

        System.out.println("Sum: " + sum);
        System.out.println("Average: " + average);
    }
}

这些示例展示了如何利用 TreeSet 进行一些基本的数据统计任务。根据你的具体需求，你可以结合其他 Java 集合和工具类来实现更复杂的数据统计功能。