题解 | #牛牛的三元组问题#

牛牛的三元组问题

https://www.nowcoder.com/practice/72c6d735fb1144a2ba162976a4510839

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型二维数组
     */
    public int[][] findTriplets (int[] nums) {
        // write code here
        Arrays.sort(nums);
        List<Integer> path=new ArrayList<>();
        List<int[]> list=new ArrayList<>();
        boolean[] used=new boolean[nums.length];
        backTrack(0, used, nums, path, list);
        return list.toArray(new int[list.size()][3]);
    }

    private void backTrack(int start, boolean[] used, int[] nums,
     List<Integer> path, List<int[]> list){
        if(path.size()==3){
            int sum=path.get(0)+path.get(1)+path.get(2);
            if(sum==0){
                int[] tmp=new int[3];
                for (int i = 0; i < 3; i++)
                    tmp[i] = path.get(i);
                list.add(tmp);
            }
            return;
        }
        for(int i=start; i<nums.length; i++){
            if(used[i] || i>0 && nums[i-1]==nums[i] &&!used[i-1])continue;
            
            used[i]=true;
            path.add(nums[i]);
            backTrack(i, used, nums, path, list);
            used[i]=false;
            path.remove(path.size()-1);

        }
    }
}

这道题目和力扣上的数组组合问题十分相似,只是变了一个说法,这题要求按字典序返回,所以我们要联想到排序,这样每次取到的数字都是按照顺序的,另外为了保证结果不重复,也要把下一次开始的位置记录下来传到下一个递归过程,并且使用used数组标记使用过的元素。

#回溯法#
全部评论

相关推荐

2024-12-30 22:31
吉首大学 Web前端
工字钢写代码:改成吉林就OK了
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客企业服务