配列内の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に等しいかどうかを見ます
コード#コード#
タイトル
まず配列の異なる同位体に対して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;
}