两数之和(Python + Java)

给定一个整数数组 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]

Python:
思路一:暴力求解:两层for循环遍历列表

def sums(nums,target):
    for i in range(len(nums)):
        num = nums[i]
        for j in range(i+1,len(nums)):
            if nums[j] == target - num:
                return i,j

思路二:采用枚举对象,enumerate() 函数将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标,用字典存储整数及其下标

def enum(nums, target):
    dic = {
    
    }
    for index,el in enumerate(nums):
        if target - el not in dic:   # in操作符用于判断键是否存在于字典中
            dic[el] = index  # 存储数值和下标
        else:
            return [dic[target - el], index]

if __name__ == '__main__':
    nums = [2, 7, 11, 15]
    target = 9
    print(enum(nums,target))

enumerate(nums)返回值:
0 2
1 7
2 11
3 15
输出:[0, 1]

Java:使用HashMap实现键值对的映射

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

// 两数之和
public class TwoSum {
    
    
	public static void main(String[] arg){
    
    
		int[] arr = {
    
    2,7,11,15};
		int target = 9;
		for(int i : twosum(arr, target)){
    
    
			System.out.println(i);
		}
	}
	
	public static int[] twosum(int[] nums,int target) {
    
    
		int[] arr = new int[2];
		Map<Integer, Integer> map = new HashMap<>();
		for(int i = 0;i < nums.length;i++){
    
    
			int num = target - nums[i];
			if(map.containsKey(num)){
    
      //检查 hashMap 中是否存在指定的 key 对应的映射关系
				arr[0] = map.get(num);  //获取指定 key 对应对 value
				arr[1] = i;
			}
			map.put(nums[i], i);
		}
		return arr;
	}
}

猜你喜欢

转载自blog.csdn.net/qq_43325582/article/details/122528634