程序设计-寻找三数之和为零的三元组(Java)

程序设计-寻找三数之和为零的三元组(Java)分享一个大牛的人工智能教程。零基础!通俗易懂!风趣幽默!希望你也加入到人工智能的队伍中来!请点击http://www.captainbed.netpackagelive.every.day.Programming;importjava.util.ArrayList;importjava.util.Arrays;/***给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a、b、c,使得a+b+c=0。*找出所有满足条件且不重复的三元组。**@auth

大家好,又见面了,我是你们的朋友全栈君。

分享一个大牛的人工智能教程。零基础!通俗易懂!风趣幽默!希望你也加入到人工智能的队伍中来!请点击http://www.captainbed.net

package live.every.day.Programming;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * 给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a、b、c,使得a+b+c=0。
 * 找出所有满足条件且不重复的三元组。
 *
 * @author Created by LiveEveryDay
 */
public class ThreeNumbersSumEqualsZero {

    public static ArrayList<ArrayList<Integer>> solve(int[] nums) {

        // 三元组用ArrayList<Integer>来表示
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();

        if (nums == null || nums.length == 0) {
            return result;
        }

        // 排序
        Arrays.sort(nums);

        for (int i = 0; i < nums.length - 2; i++) {

            // 避免计算重复元素
            if (i != 0 && nums[i] == nums[i - 1]) {
                continue;
            }

            /**
             * 固定一个数,从后面的数中选出两个数。
             * 因为数组是有序的,所以可以用两个数组下标left和right,left指向当前元素的后一个位置,right指向数组的最后一个位置。
             * 三数相加的和等于0时,加入解集;
             * 三数相加的和小于0时,把left往右边移动;
             * 三数相加的和大于0时,把right往左边移动。
             */
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum == 0) {
                    ArrayList<Integer> r = new ArrayList<>();
                    r.add(nums[i]);
                    r.add(nums[left]);
                    r.add(nums[right]);
                    result.add(r);
                    left++;
                    right--;

                    // 避免计算重复元素
                    while (left < right && nums[left] == nums[left - 1]) {
                        left++;
                    }

                    // 避免计算重复元素
                    while (left < right && nums[right] == nums[right + 1]) {
                        right--;
                    }
                } else if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return result;
    }

    public static void main(String[] args) {
        int[] nums1 = {-1, 0, 1, 2, -1, -4};
        System.out.println(solve(nums1));

        int[] nums2 = {0};
        System.out.println(solve(nums2));

        int[] nums3 = {1, 2, 3, 4, 5, 6};
        System.out.println(solve(nums3));
    }

}

// Output:
/*
[[-1, -1, 2], [-1, 0, 1]]
[]
[]
*/

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/151780.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号