Java 查找元素在TreeMap中的位置

在Java中，TreeMap类提供了一种有序存储键值对的有效方式。有时，我们可能需要在TreeMap中找到特定元素的位置。在本文中，我们将探讨不同的方法来实现这个任务。我们将讨论语法、算法，并为每种方法提供可执行的代码示例。

语法

要在Java TreeMap中找到元素的位置，我们可以使用以下语法−

int position = Collections.binarySearch(treeMap.values(), element);

语法解释

Collections.binarySearch()策略用于对排序列表进行二次查找。在我们的示例中，我们将TreeMap的值和要查找位置的元素传递给该策略。如果元素在列表中找到，则策略返回该组件的位置列表；否则返回负值。

方法1：使用binarySearch()

步骤

使用values()方法从TreeMap中获取值。
对值使用Collections.binarySearch()进行二分查找。
将结果存储在名为position的变量中。
如果position大于或等于0，则表示元素已找到。否则，元素不在TreeMap中。

示例

import java.util.Collections;
import java.util.TreeMap;

public class TreeMapPositionFinder {
   public static void main(String[] args) {
      TreeMap<Integer, String> treeMap = new TreeMap<>();
      treeMap.put(1, "Apple");
      treeMap.put(2, "Banana");
      treeMap.put(3, "Orange");
      treeMap.put(4, "Mango");

      String element = "Banana";

      int position = Collections.binarySearch(treeMap.values(), element);

      if (position >= 0) {
         System.out.println("Element found at position: " + (position + 1));
      } else {
         System.out.println("Element not found in the TreeMap.");
      }
   }
}

说明

在这种方法中，我们创建一个TreeMap，并用几个键值对来填充它。然后，我们定义我们想要找到的元素，这里是”Banana”。使用Collections.binarySearch()方法在TreeMap的值中搜索该元素。如果找到该元素，我们通过将位置变量加1来打印它的位置。否则，我们显示该元素在TreeMap中未显示。

方法2：使用TreeMap的keySet()和get()方法

步骤

使用keySet()方法从TreeMap获取键集合。
遍历键集合。
检查每个键对应的值是否等于我们要查找的元素。
如果找到匹配项，将对应的键存储在一个名为position的变量中。
如果position不为null，则表示找到了该元素。否则，该元素不存在于TreeMap中。

示例

import java.util.TreeMap;

public class TreeMapPositionFinder {
   public static void main(String[] args) {
      TreeMap<Integer, String> treeMap = new TreeMap<>();
      treeMap.put(1, "Apple");
      treeMap.put(2, "Banana");
      treeMap.put(3, "Orange");
      treeMap.put(4, "Mango");

      String element = "Banana";
      Integer position = null;

      for (Integer key : treeMap.keySet()) {
         if (treeMap.get(key).equals(element)) {
            position = key;
            break;
         }
      }

      if (position != null) {
         System.out.println("Element found at position: " + position);
      } else {
         System.out.println("Element not found in the TreeMap.");
      }
   }
}

输出

Element found at position: 2

解释

在这种方法中，我们再次创建一个TreeMap，并用键值对填充它。我们定义要查找的元素，这里是“Banana”。然后我们使用for-each循环遍历键，检查与每个键关联的值是否与要查找的元素匹配。如果找到匹配项，我们将对应的键存储在position变量中。最后，我们检查position是否为非空，以确定该元素是否在TreeMap中显示。

方法3：使用TreeMap的entrySet()和getValue()方法

步骤

使用entrySet()方法从TreeMap中获取entrySet。
遍历entries。
检查每个entry的值是否等于要查找的元素。
如果找到匹配项，将对应的键存储在名为position的变量中。
如果position不为空，则找到该元素。否则，该元素不在TreeMap中。

示例

import java.util.Map;
import java.util.TreeMap;

public class TreeMapPositionFinder {
   public static void main(String[] args) {
      TreeMap<Integer, String> treeMap = new TreeMap<>();
      treeMap.put(1, "Apple");
      treeMap.put(2, "Banana");
      treeMap.put(3, "Orange");
      treeMap.put(4, "Mango");

      String element = "Banana";
      Integer position = null;

      for (Map.Entry<Integer, String> entry : treeMap.entrySet()) {
         if (entry.getValue().equals(element)) {
            position = entry.getKey();
            break;
         }
      }

      if (position != null) {
         System.out.println("Element found at position: " + position);
      } else {
         System.out.println("Element not found in the TreeMap.");
      }
   }
}

输出

Element found at position: 2

说明

与方法2相似，我们创建一个TreeMap，将其填充，并对需要查找的组件进行特征化。然后，我们使用for-each循环在TreeMap的条目上进行遍历，并检查每个条目的值是否与组件匹配。如果找到匹配项，我们将对应的键存储在位置变量中。最后，我们检查位置是否有效，以确定组件是否显示在TreeMap中。

方法4：使用TreeMap的values()方法和indexOf()

步骤

使用values()方法从TreeMap获取值。
使用indexOf()方法找到元素的索引。
如果索引大于等于0，则找到该元素。否则，该元素不存在于TreeMap中。

示例

import java.util.ArrayList;
import java.util.TreeMap;

public class TreeMapPositionFinder {
   public static void main(String[] args) {
      TreeMap<Integer, String> treeMap = new TreeMap<>();
      treeMap.put(1, "Apple");
      treeMap.put(2, "Banana");
      treeMap.put(3, "Orange");
      treeMap.put(4, "Mango");

      String element = "Mango";

      ArrayList<String> values = new ArrayList<>(treeMap.values());
      int position = values.indexOf(element);

      if (position >= 0) {
         System.out.println("Element found at position: " + (position + 1));
      } else {
         System.out.println("Element not found in the TreeMap.");
      }
   }
}

输出

Element found at position: 4

解释

在这种方法中，我们首先创建一个TreeMap并填充它。我们对需要找到的组件进行特征描述，在本例中是“Banana”。然后使用values()方法创建一个包含TreeMap值的ArrayList。我们使用indexOf()方法在ArrayList中找到组件的索引。如果索引大于或等于0，我们打印出组件的位置。否则，我们表示该元素在TreeMap中没有显示。