兩數之和及三數之和對比

Leetcode1-兩數之和

• 給定一個整數陣列 nums 和一個整數目標值 target，請你在該陣列中找出 和為目標值 target 的那 兩個 整數，並返回它們的陣列下標。
• 你可以假設每種輸入只會對應一個答案。但是，陣列中同一個元素在答案裡不能重複出現。
• 你可以按任意順序返回答案。
• 輸入：nums = [2,7,11,15], target = 9
• 輸出：[0,1]

    public int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> map=new HashMap<>();
        int[] res=new int[2];
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(target-nums[i])){
                res[0]=map.get(target-nums[i]);//get得到value所以這裡把下標作為value
                res[1]=i;
                break;
            }else{
                map.put(nums[i],i);
            }
        }
        return res;
    }
 

Leetcode15-三數之和

• 給你一個包含 n 個整數的陣列 nums，判斷 nums 中是否存在三個元素 a，b，c ，使得 a + b + c = 0 ？請你找出所有和為 0 且不重複的三元組。

• 注意：答案中不可以包含重複的三元組。

• 輸入：nums = [-1,0,1,2,-1,-4]

• 輸出：[[-1,-1,2],[-1,0,1]]

• 思路：兩數之和直接採用hashmap即可，這邊最好使用排序+雙指標方式

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
            List<List<Integer>> integetLists=new ArrayList<>();
            Set<List<Integer>> set=new HashSet<>();
            Arrays.sort(nums);
            if(nums.length==0 || nums.length==1 || nums.length==2){
                return integetLists;
            }

            if(nums[0]>0 || nums[nums.length-1]<0){
                return integetLists;
            }

            int i=0;
            int left=i+1;
            int right=nums.length-1;

            while(i<=nums.length-3){
                while(left!=right){
                    if(nums[i]+nums[left]+nums[right]<0){
                        left++;
                    }
                    else if(nums[i]+nums[left]+nums[right]>0){
                        right--;
                    }else{
                        List integerNums=new ArrayList<Integer>();
                        integerNums.add(nums[i]);
                        integerNums.add(nums[left]);
                        integerNums.add(nums[right]);
                        set.add(integerNums);
                        left++;
                    }
                }
                i++;
                left=i+1;
                right=nums.length-1;
            }

            for(List list:set){
                integetLists.add(list);
            }


            return integetLists;
    }
}