计算列表中的对

Question

我最近开始使用 HackerRank，我正在尝试“按匹配销售”。在利用 Kotlin 的函数编程功能方面，我找到了一个令我满意的解决方案。但是，我没有得到预期的答案...

问题总结：

Given an Array: -> find and return the total number of pairs.

i.e:

input -> [10, 20, 20, 10, 10, 30, 50, 10, 20]

number of pairs -> 3

这是我的代码和一些注释来解释它：

fun sockMerchant(n: Int, pile: Array<Int>): Int{
    var count = 0
    mutableMapOf<Int, Int>().withDefault { 0 }.apply {
        // the [attempted] logic behind this piece of code here is 
        // that as we iterate through the list, the 'getOrPut()'
        // function will return either the value for the given [key]             
        // or throw an exception if there is no such key
        // in the map. However, since our map was created by 
        // [withDefault], the function resorts to its `defaultValue` <-> 0
        // instead of throwing an exception.
        for (e in values) {
            // this simplifies our code and inserts a zero [+1] where needed.
            // if (key exists)
            //      // return its associated value and MOD it:
            //      case: even  -> increment counter
            //            else  -> do nothing
            // else if (key dne)
            //      // insert default value <-> [0] + 1
            //              ....
            //              ....
            //              ....
            if (getOrPut(e, { getValue(e) + 1 } ) % 2 == 0) count++
        }
    }
    return count
}


fun main(args: Array<String>) {
    val scan = Scanner(System.`in`)
    val n = scan.nextLine().trim().toInt()
    val ar = scan.nextLine().split(" ").map{ it.trim().toInt() }.toTypedArray()
    val result = sockMerchant(n, ar)
    println(result)
}

-- 任何帮助或提示在这里都会有很长的路要走:)

Answer 1

我对其进行了一些修改以使其更易于测试，但这里是修复程序：

import java.util.*

fun sockMerchant(n: Int, pile: Array<Int>): Int{
    var count = 0
    mutableMapOf<Int, Int>().withDefault { 0 }.apply {
        // the [attempted] logic behind this piece of code here is
        // that as we iterate through the list, the 'getOrPut()'
        // function will return either the value for the given [key]
        // or throw an exception if there is no such key
        // in the map. However, since our map was created by
        // [withDefault], the function resorts to its `defaultValue` <-> 0
        // instead of throwing an exception.
        for (e in pile) {
            // this simplifies our code and inserts a zero [+1] where needed.
            // if (key exists)
            //      // return its associated value and MOD it:
            //      case: even  -> increment counter
            //            else  -> do nothing
            // else if (key dne)
            //      // insert default value <-> [0] + 1
            //              ....
            //              ....
            //              ....
            println(e)
            put(e, getValue(e) + 1)
            if (getValue(e) % 2 == 0) count++
            println(entries)
        }
    }
    return count
}


val n = 5
val ar = "10 10 10 10 20 20 30 40".split(" ").map{ it.trim().toInt() }.toTypedArray()
val result = sockMerchant(n, ar)
println(result)

输出：

10
[10=1]
10
[10=2]
10
[10=3]
10
[10=4]
20
[10=4, 20=1]
20
[10=4, 20=2]
30
[10=4, 20=2, 30=1]
40
[10=4, 20=2, 30=1, 40=1]
3
Pair.kts:3:18: warning: parameter 'n' is never used
fun sockMerchant(n: Int, pile: Array<Int>): Int{
                 ^

Process finished with exit code 0

解释：

你遍历了“值”，它在开始时是空的，所以你没有对你的代码做任何事情
即使循环遍历 pile，您的递增逻辑也不会超过 1，因此条件永远不会满足，计数也永远不会递增。

但背后的主要推理是正确的。

Answer 2

我可以通过将数字组合在一起，获取结果列表，然后对每个包含的对数求和来做到这一点：

fun sockMerchant(n: Int, pile: Array<Int>): Int =
    pile.groupBy { it }.values.sumBy { it.size / 2 }

在我们做了pile.groupBy { it }之后，我们有这样的结构：

{10=[10, 10, 10, 10], 20=[20, 20, 20], 30=[30], 50=[50]}

我们取这些值，然后将每个值的大小除以 2 求和。这会将 half-pairs 舍入为 0，将整对舍入为 1。

注意：我不完全清楚n在这种情况下的目的是什么。

计算列表中的对

counting pairs in a list

algorithm

kotlin

mutablemap