【劍指offer】面試題57. 和爲s的兩個數字

思路

剛開始自己的想法,對數組進行遍歷,取第一個數時,通過二分法來找target-nums[i],但發現最終這樣的代碼超時,時間複雜度O(nlogn)

改進的想法
使用雙指針法,左指針指向最小元素,右指針指向最大元素,
比較兩個指針所指元素的和與target的大小,如果相等,則說明找到,如果sum>target,則左指針++,反之,右指針++

代碼

正確解法

class Solution {
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		vector<int> ans;
		int sum = 0;
		int left = 0, right = nums.size()-1;
		while (left<right)
		{
			sum = nums[left] + nums[right];
			if (sum==target)
			{
				ans.push_back(nums[left]);
				ans.push_back(nums[right]);
				break;
			}
			if (sum > target) right--;
			else left++;
		}
		return ans;
	}
};

超時解法

class Solution {
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		vector<int> ans(2);
		for (int i=0;i<nums.size();i++)
		{
			ans.push_back(nums[i]);
			if (binarySearch(nums, target - nums[i])) {
				ans.push_back(target - nums[i]);
				return ans;
			}
			ans.pop_back();
		}
		return ans;
	}
	bool binarySearch(vector<int> nums, int t)
	{
		int left = 0;
		int right = nums.size() - 1;
		while (left<=right)
		{
			int mid = left + (right - left) / 2;
			if (nums[mid] == t) return true;
			else if (nums[mid] > t) {//向左找
				right = mid - 1;
			}else if (nums[mid]<t)//向右找
			{
				left = mid + 1;
			}
		}
		return false;
	}
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章