题解 | #牛群的标签和#

牛群的标签和

https://www.nowcoder.com/practice/42ae88bedeb74da99813f6150769d07e

知识点:双指针

要求得到四个元素和为target的四元组,且按升序排列,故原数组的顺序不会影响答案,我们可以先对原数组进行排序,我们可以先确定前两个数,然后对后两个数采用双指针的方式遍历剩余位置,先将左右指针放置在剩余位置两侧,再将左右指针向内靠拢,以遍历剩余位置,若当前和小于目标和,则左指针向右移动,若当前和大于目标和,则右指针向左移动,若相等,保存当前四元组,左右指针同时向内移动,以遍历剩余数组。

Java题解如下

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @param target int整型 
     * @return int整型二维数组
     */
    public int[][] fourSum (int[] nums, int target) {
        // write code here
        Arrays.sort(nums);
        List<int[]> list = new ArrayList<>();
        int n = nums.length;
        for(int i = 0; i < n; i++) {
            if(i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for(int j = i + 1; j < n; j++) {
                if(j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }
                int left = j + 1, right = n - 1;
                while(left < right) {
                    int sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if(sum < target) {
                        left++;
                    }else if(sum > target) {
                        right--;
                    }else {
                        list.add(new int[]{nums[i], nums[j], nums[left], nums[right]});
                        left++;
                        right--;
                    }
                }
            }
        }
        int[][] res = new int[list.size()][4];
        for(int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务