comments: true
difficulty: 简单
edit_url: https://github.com/doocs/leetcode/edit/main/lcof/%E9%9D%A2%E8%AF%95%E9%A2%9857.%20%E5%92%8C%E4%B8%BAs%E7%9A%84%E4%B8%A4%E4%B8%AA%E6%95%B0%E5%AD%97/README.md
面试题 57. 和为 s 的两个数字
题目描述
输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,则输出任意一对即可。
示例 1:
输入:nums = [2,7,11,15], target = 9 输出:[2,7] 或者 [7,2]
示例 2:
输入:nums = [10,26,30,31,47,60], target = 40 输出:[10,30] 或者 [30,10]
限制:
1 <= nums.length <= 105
1 <= nums[i] <= 106
解法
我们用双指针 l l l 和 r r r 分别指向数组的左右两端,然后不断移动指针,直到找到一组和为 t a r g e t target target 的连续正整数序列。
时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( 1 ) O(1) O(1)。其中 n n n 为数组的长度。
方法一:双指针
Python3
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
l, r = 0, len(nums) - 1
while l < r:
if nums[l] + nums[r] == target:
return [nums[l], nums[r]]
if nums[l] + nums[r] > target:
r -= 1
else:
l += 1
数字非递增的解法
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
st=set()
for n in nums:
if target-n in st:return [n,target-n]
st.add(n)
Java
class Solution {
public int[] twoSum(int[] nums, int target) {
int l = 0, r = nums.length - 1;
while (true) {
if (nums[l] + nums[r] == target) {
return new int[] {nums[l], nums[r]};
}
if (nums[l] + nums[r] > target) {
--r;
} else {
++l;
}
}
}
}
C++
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int l = 0, r = nums.size() - 1;
while (1) {
if (nums[l] + nums[r] == target) {
return {nums[l], nums[r]};
}
if (nums[l] + nums[r] > target) {
--r;
} else {
++l;
}
}
}
};
Go
func twoSum(nums []int, target int) []int {
l, r := 0, len(nums)-1
for {
if nums[l]+nums[r] == target {
return []int{nums[l], nums[r]}
}
if nums[l]+nums[r] > target {
r--
} else {
l++
}
}
}
JavaScript
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function (nums, target) {
let l = 0;
let r = nums.length - 1;
while (1) {
if (nums[l] + nums[r] == target) {
return [nums[l], nums[r]];
}
if (nums[l] + nums[r] > target) {
--r;
} else {
++l;
}
}
};
TypeScript
function twoSum(nums: number[], target: number): number[] {
let l = 0;
let r = nums.length - 1;
while (nums[l] + nums[r] !== target) {
if (nums[l] + nums[r] < target) {
l++;
} else {
r--;
}
}
return [nums[l], nums[r]];
}
Rust
use std::cmp::Ordering;
impl Solution {
pub fn two_sum(nums: Vec<i32>, target: i32) -> Vec<i32> {
let mut l = 0;
let mut r = nums.len() - 1;
loop {
match target.cmp(&(nums[l] + nums[r])) {
Ordering::Less => {
r -= 1;
}
Ordering::Greater => {
l += 1;
}
Ordering::Equal => {
break vec![nums[l], nums[r]];
}
}
}
}
}
C#
public class Solution {
public int[] TwoSum(int[] nums, int target) {
int l = 0, r = nums.Length - 1;
while (true) {
if (nums[l] + nums[r] == target) {
return new int[] {nums[l], nums[r]};
}
if (nums[l] + nums[r] > target) {
--r;
} else {
++l;
}
}
}
}
Swift
class Solution {
func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
var l = 0
var r = nums.count - 1
while l < r {
let sum = nums[l] + nums[r]
if sum == target {
return [nums[l], nums[r]]
} else if sum > target {
r -= 1
} else {
l += 1
}
}
return []
}
}