Atualizando mapas

Parte da seção Lógica e Fluxo do Journey de Kotlin da Coddy. Lição 3 de 63.

mutableMapOf permite adicionar e substituir entradas com atribuição usando colchetes. remove(key) exclui uma entrada, se ela estiver presente. Um val pode fazer referência a um mapa mutável: a referência permanece fixa enquanto seu conteúdo muda.

Dentro de main:

val counts = mutableMapOf("a" to 2)
counts["a"] = 5
counts["b"] = 3
println(counts.size)

Atualizar a altera seu valor; adicionar b introduz a segunda chave.

A atribuição com colchetes insere uma chave ausente ou substitui o valor de uma chave existente.

Desafio

Médio

Complete updatedStock com os parâmetros stock: Map<String, Int>, key: String, amount: Int. Copie stock para um mapa mutável, defina a chave solicitada como amount e retorne o novo número de chaves.

A menos que limites mais rigorosos sejam especificados acima, as coleções contêm no máximo 100 elementos em cada nível, e os argumentos inteiros e os valores inteiros das coleções estão entre -1000 e 1000. O texto fornecido para análise pode incluir representações inválidas ou fora do intervalo, conforme descrito.

Retorne um valor do tipo Int. Mantenha o código de entrada/saída fornecido. Ele lê argumentos escalares, um por linha; uma lista começa com sua quantidade, seguida de seus elementos, e um mapa começa com a quantidade de suas entradas, seguida de cada chave e valor. Listas aninhadas repetem o formato de quantidade e elementos para cada linha. O main fornecido imprime o valor retornado seguido por uma nova linha. As listas usam o formato padrão entre colchetes do Kotlin; não imprima rótulos adicionais.

Experimente você mesmo

fun updatedStock(stock: Map<String, Int>, key: String, amount: Int): Int {
    // Escreva sua solução aqui.
    return 0
}
fun main() {
    val stockCount = readln().toInt()
    val stock = mutableMapOf<String, Int>()
    for (i0 in 0 until stockCount) {
        val stockKey = readln()
        val stockValue = readln().toInt()
        stock[stockKey] = stockValue
    }
    val key = readLine().orEmpty()
    val amount = readln().toInt()
    println(updatedStock(stock, key, amount))
}