add two sum problem (#4364)
* add two sum problem * linter solved * linter solved * improve code * linter solved * improve code * mini linter solved * update code --------- Co-authored-by: Piotr Idzik <65706193+vil02@users.noreply.github.com>
This commit is contained in:
parent
94621fb63e
commit
34cf6dab28
33
src/main/java/com/thealgorithms/misc/TwoSumProblem.java
Normal file
33
src/main/java/com/thealgorithms/misc/TwoSumProblem.java
Normal file
@ -0,0 +1,33 @@
|
||||
package com.thealgorithms.misc;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Optional;
|
||||
import org.apache.commons.lang3.tuple.Pair;
|
||||
|
||||
public final class TwoSumProblem {
|
||||
private TwoSumProblem() {
|
||||
}
|
||||
|
||||
/**
|
||||
* The function "twoSum" takes an array of integers and a target integer as input, and returns an
|
||||
* array of two indices where the corresponding elements in the input array add up to the target.
|
||||
* @param values An array of integers.
|
||||
* @param target The target is the sum that we are trying to find using two numbers from the given array.
|
||||
* @return A pair or indexes such that sum of values at these indexes equals to the target
|
||||
* @author Bama Charan Chhandogi (https://github.com/BamaCharanChhandogi)
|
||||
*/
|
||||
|
||||
public static Optional<Pair<Integer, Integer>> twoSum(final int[] values, final int target) {
|
||||
HashMap<Integer, Integer> valueToIndex = new HashMap<>();
|
||||
for (int i = 0; i < values.length; i++) {
|
||||
final var rem = target - values[i];
|
||||
if (valueToIndex.containsKey(rem)) {
|
||||
return Optional.of(Pair.of(valueToIndex.get(rem), i));
|
||||
}
|
||||
if (!valueToIndex.containsKey(values[i])) {
|
||||
valueToIndex.put(values[i], i);
|
||||
}
|
||||
}
|
||||
return Optional.empty();
|
||||
}
|
||||
}
|
61
src/test/java/com/thealgorithms/misc/TwoSumProblemTest.java
Normal file
61
src/test/java/com/thealgorithms/misc/TwoSumProblemTest.java
Normal file
@ -0,0 +1,61 @@
|
||||
package com.thealgorithms.misc;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
import static org.junit.jupiter.api.Assertions.assertFalse;
|
||||
|
||||
import org.apache.commons.lang3.tuple.Pair;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
/**
|
||||
* Test case for Two sum Problem.
|
||||
* @author Bama Charan Chhandogi (https://github.com/BamaCharanChhandogi)
|
||||
*/
|
||||
|
||||
public class TwoSumProblemTest {
|
||||
|
||||
@Test
|
||||
void testTwoSumExists() {
|
||||
final int[] values = new int[] {2, 7, 11, 15};
|
||||
final int target = 9;
|
||||
final var expected = Pair.of(0, 1); // values[0] + values[1] = 2 + 7 = 9
|
||||
assertEquals(expected, TwoSumProblem.twoSum(values, target).get());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTwoSumNoSolution() {
|
||||
final int[] values = new int[] {2, 7, 11, 15};
|
||||
final int target = 3;
|
||||
assertFalse(TwoSumProblem.twoSum(values, target).isPresent());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTwoSumMultipleSolutions() {
|
||||
final int[] values = {3, 3};
|
||||
final int target = 6;
|
||||
final var expected = Pair.of(0, 1); // values[0] + values[1] = 3 + 3 = 6
|
||||
assertEquals(expected, TwoSumProblem.twoSum(values, target).get());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTwoSumMultipleSolution() {
|
||||
final int[] values = {3, 4, 3, 3};
|
||||
final int target = 6;
|
||||
final var expected = Pair.of(0, 2); // values[0] + values[2] = 3 + 3 = 6
|
||||
assertEquals(expected, TwoSumProblem.twoSum(values, target).get());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTwoSumNegativeNumbers() {
|
||||
final int[] values = {-1, -2, -3, -4, -5};
|
||||
final int target = -8;
|
||||
final var expected = Pair.of(2, 4); // values[2] + values[4] = -3 + (-5) = -8
|
||||
assertEquals(expected, TwoSumProblem.twoSum(values, target).get());
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTwoSumNoSolutionDuplicatedInputs() {
|
||||
final int[] values = {0, 0, 0};
|
||||
final int target = 100;
|
||||
assertFalse(TwoSumProblem.twoSum(values, target).isPresent());
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user