Python 3 LeetCode三数之和

 

大体思路:给原始数据排序,然后值在负数部分循环,去除相同值得时候因为排过序,所以相同值都挨在一起,比较方便去除相同值。注意三个数都应该判断是否有相同值。

# -*- coding :UTF-8 -*-

class Solution:
    def threeSum(self, nums):
        result = []
        temp_nums = sorted(nums)
        l = len(temp_nums)
        for i in range(l - 2):
            if i > 0 and temp_nums[i] == temp_nums[i -1]:
                continue
            else:
                if temp_nums[i] <= 0:#只循环负数
                    j = i + 1
                    k = l - 1
                    while j < k:
                        x = temp_nums[i] + temp_nums[j] + temp_nums[k]
                        if x == 0:
                            while j < k and temp_nums[j] == temp_nums[j+1]:#去掉重复部分左边
                                j += 1
                            while j < k and temp_nums[k] == temp_nums[k-1]:#去掉重复部分右边
                                k -= 1
                            result.append((temp_nums[i], temp_nums[j], temp_nums[k]))
                            j += 1
                        elif x > 0:
                            k -= 1
                        else:
                            j += 1
                else:
                    break
        return result

 

你可能感兴趣的:(Python 3 LeetCode三数之和)