給定四個包含整數(shù)的數(shù)組列表 A , B , C , D ,計算有多少個元組 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。
首先將四個數(shù)組分割為兩兩數(shù)組,前兩個數(shù)組值相加,后兩個數(shù)組相加,入股前兩個數(shù)組相加和與后兩個數(shù)組相加和正好為相反數(shù),四個元素之和為0.
將兩數(shù)組的元素進(jìn)行遍歷相加,相加之和為map的索引。所指向的元素,就是出現(xiàn)的次數(shù)。
func foursumcount(A []int, B []int, C []int, D []int) int{ des :=map[int]int{} for _,v:=range A{ for _,w:=range B{ des[v+w]++ } } }
再次遍歷另兩個數(shù)組,將兩個數(shù)組的元素進(jìn)行相加,取和的相反數(shù),通過使用相反數(shù)在map中查找,如果沒出現(xiàn),所指向的數(shù)是0,如果出現(xiàn)過這個數(shù)的相反數(shù),則所指向的數(shù)大于一。
func foursumcount(A []int, B []int, C []int, D []int) int{ des :=map[int]int{} ans:=0 for _,v:=range C{ for _,w:=range D{ ans +=des[-v-w] } } }
全部代碼
func fourSumCount(A []int, B []int, C []int, D []int) int { des := map[int]int{} ans:=0 for _,v :=range A{//遍歷兩個數(shù)組,將兩個數(shù)組的和作為一個索引,進(jìn)行+1操作 for _,w:=range B{ des[v+w]++ } } for _,v :=range C{//遍歷另兩個數(shù)組,如果這兩個數(shù)組進(jìn)行相加的和的相反數(shù)在map中不為1,則證明出現(xiàn)過 for _,w:=range D{ ans +=des[-v-w] } } return ans//返回總數(shù) }
補(bǔ)充:算法題:三個數(shù)相加等于某個特定值
題目來自于leetcode第十五題
給定一個n個整數(shù)的數(shù)組S,是否存在S中的元素a,b,c,使得a + b + c = 0? 查找數(shù)組中所有唯一的三元組,它們的總和為零。
注意:解決方案集不能包含重復(fù)的三元組。
給定數(shù)組:
S = [-1, 0, 1, 2, -1, -4]
解決方案:
[[-1, 0, 1],[-1, -1, 2]]
在剛看到這道題目的題目的時候,首先想到的就是暴力解法,將數(shù)組排序后直接嵌套三個循環(huán),這樣子雖然簡單,但是時間復(fù)雜度確實(shí)n^3,遇到數(shù)據(jù)量過大的時候消耗太大,提交的時候并沒有通過。
自己在想了一段時間后想到了一些優(yōu)化方案,但是本質(zhì)上都沒有將次方縮減,所以仍然需要改進(jìn),目標(biāo)為n^2。
首先,目標(biāo)為n^2的話,就需要將數(shù)組掃描兩遍,第一層循環(huán)沒有問題,但要將第二層和第三層循環(huán)縮減為掃描一遍,因為是要將兩個數(shù)相加等于某個值,所以可將有序數(shù)組分別從前往后和從后往前掃描,直至碰頭,碰頭后如果繼續(xù)循環(huán)的話,所得到的結(jié)果會重復(fù),
所以到碰頭后可以跳出循環(huán)。這樣子只需要掃描數(shù)組一遍就可達(dá)到兩層循環(huán)的結(jié)果。思路簡單是這樣,在實(shí)現(xiàn)的時候要考慮一些其他的問題,具體實(shí)現(xiàn)的代碼如下:
public class Solution { public ListListInteger>> threeSum(int[] nums) { ListListInteger>> result = new LinkedListListInteger>>(); if(nums.length3){ return result; } Arrays.sort(nums); int left=0,right=nums.length-1; for(int mid=0;mid nums.length-2;mid++){ if(nums[mid]>0) break; if(mid == 0 || (mid > 0 nums[mid] != nums[mid-1])){ left=mid+1; right=nums.length-1; while(leftright){ if(nums[left]+nums[mid]+nums[right] ==0){ result.add(Arrays.asList(nums[mid],nums[left],nums[right])); while (left right nums[left] == nums[left+1]) left++; while (left right nums[right] == nums[right-1]) right--; left++; right--; }else if(nums[left]+nums[mid]+nums[right]0){ left++; }else if(nums[left]+nums[mid]+nums[right]>0){ right--; } } } } return result; } }
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。
標(biāo)簽:梅河口 重慶 欽州 蘭州 吐魯番 雞西 銅川 汕頭
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《go語言的四數(shù)相加等于指定數(shù)算法》,本文關(guān)鍵詞 語言,的,四數(shù),相加,等于,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。