当前位置：首页 > news >正文

每天一道面试题-两数之和

news 来源：原创 2025/5/10 17:10:57

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

Java实现（哈希表法）

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> numMap = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            if (numMap.containsKey(complement)) {
                return new int[]{numMap.get(complement), i};
            }
            numMap.put(nums[i], i);
        }
        throw new IllegalArgumentException("No solution found");
    }
}

其他方法分析

1. 暴力法（双重循环）

public int[] twoSumBruteForce(int[] nums, int target) {
    for (int i = 0; i < nums.length; i++) {
        for (int j = i + 1; j < nums.length; j++) {
            if (nums[i] + nums[j] == target) {
                return new int[]{i, j};
            }
        }
    }
    throw new IllegalArgumentException("No solution found");
}

时间复杂度：O(n²)，遍历所有可能的元素对。
空间复杂度：O(1)，无需额外存储。
缺点：效率低，不适用于大规模数据。

2. 排序 + 双指针法

import java.util.Arrays;

public int[] twoSumWithSorting(int[] nums, int target) {
    int[][] sorted = new int[nums.length][2];
    for (int i = 0; i < nums.length; i++) {
        sorted[i][0] = nums[i];
        sorted[i][1] = i;
    }
    Arrays.sort(sorted, (a, b) -> Integer.compare(a[0], b[0]));
    
    int left = 0, right = nums.length - 1;
    while (left < right) {
        int sum = sorted[left][0] + sorted[right][0];
        if (sum == target) {
            return new int[]{sorted[left][1], sorted[right][1]};
        } else if (sum < target) {
            left++;
        } else {
            right--;
        }
    }
    throw new IllegalArgumentException("No solution found");
}

时间复杂度：O(n log n)，排序占主导。
空间复杂度：O(n)，需存储原始索引。
缺点：需额外空间存储索引，且会破坏原始数组的顺序。

方法对比

方法	时间复杂度	空间复杂度	适用场景
哈希表法	O(n)	O(n)	通用场景，最优解
暴力法	O(n²)	O(1)	数据量极小
排序双指针	O(n log n)	O(n)	需要有序数据或空间限制较宽松