【算法题】数组中找到两个数,使其和等于某个数

From : http://leetcode.com/onlinejudge

Two Sum

Given an array of integers, find two numbers such that they add up to a specific target number.

The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.
You may assume that each input would have exactly one solution.
Input: numbers={2, 7, 11, 15}, target=9

Output: index1=1, index2=2


    今天在http://leetcode.com/onlinejudge上碰到该题,由于题中没有提出时间复杂度的要求,直接用的最简单的做法——依次求数组两两元素的和,找到结果等于target的两个数,很明显,该算法的时间复杂度为O(n^2)。

    显然,该解法不是该题最理想的答案,希望能给出时间复杂度更优的解,《编程之美》2.12对该题进行了讲解,给出了不同的时间复杂度的解法。

  1. O(n*log(n))的解法:先对数组进行排序(时间复杂度为O(n*log(n))),在排序的数组中依次查找target-arr[i]是否在数组中,查找target-arr[i]时可以用二分查找(时间复杂度为O(log(n))),遍历数组一遍知道找到target-arr[i]在数组中为止,所以查找的时间复杂度为O(n*log(n)),总的时间复杂度为O(n*log(n))。
  2. 另一种O(n*log(n))的解法:《编程之美》2.12给出的解法三比较有意思,比1有一些改进,主要改进了查找的过程。首先也是要对数组进行排序(时间复杂度为O(n*log(n))),然后再已排序的数组中,令i=0,j=n-1,判断arr[i]+arr[j]是否等于target,等于则结束,如果小于target,则i=i+1;如果大于target,则j=j-1,这样查找符合条件的两个元素的过程只需要遍历数组一次,因此查找的时间复杂度降低到O(n),总的时间复杂度还是O(n*log(n))。
  3. O(n)的解法:将数组中的所有元素存入hash表中,遍历一遍时间复杂度O(n),然后再遍历一遍数组,查找target-arr[i]是否出现在hash表中,因为在hash表中查找的过程只需O(1)时间,所以第二次遍历的时间复杂度也是O(n),总的时间复杂度为O(n)。但是该方法也是有弊端的(也是hash表固有的弊端),就是较高的空间复杂度——需要O(n)的hash表存储空间,例如{1,3,6,100000}这样一个数组就需要用100000的空间来存储,存储效率太低太低!另外说明一下,因为这里只需要标记数是否出现过,所以这里用Bitmap数据结构(见:)可以适当的节省空间。hash表是一种用空间换时间的方法,所以该题需要根据实际情况选取合适的方法。
    另外,http://www.dewen.org/q/6111中一位朋友给出了这样的答案,代码如下:
#include <iostream>
#include <map>
using namespace std;

int main()
{
	int a[] = {1,9,5,6,3,4,23,456,32,2,41,23,34}; //数组
	map<int,int> m;
	int sum = 10; //给定的整数
	int count = sizeof(a)/sizeof(int);
	for(int i=0; i < count; i++) {
		m.insert(map<int,int>::value_type(sum-a[i], a[i])); //放到hash map中
	}
	for(int i=0; i < count; i++) {
		if(m.find(a[i])->second == sum-a[i]) cout << "find " << i <<endl; //hash值查找
	}
	return 0;
}

C++大概的实现就是这样了。

    显然他想用上面3中的hash表的方法,但是这里使用的是STL中的map数据结构,而STL中的map底层是用红黑树实现的,所以map中查找的时间复杂度是不是O(log(n))呢?因此该程序实际的时间复杂度是不是还是O(n*log(n))呢?

目描述: 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2,请你找出这两个有序数组的中位,并且要求算法的时间复杂度为 O(log(m + n))。 示例 1: nums1 = [1, 3] nums2 = [2] 则中位是 2.0 示例 2: nums1 = [1, 2] nums2 = [3, 4] 则中位是 (2 + 3)/2 = 2.5 解思路: 1. 将两个数组合并成一个数组,再对合并后的数组进行排序,找到中位 时间复杂度为O(m+n)log(m+n),不符合目要求 2. 利用二分法查找 首先,我们可以理解中位的定义:如果某个有序数组的长度是奇,那么其中位就是最中间那个元素,如果是偶,那么就是最中间两个元素的平均值。 在本中,我们需要找到两个有序数组 A 和 B 的中位,我们可以考虑从中位的定义入手。如果我们能够将两个数组分别分成部分,并且使得左半部分和右半部分分别满足以下两个条件: 1.左半部分的所有元素都小于右半部分的所有元素。 2.左半部分和右半部分分别包含了 A 和 B 数组的一半元素。 那么中位就可以通过以下公式得到: median = max(left_part) (len(A) + len(B) 为偶) median = (max(left_part) + min(right_part))/2 (len(A) + len(B) 为奇) 为了使得划分满足上述两个条件,我们可以对较短的那个数组进行二分查找,然后根据查找到的位置将两个数组进行划分,最后再根据上述公式计算中位。 具体实现见代码。 Java代码: class Solution { public double findMedianSortedArrays(int[] nums1, int[] nums2) { int m = nums1.length; int n = nums2.length; //确保nums1比nums2短 if (m > n) { int[] temp = nums1; nums1 = nums2; nums2 = temp; int tmp = m; m = n; n = tmp; } int iMin = 0, iMax = m; while (iMin <= iMax) { int i = (iMin + iMax) / 2; int j = (m + n + 1) / 2 - i; if (i < iMax && nums2[j - 1] > nums1[i]) { iMin = i + 1; } else if (i > iMin && nums1[i - 1] > nums2[j]) { iMax = i - 1; } else { int maxLeft = 0; if (i == 0) { maxLeft = nums2[j - 1]; } else if (j == 0) { maxLeft = nums1[i - 1]; } else { maxLeft = Math.max(nums1[i - 1], nums2[j - 1]); } if ((m + n) % 2 == 1) { return maxLeft; } int minRight = 0; if (i == m) { minRight = nums2[j]; } else if (j == n) { minRight = nums1[i]; } else { minRight = Math.min(nums1[i], nums2[j]); } return (maxLeft + minRight) / 2.0; } } return 0.0; } }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值