两数之和、三数之和、四数之和java实现

两数之和

https://leetcode-cn.com/problems/two-sum

题目描述

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

 

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路:for循环遍历或者DFS遍历都可以,最快速方法是HashMap

import java.util.*;
class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] res=new int[2];
        if(nums==null||nums.length==0){
            return res;
        }
        Map map=new HashMap();
        for(int i=0;i

三数之和

https://leetcode-cn.com/problems/3sum

题目描述

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

 

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思路:找所有有效解的问题,第一反应是DFS,但本题DFS会超时

//DFS方法
import java.util.*;
class Solution {
    List> res=new ArrayList>();
    public List> threeSum(int[] nums) {
        if(nums==null||nums.length==0){
            return res;
        }
        Arrays.sort(nums);
        List list=new ArrayList();
        dfs(nums,list,res,0,0,0);
        return res;
    }
    public void dfs(int[] nums,List list,List> res,int start,int index,int sum){
        if(index==3&&sum==0&&!res.contains(list)){   
            res.add(new ArrayList(list));
            return;
        }
        if(index==3){
            return;
        }
        for(int i=start;i

因为超时故采用双指针法

import java.util.*;
class Solution {
    public List> threeSum(int[] nums) {
        List> res=new ArrayList>();
        if(nums==null||nums.length==0){
            return res;
        }
        Arrays.sort(nums);//数组排序
        for(int i=0;i0){//若第一个数就大于0则后续不可能和为0,之后的就不用做了break
                break;
            }
            if(i-1>=0&&nums[i]==nums[i-1]){
                continue;//当前数和之前的一样则跳过
            }
            int left=i+1;//左指针
            int right=nums.length-1;//右指针
            while(left-nums[i]){//当前和偏大,则右指针左移
                    right--;
                }
                else{
                    List list=new ArrayList();
                    list.add(nums[i]);
                    list.add(nums[left]);
                    list.add(nums[right]);
                    res.add(list);//等于0时加入res中
                    while(leftleft&&nums[right-1]==nums[right]){
                        right--;//去除重复的右指针数
                    }
                    left++;
                    right--;
                }
            }
        }
        return res;
    }
}

 四数之和

https://leetcode-cn.com/problems/4sum

题目描述

给定一个包含 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]
]

思路:采用左右指针的思路

import java.util.*;
class Solution {
    public List> fourSum(int[] nums, int target) {
        List> res=new ArrayList>();
        if(nums==null||nums.length==0){
            return res;
        }
        Arrays.sort(nums);//数组排序
        for(int i=0;i0&&nums[i]>target){//当第一个数大于target时后续的就不用做了break
                break;
            }
            for(int j=i+1;j0&&nums[i]+nums[j]>target){
                    break;//当第一第二个数和大于targer时后续不用做
                }
                int left=j+1;
                int right=nums.length-1;
                while(lefttarget){
                        right--;
                    }
                    else if(nums[left]+nums[right]+nums[i]+nums[j] list=new ArrayList();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[left]);
                        list.add(nums[right]);
                        if(!res.contains(list)){//若res中没有重复的则加入res
                            res.add(list);
                        }
                        while(leftleft&&nums[right-1]==nums[right]){
                            right--;
                        }
                        left++;
                        right--;
                    }
                }
            }
        }
        return res;
    }
}

 

你可能感兴趣的:(DFS,LeetCode)