LeetCode15【DFS】：三数之和

超时：2个用力没过

package leetcode.editor.cn;

//给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复
//的三元组。 
//
// 注意：答案中不可以包含重复的三元组。 
//
// 
//
// 示例： 
//
// 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
//
//满足要求的三元组集合为：
//[
//  [-1, 0, 1],
//  [-1, -1, 2]
//]
// 
// Related Topics 数组 双指针


import java.util.*;

//Java：三数之和
public class P15三数之和{
    public static void main(String[] args) {
        Solution solution = new P15三数之和().new Solution();
        int[] nums = {-1, 0, 1, 2, -1};
        List<List<Integer>> ans = solution.threeSum(nums);
        for (int i = 0; i < ans.size(); ++i) {
            System.out.println(ans.get(i).toString());
        }
        // TO TEST
    }
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
        HashSet<List<Integer>> hashtable = new HashSet<>();
    public List<List<Integer>> threeSum(int[] nums) {
        if (nums == null || nums.length < 3) {
            return new ArrayList<>();
        }
        int len = nums.length;
        boolean[] flag = new boolean[len];
        Arrays.sort(nums);
        dfs(nums, 0, 0, flag, new ArrayList<>());
        List<List<Integer>> ans2 = new ArrayList<>();
        for (List<Integer> lists : hashtable) {
            ans2.add(lists);
        }
        return ans2;
    }
    public void dfs(int[] nums, int idx, int sum, boolean[] flag, List<Integer> list) {
        if (sum > 0) {
            return;
        }
        if (list.size() == 3) {
            if (sum == 0) {
                hashtable.add(new ArrayList<>(list));
            }
            return;
        }
        for (int i = idx; i < nums.length; ++i)  {
            if (flag[i]) {
                continue;
            }
            flag[i] = true;
            list.add(nums[i]);
            dfs(nums, i, sum + nums[i], flag, list);
            list.remove(list.size() - 1);
            flag[i] = false;
        }

    }
}
//leetcode submit region end(Prohibit modification and deletion)

}