Programming example sum of squares

This commit is contained in:
p1ng0ut 2023-02-20 19:33:57 +01:00
parent 41e5f46eef
commit 14f5c1624c
4 changed files with 36 additions and 8 deletions

View File

@ -0,0 +1,8 @@
package space.dezentrale.prgrnd
class SumOfSquares {
fun calculate(intArray: IntArray): Int {
return intArray.sumOf { it * it }
}
}

View File

@ -13,12 +13,12 @@ class SumOfTwo {
}
fun optimalSolution(nums: IntArray, target: Int): IntArray {
val numsToIndices = nums.mapIndexed { index: Int, num: Int -> num to index }.toMap()
println("numsToIndices = $numsToIndices")
val numsToIndices = mutableMapOf<Int, Int>()
for ((index, num) in nums.withIndex()) {
if (numsToIndices.containsValue(target - num)) {
return intArrayOf(numsToIndices[target - num]!!, index)
}
numsToIndices[target] = index
}
return intArrayOf(-1, -1)
}

View File

@ -0,0 +1,20 @@
package space.dezentrale.prgrnd
import kotlin.test.Test
import kotlin.test.assertEquals
class SumOfSquaresTest {
@Test
fun `it calculates sum of one`() {
val sumOfSquares = SumOfSquares()
assertEquals(4, sumOfSquares.calculate(intArrayOf(-2)))
}
@Test
fun `it calculates sum of two`() {
val sumOfSquares = SumOfSquares()
assertEquals(13, sumOfSquares.calculate(intArrayOf(-2, 3)))
}
}

View File

@ -10,7 +10,7 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.simpleSolution(intArrayOf(2, 7, 11, 15), 9)
assertContentEquals(twoSum, intArrayOf(0, 1))
assertContentEquals(intArrayOf(0, 1), twoSum)
}
@Test
@ -19,7 +19,7 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.simpleSolution(intArrayOf(3, 2, 4), 6)
assertContentEquals(twoSum, intArrayOf(1, 2))
assertContentEquals(intArrayOf(1, 2), twoSum)
}
@Test
@ -28,7 +28,7 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.simpleSolution(intArrayOf(3, 3), 6)
assertContentEquals(twoSum, intArrayOf(0, 1))
assertContentEquals(intArrayOf(0, 1), twoSum)
}
@Test
@ -37,7 +37,7 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.optimalSolution(intArrayOf(2, 7, 11, 15), 9)
assertContentEquals(twoSum, intArrayOf(0, 1))
assertContentEquals(intArrayOf(0, 1), twoSum)
}
@Test
@ -46,7 +46,7 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.optimalSolution(intArrayOf(3, 2, 4), 6)
assertContentEquals(twoSum, intArrayOf(1, 2))
assertContentEquals(intArrayOf(1, 2), twoSum)
}
@Test
@ -55,6 +55,6 @@ class SumOfTwoTest {
val twoSum = sumOfTwo.optimalSolution(intArrayOf(3, 3), 6)
assertContentEquals(twoSum, intArrayOf(0, 1))
assertContentEquals(intArrayOf(0, 1), twoSum)
}
}