LeetCode-Array-四數之和(18)

題目描述:

給定一個包含 n 個整數的數組 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。

注意:

答案中不可以包含重複的四元組。

示例:

給定數組 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

滿足要求的四元組集合爲:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

方法:

三指針法:
  • 首先判斷nums的長度是否小於4,是則直接返回list

  • 將nums排序,此時時間複雜度爲O(nlogn)

  • 第一層循環固定k=0,第二層循環固定指針i=k+1,之後再初始化指針m =i+1,j=nums.length-1;

  • 判斷sum和target的大小

    • 如果(sum==target)將nums[k],nums[i],nums[m],nums[j]加入list,並 j–;m++;
    • else if(sum>target) 執行j–;
    • else m++;
  • 注意隨後要用hashset去重

  • 時間複雜度爲:O(nlogn)+O(n^3)=O(n^3)

詳細代碼:
class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> list = new ArrayList<>();
        if(nums.length<4) return list;
        Arrays.sort(nums);
        for(int k = 0;k<nums.length;k++){
            for(int i =k+1;i<nums.length;i++){
                int m =i+1,j=nums.length-1;
                while(m<j){
                    int sum = nums[k]+nums[i]+nums[m]+nums[j];
                    if(sum==target){
                        list.add(Arrays.asList(nums[k],nums[i],nums[m],nums[j]));
                        j--;m++;
                    }else if(sum>target) j--;
                    else m++;
                }
            }
        }
        HashSet set = new HashSet(list);
        list.clear();
        list.addAll(set);
        return list;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章