如何在 Java 函数中正确返回一个整型数组

在 java 中,函数需通过 `return` 语句直接返回已构建好的数组对象(如 `int[]`),不能使用语法错误的 `new int[]{}array`;应先将 `list` 转为 `int[]`,再直接 `return array;`。

在 LeetCode 第 496 题(Next Greater Element I)等算法题中,函数签名明确要求返回 int[] 类型(例如 public int[] nextGreaterElement(int[] nums1, int[] nums2))。这意味着你必须构造并返回一个合法的整型数组对象,而非打印、拼接字符串或尝试非法语法。

你原代码中的关键错误在于这一行:

return new int[]{}array; // ❌ 编译错误:非法表达式,语法不成立

new int[]{} 是创建一个空数组的字面量,而紧随其后的 array 是一个变量名——Java 不支持这种“拼接式”返回。编译器会报错:unexpected tokenillegal start of expression

✅ 正确做法是:在完成数组构建后,直接返回该数组引用。你已用 Stream 正确完成了 List 到 int[] 的转换:

int[] array = list.stream().mapToInt(i -> i).toArray();
return array; // ✅ 正确:返回已初始化的数组对象

此外,还需注意逻辑缺陷(虽非本题核心,但影响结果正确性):

  • 原内层循环未在匹配到 nums1[i] == nums2[j] 后及时跳出,可能导致重复添加;
  • try-catch 判断 nums2[j+1] 是否更大是错误策略——题目要求的是 该元素在 nums2 中「右侧第一个」严格更大的元素,而非仅看紧邻下一位;正确解法应使用单调栈或预处理 nextGreater 映射。

以下是修复语法 + 优化逻辑的最小可运行版本(满足 LeetCode 接口):

import java.util.*;

public class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // 构建 nums2 中每个元素的「下一个更大元素」映射
        Map nextGreater = new HashMap<>();
        Stack stack = new Stack<>();

        // 从右向左遍历 nums2,维护单调递减栈
        for (int i = nums2.length - 1; i >= 0; i--) {
            while (!stack.isEmpty() && stack.peek() <= nums2[i]) {
                stack.pop();
            }
            nextGreater.put(nums2[i], stack.isEmpty() ? -1 : stack.peek());
            stack.push(nums2[i]);
        }

        // 根据 nums1 构造结果数组
        int[] result = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            result[i] = nextGreater.get(nums1[i]);
        }
        return result; // ✅ 直接返回 int[],无任何包装或语法错误
    }
}

? 关键总结

  • ✅ return 后必须跟一个与方法声明类型兼容的表达式(此处为 int[] 变量或字面量);
  • ❌ 禁止写 new int[]{}xxx、return array[]、return {1,2,3}(除非是方法内联初始化)等非法语法;
  • ✅ 数组一旦创建(如 new int[n] 或 list.stream().toArray()),即可直接 return;
  • ? 调试时可用 System.out.println(Arrays.toString(result)) 查看输出格式(如 [-1, 3, -1]),但提交时务必删除所有 System.out,仅保留 return。

遵循以上规则,即可确保函数合规

返回数组,顺利通过 LeetCode 等平台的自动判题系统。