Java算法题(来自lettcode)----01两数之和

题目是这样的


 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
   给定 nums = [2, 7, 11, 15], target = 9
   因为 nums[0] + nums[1] = 2 + 7 = 9
   所以返回 [0, 1]

   自己初步的想法我相信多数人也应该想到了,就是两个for循环搞定。


public class TwoNumSum {

	public int[] twoSum(int[] nums, int target) {
//		int[] result = new int[2];
//		for (int i = 0; i < nums.length ; i++) {
//			for (int j = i + 1; j < nums.length ; j++) {
//				if (nums[i] + nums[j] == target) {
//					result[0] = nums[i];
//					result[1] = nums[j];
//					break;
//				}
//			}
//		}
		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 two solution");
	}
}

下面是lettcode的两个更好的解法

public class TwoNumSum2 {

	public int[] twoSum(int[] nums, int target) {
		Map<Integer, Integer> map = new HashMap<>();
		for (int i = 0; i < nums.length; i++) {
			map.put(nums[i], i);
		}
		for (int i = 0; i < nums.length; i++) {
			int temp = target - nums[i];
			System.out.println("temp:" + temp);
			if (map.containsKey(temp) && (map.get(temp) != i)) {
				return new int[] { i, map.get(temp) };
			}
		}
		throw new IllegalArgumentException("No two sum solution");
	}
}

 还有一种

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

猜你喜欢

转载自blog.csdn.net/loli_kong/article/details/88146228