0167. Tổng hai số II - Mảng đã sắp xếp

Thẻ: Mảng, Con trỏ kép, Tìm kiếm nhị phân
Độ khó: Trung bình

Tóm tắt đề bài

Mô tả: Cho một mảng số nguyên $n u mb ers$ được sắp xếp theo thứ tự tăng dần và một giá trị mục tiêu $t a r g e t$ .

Yêu cầu: Tìm hai số trong mảng có tổng bằng $t a r g e t$ và trả về chỉ số của hai số đó trong mảng.

Giải thích:

$2 \leq numbers.length \leq 3 \times 1 0^{4}$ .
$- 1000 \leq numbers[i] \leq 1000$ .
$n u mb ers$ được sắp xếp theo thứ tự không giảm.
$- 1000 \leq target \leq 1000$ .
Chỉ có một kết quả duy nhất.

Ví dụ:

Ví dụ 1:

Input: numbers = [2,7,11,15], target = 9
Output: [1,2]
Explanation: 2 + 7 = 9. Vì vậy, index1 = 1, index2 = 2. Trả về [1, 2].

Ví dụ 2:

Input: numbers = [2,3,4], target = 6
Output: [1,3]
Explanation: 2 + 4 = 6. Vì vậy, index1 = 1, index2 = 3. Trả về [1, 3].

Ý tưởng

Ý tưởng 1: Tìm kiếm nhị phân

Vì mảng đã được sắp xếp, nên có thể sử dụng tìm kiếm nhị phân để giảm độ phức tạp thời gian. Cách làm cụ thể như sau:

Sử dụng một vòng lặp duy nhất để duyệt qua mảng, lần lượt chọn một số cố định, tức là $n u mb ers [i]$ .
Sau đó, sử dụng tìm kiếm nhị phân để tìm số thứ hai thỏa mãn yêu cầu.
Sử dụng hai con trỏ $l e f t$ và $r i g h t$ . $l e f t$ trỏ đến phần tử đầu tiên của mảng, $r i g h t$ trỏ đến phần tử có giá trị lớn nhất trong mảng.
Kiểm tra tổng của số thứ nhất $n u mb ers [i]$ và số giữa hai con trỏ $n u mb ers [mi d]$ với giá trị mục tiêu.
1. Nếu $n u mb ers [mi d] + n u mb ers [i] < t a r g e t$ , loại bỏ khoảng không thể có $[l e f t, mi d]$ , tiếp tục tìm kiếm trong khoảng $[mi d + 1, r i g h t]$ .
2. Nếu $n u mb ers [mi d] + n u mb ers [i] \geq t a r g e t$ , số thứ hai có thể nằm trong khoảng $[l e f t, mi d]$ , tiếp tục tìm kiếm trong khoảng $[l e f t, mi d]$ .
Tiếp tục cho đến khi $l e f t$ và $r i g h t$ di chuyển đến cùng một vị trí dừng lại. Nếu $n u mb ers [l e f t] + n u mb ers [i] == t a r g e t$ , trả về vị trí của hai số $[l e f t + 1, i + 1]$ (chỉ số bắt đầu từ $1$ ).
Nếu không tìm thấy, trả về $[- 1, - 1]$ .

Ý tưởng 1: Code

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        for i in range(len(numbers)):
            left, right = i + 1, len(numbers) - 1
            while left < right:
                mid = left + (right - left) // 2
                if numbers[mid] + numbers[i] < target:
                    left = mid + 1
                else:
                    right = mid
            if numbers[left] + numbers[i] == target:
                return [i + 1, left + 1]
 
        return [-1, -1]

Ý tưởng 1: Độ phức tạp

Độ phức tạp thời gian: $O (n \times lo g n)$ .
Độ phức tạp không gian: $O (1)$ .

Ý tưởng 2: Con trỏ kép

Có thể sử dụng con trỏ kép để giảm độ phức tạp thời gian. Cách làm cụ thể như sau:

Sử dụng hai con trỏ $l e f t$ và $r i g h t$ . $l e f t$ trỏ đến phần tử đầu tiên của mảng, $r i g h t$ trỏ đến phần tử có giá trị lớn nhất trong mảng.
Kiểm tra tổng của hai phần tử tại vị trí $l e f t$ và $r i g h t$ với giá trị mục tiêu.
1. Nếu tổng bằng giá trị mục tiêu, trả về vị trí của hai phần tử.
2. Nếu tổng lớn hơn giá trị mục tiêu, di chuyển $r i g h t$ sang trái để giảm tổng.
3. Nếu tổng nhỏ hơn giá trị mục tiêu, di chuyển $l e f t$ sang phải để tăng tổng.
Tiếp tục cho đến khi $l e f t$ và $r i g h t$ di chuyển đến cùng một vị trí dừng lại.
Nếu không tìm thấy, trả về $[- 1, - 1]$ .

Ý tưởng 2: Code

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        left = 0
        right = len(numbers) - 1
        while left < right:
            total = numbers[left] + numbers[right]
            if total == target:
                return [left + 1, right + 1]
            elif total < target:
                left += 1
            else:
                right -= 1
        return [-1, -1]

Ý tưởng 2: Độ phức tạp

Độ phức tạp thời gian: $O (n)$ .
Độ phức tạp không gian: $O (1)$ . Chỉ sử dụng một số biến hằng số để lưu trữ.

Hung Nguyen 🧠🌳

Explorer

LeetCode 0167