[leetcode]3Sum @ Python

3077 단어 LeetCode
원제 주소:http://oj.leetcode.com/problems/3sum/
제목: 한 배열 에서 세 개의 수 를 찾 아 이 세 개의 수 를 0 으로 합 친다.여러 조 의 해 가 존재 할 수도 있 고 중복 되 는 해 가 존재 할 수도 있 기 때문에 다시 해 야 한다.예 를 들 어 num = [- 1, 0, 1, 2, - 1, - 4];그러면 두 조 의 해 가 존재 한다. [- 1, 0, 1], [- 1, - 1, 2]. 해 중의 수 는 어 릴 때 부터 큰 정렬 상태 가 필요 하 다.
문제 풀이 방향: 1. 먼저 배열 을 정렬 합 니 다.
2. 정렬 후 투 썸 의 사고방식 에 따라 문 제 를 풀 수 있다.어떻게 풀 지?num [i] 의 반대 수 인 num [i] 를 target 으로 한 다음 i + 1 에서 len (num) - 1 까지 의 배열 요소 에서 두 개의 수 를 찾 아 합 친 것 을 - num [i] 로 하면 된다.아래 표 시 된 i 의 범 위 는 0 에서 len (num) - 3 입 니 다.
3. 이 과정 은 무 거 운 것 을 주의해 야 한다.
       4. 시간 복잡 도 는 O (N ^ 2) 입 니 다.
코드:
class Solution:
    # @return a list of lists of length 3, [[val1,val2,val3]]
    def threeSum(self, num):
        num.sort()
        res = []
        for i in range(len(num)-2):
            if i == 0 or num[i] > num[i-1]:
                left = i + 1; right = len(num) - 1
                while left < right:
                    if num[left] + num[right] == -num[i]:
                        res.append([num[i], num[left], num[right]])
                        left += 1; right -= 1
                        while left < right and num[left] == num[left-1]: left +=1
                        while left < right and num[right] == num[right+1]: right -= 1
                    elif num[left] + num[right] < -num[i]:
                        while left < right:
                            left += 1
                            if num[left] > num[left-1]: break
                    else:
                        while left < right:
                            right -= 1
                            if num[right] < num[right+1]: break
        return res

좋은 웹페이지 즐겨찾기