配列内の3つの数の和が0の組合せを見つける

14418 ワード

配列内の3つの数の和が0の組合せを見つける
  • 構想
  • コード

  • タイトル
  • は、無秩序反復可能な整数シーケンスを与え、シーケンスの任意の3つの数の和が0に等しい場合、この3つの数を出力する.例えば、シーケンスnums=[−1,0,1,2,−1,−4],出力[−1,0,1],[−1,−1,2]]
  • 構想
    まず配列の異なる同位体に対して2つの結合を行って、1ラウンドの循環を行って、さっき結合した2組の加算と異なる同位体の数の和が0に等しいかどうかを見ます
    コード#コード#
    public class ThreeNumbersSum {
         
    
        public static void main(String[] args) {
         
            int[] nums = new int[]{
         -1, 0, 1, 2, -3, -4};
            List<int[]> ts = ts(nums);
            ts.forEach(arr -> {
         
                System.out.println(Arrays.toString(arr));
            });
        }
    
        public static List<int[]> ts(int[] nums) {
         
    
            if (nums == null || nums.length < 3) {
         
                return new ArrayList<>();
            }
            int length = nums.length;
            List<int[]> resultList = new ArrayList<>();
            if (length == 3) {
         
                if (nums[0] + nums[1] + nums[2] != 0) {
         
                    return null;
                } else {
         
                    resultList.add(nums);
                    return resultList;
                }
            }
    
            //               ,     
            List<Temp> maps = new ArrayList<>();
            for (int i = 0; i < length - 1; i++) {
         
                for (int j = i + 1; j < length; j++) {
         
                    Temp temp = new Temp();
                    temp.index[0] = i;
                    temp.index[1] = j;
                    temp.sum = nums[i] + nums[j];
                    maps.add(temp);
                }
            }
            //                 ,     0
            for (int i = 0; i < length; i++) {
         
                for (Temp temp : maps) {
         
                    int[] index = temp.index;
                    if (index[0] >= i || index[1] >= i) {
         
                        continue;
                    }
                    if (temp.sum + nums[i] == 0) {
         
                        temp.index[2] = i;
                        resultList.add(new int[]{
         nums[index[0]], nums[index[1]], nums[i]});
                    }
                }
            }
    
            return resultList;
        }
    }
    class Temp {
         
    
        int[] index = new int[3];
    
        int sum;
    }