IOS 算法 三數(shù)之和求解問題
IOS 算法三數(shù)之和求解問題
1、三數(shù)求和簡單介紹
對于一個整數(shù)的數(shù)組, 是否存在a, b, c 使得 a + b + c = 0, 返回a b c 數(shù)組,相同數(shù)組只返回一個,:
例如:
[-1, -2, 6, 5, 0, 1, 2, -1, -1] 返回 [[-1, 0, 1], [-2, 0, 2], [-1, -1, 2]]
關鍵點:
① 找到和為0的三個數(shù)
② 去除相同項, 比如: 上面的數(shù)組 其實 [-1, 0, 1], 有三組, 但是我們只要添加1組
千萬不要用 for循環(huán)
又套一層for循環(huán)
處理這個問題, 有些認為兩層for循環(huán)求解, 可以啊, 一層尋找A, 2層尋找B, 判斷數(shù)組是否存在C = - (A + B),
思路正確, 但是這種時間復雜度很高O(n^2)
而且上手時候你會發(fā)現(xiàn), 去重問題處理起來比較繁瑣
方法思路是:
數(shù)組nums 先正序排列
然后for循環(huán)
, 設置最小值下標 low = i + 1
, 最大值下標 high = nums.count - 1
最大值, 最小值 不斷收縮查找, 重復的去掉 且始終保持 low < high
(因為是正序排列 大值 >= 小值)
使得 0 - nums[i] = nums[low] + nums[high] (即: 0 = nums[low] + nums[high] + nums[i])
創(chuàng)建新數(shù)組 添加符合條件的 [nums[low], nums[high], nums[i]], 循環(huán)結束返回即可
接下來我們看下代碼
2、代碼
let num:[Int] = [0, 0, 0, 0, -1, 0, 1, 9, 7, 4] print("返回結果: \(self.caculate(nums: num))") func caculate(nums: [Int]) -> [[Int]] { //數(shù)組元素小于2直接返回 if nums.count < 2 { return [] } //創(chuàng)建空數(shù)組, 用來添加 [A,B,C] var result:[[Int]] = [] //將原數(shù)組數(shù)組正序排列, 這一步很重要, 亂序數(shù)組很難處理 let sort:[Int] = nums.sorted() //循環(huán)正序數(shù)組 for i in 0..<sort.count-1 { //創(chuàng)建最小值下標, 最大值下標 var low:Int = i+1 var high:Int = sort.count - 1 //A+B+C=0 定義-C 為了之后讓 A+B=-C let target:Int = 0 - sort[i] //如果兩個數(shù)相等直接跳過下一次循環(huán) if i>0 && sort[i] == sort[i-1] { continue } //始終保證 最大值下標 > 最小值下標 //思路就是最大值不減小, 最小值不斷增大, 最小值不會超過最大值 //直到找到對應值, 相同值去重 while low < high { //創(chuàng)建sum為: 兩數(shù)字和 A+B let sum:Int = sort[low] + sort[high] //如果A+B == -C 即 A+B+C == 0 if sum == target { //數(shù)組添加新元素 result.append([sort[low], sort[high], sort[i]]) //如果當前最小值和下一位相等, 下標往前移位, 去重 while low < high && sort[low] == sort[low + 1] { low += 1 } //如果當前最大值和前一位相等, 下標往前移位, 去重 while low < high && sort[high] == sort[high - 1] { high -= 1 } //最小值向后移動一位, 最大值向前移動一位 繼續(xù)收縮, 直到跳出while low += 1 high -= 1 }else if sum < target{ //如果A+B == -C 即 A+B+C == 0 low += 1 }else { //如果A+B == -C 即 A+B+C == 0 high -= 1 } } } return result }
返回結果: [[0, 1, -1], [0, 0, 0]]
到此這篇關于IOS 算法之三數(shù)之和求解問題的文章就介紹到這了,更多相關IOS 算法三數(shù)之和求解內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
詳解iOS應用UI開發(fā)中的九宮格坐標計算與字典轉換模型
這篇文章主要介紹了iOS應用UI開發(fā)中的九宮格坐標計算與字典轉換模型,代碼基于傳統(tǒng)的Objective-C,需要的朋友可以參考下2016-01-01淺談iOS開發(fā)如何適配暗黑模式(Dark Mode)
這篇文章主要介紹了淺談iOS開發(fā)如何適配暗黑模式(Dark Mode),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2019-09-09