门|门 IT 2007 |第 50 题

这是一道经典的编程题目，旨在考察程序员的算法能力和编程实现能力。

题目描述

给定一个由若干个数字组成的数组，求该数组中所有的三元组（a, b, c），使得a + b + c = 0。

解题思路

该问题可以使用双指针法来解决，具体步骤如下：

1. 将数组进行排序，方便后续的操作；
2. 逐个枚举第一个数，从左到右依次枚举；
3. 确定第一个数后，从它的下一个位置开始，使用双指针法查找另外两个数；
4. 双指针法的过程中，使用指针i和j指向第一个数之后的位置，使用指针k指向数组末尾，如果a[i] + a[j] + a[k] = 0，则说明找到了一个解；
5. 如果a[i] + a[j] + a[k] > 0，则说明解必须向左移动，因为数组已经排过序，故k指针左移；
6. 如果a[i] + a[j] + a[k] < 0，则说明解必须向右移动，因为数组已经排过序，故i和j指针右移；

代码实现

def three_sum(nums):
    """
    :type nums: List[int]
    :rtype: List[List[int]]
    """
    res = []
    nums.sort()
    for i in range(len(nums)-2):
        if i > 0 and nums[i] == nums[i-1]:
            continue
        j, k = i + 1, len(nums) - 1
        while j < k:
            s = nums[i] + nums[j] + nums[k]
            if s < 0:
                j += 1
                while j < k and nums[j] == nums[j-1]:
                    j += 1
            elif s > 0:
                k -= 1
                while j < k and nums[k] == nums[k+1]:
                    k -= 1
            else:
                res.append([nums[i], nums[j], nums[k]])
                j += 1
                k -= 1
                while j < k and nums[j] == nums[j-1]:
                    j += 1
                while j < k and nums[k] == nums[k+1]:
                    k -= 1
    return res

以上就是一种python实现的双指针方法，可以通过leetcode OJ测试(leetcode第15题)。

总结

双指针是一种常用的算法思想，可以有效地降低时间复杂度，因此在解决数组相关问题时可以优先考虑使用双指针法。