两数之和

两数之和

 

/*
给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
*/

//方法一
	public int[] twoSum(int[] nums, int target) {
		int a = nums[0];
		int b = nums[0];
                for(int i = 0; i < nums.length; i++) {
        	    for(int j = i + 1; j < nums.length; j++) {
        		if(nums[i] + nums[j] == target) {
        			a = i;
        			b = j;
        		}
        	    }
                }
                return new int[] {a, b};
        }
	
	//方法二，hash的方式，一遍解决，效率更高
	public int[] second(int[] nums, int target) {
		Map<Integer, Integer> map = new HashMap<>();
	        for(int i = 0; i < nums.length; i++) {
                //map中已经存在该值，判断target是否是nums[i]的两倍即可
	    	if(map.containsKey(nums[i])) {
	    		if(target == nums[i] + nums[i])
	    			return new int[] {map.get(nums[i]), i};
	    	} else {
		        map.put(nums[i], i);
		        int complement = target - nums[i];
		        if(map.containsKey(complement) && i != map.get(complement)) {
		            return new int[] {map.get(complement), i};
		        }
	    	}
	    }
	    throw new IllegalArgumentException("No two sum solution");
	}
	
	@Test
	public void main() {
		int[] nums = new int[]{2, 7, 11, 15};
		int target = 9;
		int[] twoSum = twoSum(nums, target);
		System.out.println(twoSum);
	}