Count Quadruplets - KOTLIN

// Author : Saurav Kalsoor
// Count Quadruplets - KOTLIN

import java.util.*

var sc = Scanner(System.`in`)

fun main() {
    val n = sc.nextInt()
    val m = sc.nextInt()
    val p = sc.nextInt()
    val q = sc.nextInt()
    val target = sc.nextInt()

    val a = ArrayList<Int>()
    val b = ArrayList<Int>()
    val c = ArrayList<Int>()
    val d = ArrayList<Int>()

    for (i in 0 until n) {
        a.add(sc.nextInt())
    }
    for (i in 0 until m) {
        b.add(sc.nextInt())
    }
    for (i in 0 until p) {
        c.add(sc.nextInt())
    }
    for (i in 0 until q) {
        d.add(sc.nextInt())
    }
    println(countQuadruplets(a, b, c, d, target))
}

fun countQuadruplets(a: ArrayList<Int>, b: ArrayList<Int>, c: ArrayList<Int>, d: ArrayList<Int>, target: Int): Int {
    val map = HashMap<Int, Int>()
    for (i in a) {
        for (j in b) {
            if (map.containsKey(i + j)) {
                map[i + j] = map[i + j]!! + 1
            } else map[i + j] = 1
        }
    }
    var count = 0
    for (i in c) {
        for (j in d) {
            if (map.containsKey(target - i - j)) {
                count += map[target - i - j]!!
            }
        }
    }
    return count
}