寻找和为定值的两个数

题目描述

输入一个数组和一个数字，在数组中查找两个数，使得它们的和正好是输入的那个数字。

要求时间复杂度是O(N)。如果有多对数字的和等于输入的数字，输出任意一对即可。

例如输入数组1、2、4、7、11、15和数字15。由于4+11=15，因此输出4和11。

package com.cb.java.algorithms.programmingmethod.array;

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

import com.cb.java.algorithms.datastructure.yanweimindatastructure.recursive.BinaryFind;
import com.cb.java.algorithms.sort.QuickSort;

/**
 * 寻找和为定值的两个数
 * 
 * @author 36184
 *
 */
public class FindTwoSum {

	/**
	 * 暴力查找复杂度为O(N^2)
	 * 
	 * @param arr
	 * @param num
	 */
	public static void findTwo(int[] arr, int num) {
		if (arr == null || arr.length <= 0) {
			return;
		}
		for (int i = 0; i < arr.length; i++) {
			for (int j = i; j < arr.length; j++) {
				if (arr[i] + arr[j] == num) {
					System.out.println(arr[i] + " " + arr[j]);
					break;
				}
			}
		}
	}

	/**
	 * 如果数组无序先排序，然后二分查找<br>
	 * 复杂度同样为O(Nlog N)
	 * 
	 * @param arr
	 * @param num
	 */
	public static void findTwo1(int[] arr, int num) {
		if (arr == null || arr.length <= 0) {
			return;
		}
		QuickSort.quickSort(arr, 0, arr.length - 1);
		for (int i = 0; i < arr.length; i++) {
			int other = num - arr[i];
			int index = BinaryFind.find(arr, other, 0, arr.length - 1);
			if (index != -1) {
				System.out.println(arr[i] + " " + other);
				break;
			}
		}
	}

	/**
	 * 如果数组是无序的，先排序(N log N)，<br>
	 * 然后用两个指针i，j，各自指向数组的首尾两端，令i=0，j=n-1，然后i++，j--，逐次判断a[i]+a[j]?=sum，
	 * 如果某一刻a[i]+a[j] > sum，则要想办法让sum的值减小，所以此刻i不动，j--； 如果某一刻a[i]+a[j] <
	 * sum，则要想办法让sum的值增大，所以此刻i++，j不动。 所以，数组无序的时候，时间复杂度最终为O(N log N + N)=O(N log
	 * N)。 如果原数组是有序的，则不需要事先的排序，直接用两指针分别从头和尾向中间扫描，O(N)搞定，且空间复杂度还是O(1)。
	 * 
	 * @param arr
	 * @param num
	 */
	public static void findTwo2(int[] arr, int num) {
		if (arr == null || arr.length <= 0) {
			return;
		}
		QuickSort.quickSort(arr, 0, arr.length - 1);
		int i = 0;
		int j = arr.length - 1;
		while (i < j) {
			if (arr[i] + arr[j] == num) {
				System.out.println(arr[i] + " " + arr[j]);
				break;
			} else if (arr[i] + arr[j] > num) {
				j--;
			} else {
				i++;
			}
		}
	}

	/**
	 * 使用hash表
	 * 
	 * @param arr
	 * @param num
	 */
	public static void findTwo3(int[] arr, int num) {
		if (arr == null || arr.length <= 0) {
			return;
		}
		Map<Integer, Integer> kv = new HashMap<>();
		for (int i = 0; i < arr.length; i++) {
			if (!kv.containsKey(arr[i])) {
				kv.put(arr[i], 1);
			} else {
				kv.put(arr[i], kv.get(arr[i]) + 1);
			}
		}

		for (int i = 0; i < arr.length; i++) {
			if (kv.get(num - arr[i]) != null) {
				System.out.println(arr[i] + " " + (num - arr[i]));
				break;
			}
		}
	}

	public static void main(String[] args) {
		int[] arr = { 3, 2, 1, 4, 7, 11, 15 };
		findTwo(arr, 15);
		findTwo1(arr, 15);
		findTwo2(arr, 15);
		findTwo3(arr, 15);
	}
}