Java数组技巧：找首尾数字相同的元素

本文将介绍如何使用Java编写一个方法，该方法接收一个整数数组作为输入，并返回一个包含所有首尾数字相同的元素的数组。我们将通过将数字转换为字符串，并比较字符串的首尾字符来实现此功能，同时避免重复添加元素。

方法实现

实现该功能的核心思路是将整数转换为字符串，然后比较字符串的首尾字符。为了避免重复添加相同的数字，我们可以使用一个布尔数组来记录哪些数字已经被添加到结果列表中。

以下是完整的Java代码实现：

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

public class FindSameFirstAndLastDigit {

    public int[] findNumbersWithSameFirstAndLastDigits(int[] a) {
        List result = new ArrayList<>();
        boolean[] numbersThatHaveBeenAdded = new boolean[a.length];

        for (int i = 0; i < a.length; i++) {
            for (int j = i + 1; j < a.length; j++) {

                String iNumber = String.valueOf(a[i]);
                String jNumber = String.valueOf(a[j]);

                if (iNumber.charAt(0) == jNumber.charAt(0)
                    && iNumber.charAt(iNumber.length() - 1) == jNumber.charAt(jNumber.length() - 1)) {

                    if (!numbersThatHaveBeenAdded[i]) {
                        result.add(a[i]);
                        numbersThatHaveBeenAdded[i] = true;
                    }
                    if (!numbersThatHaveBeenAdded[j]) {
                        result.add(a[j]);
                        numbersThatHaveBeenAdded[j] = true;
                    }
                }
            }
        }

        return result.stream()
            .mapToInt(Integer::intValue)
            .toArray();
    }

    public static void main(String[] args) {
        FindSameFirstAndLastDigit finder = new FindSameFirstAndLastDigit();
        int[] inputArray = {121, 232, 13, 454, 67, 787, 11, 222};
        int[] resultArray = finder.findNumbersWithSameFirstAndLastDigits(inputArray);

        System.out.print("Result Array: ");
        for (int num : resultArray) {
            System.out.print(num + " ");
        }
        System.out.println();
    }
}

代码解释：

1. findNumbersWithSameFirstAndLastDigits(int[] a) 方法:

   • 接收一个整数数组 a 作为输入。
   • 创建一个 ArrayList 类型的 result 用于存储符合条件的数字。
   • 创建一个 boolean 类型的数组 numbersThatHaveBeenAdded，用于标记哪些数字已经被添加到 result 中，避免重复添加。
   • 使用两层嵌套循环遍历数组 a 中的所有数字对。
   • 将每个数字转换为字符串，分别赋值给 iNumber 和 jNumber。
   • 使用 charAt(0) 获取字符串的首字符，使用 charAt(iNumber.length() - 1) 获取字符串的尾字符。
   • 如果两个数字的首尾字符相同，并且该数字还没有被添加到 result 中，则将其添加到 result 中，并将 numbersThatHaveBeenAdded 数组中对应的位置设置为 true。
   • 最后，将 result 转换为 int[] 并返回。

2. main 方法 (示例):

   • 创建 FindSameFirstAndLastDigit 类的实例。
   • 定义一个示例输入数组 inputArray。
   • 调用 findNumbersWithSameFirstAndLastDigits 方法，并将结果存储在 resultArray 中。
   • 打印 resultArray 中的内容。

优化建议

• 性能优化： 对于大型数组，将数字转换为字符串可能会影响性能。可以考虑使用取模运算和除法运算来提取首尾数字，避免字符串转换。但这种方法可能更复杂，可读性也会降低。
• 错误处理： 可以添加对输入数组的校验，例如检查数组是否为空，或者数组元素是否为负数。

总结

本文详细介绍了如何使用Java编写一个方法，用于查找数组中首尾数字相同的元素。通过将数字转换为字符串，并比较字符串的首尾字符，我们可以轻松地实现该功能。同时，使用布尔数组可以避免重复添加相同的数字。希望本文能够帮助你理解和掌握该问题的解决方法。