四数之和Python解法
生活随笔
收集整理的這篇文章主要介紹了
四数之和Python解法
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
給你一個(gè)由 n 個(gè)整數(shù)組成的數(shù)組 nums ,和一個(gè)目標(biāo)值 target 。請(qǐng)你找出并返回滿(mǎn)足下述全部條件且不重復(fù)的四元組 [nums[a], nums[b], nums[c], nums[d]] (若兩個(gè)四元組元素一一對(duì)應(yīng),則認(rèn)為兩個(gè)四元組重復(fù)):
0 <= a, b, c, d?< n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意順序 返回答案 。
來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/4sum
?
例:
輸入:nums = [1,0,-1,0,-2,2], target = 0 輸出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]解析:四數(shù)之和大致和三數(shù)之和相同,在三數(shù)之和外層多套一層即可。為了避免重復(fù),需要對(duì)相同的值進(jìn)行判定。最外層大循環(huán),內(nèi)層雙指針,然后指針中間便利對(duì)比判斷。
class Solution(object):def fourSum(self, nums, target):""":type nums: List[int]:type target: int:rtype: List[List[int]]"""nums.sort() # 排序n = len(nums) # 長(zhǎng)度res = [] # 初始化結(jié)果for i in range(n): # 最外層if i > 0 and nums[i] == nums[i - 1]: continue # 排除重復(fù)for k in range(i+1, n): # 第二個(gè)值if k > i + 1 and nums[k] == nums[k-1]: continuep = k + 1 # 第三個(gè)值q = n - 1 # 第四個(gè)值while p < q:if nums[i] + nums[k] + nums[p] + nums[q] > target: q -= 1elif nums[i] + nums[k] + nums[p] + nums[q] < target: p += 1else:res.append([nums[i], nums[k], nums[p], nums[q]])while p < q and nums[p] == nums[p + 1]: p += 1 # 去重while p < q and nums[q] == nums[q - 1]: q -= 1 # 去重p += 1q -= 1return res總結(jié)
以上是生活随笔為你收集整理的四数之和Python解法的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 最接近的三数之和Python解法
- 下一篇: 有效的括号Python解法