Aritmética de Matrizes

Parte da seção Lógica e Fluxo do Journey de R da Coddy. Lição 20 de 64.

A aritmética entre uma matriz e um número, ou entre duas matrizes do mesmo tamanho, funciona célula por célula:

a <- matrix(1:4, nrow = 2, byrow = TRUE)
b <- matrix(c(10, 20, 30, 40), nrow = 2, byrow = TRUE)
print(a * 2)
print(a + b)

Saída:

     [,1] [,2]
[1,]    2    4
[2,]    6    8
     [,1] [,2]
[1,]   11   22
[2,]   33   44

Uma comparação também funciona célula por célula e retorna uma matriz lógica. Use-a para contar células ou como um índice para alterar apenas as células correspondentes:

m <- matrix(c(4, 12, 7, 15), nrow = 2, byrow = TRUE)
print(sum(m > 10))
m[m > 10] <- 10
print(m)

Saída:

[1] 2
     [,1] [,2]
[1,]    4   10
[2,]    7   10

t(m) troca linhas e colunas. %*% é o produto de matrizes, em que cada célula do resultado combina uma linha da primeira matriz com uma coluna da segunda; portanto, ncol(a) deve ser igual a nrow(b):

a <- matrix(1:4, nrow = 2, byrow = TRUE)
print(t(a))
print(a %*% a)

Saída:

     [,1] [,2]
[1,]    1    3
[2,]    2    4
     [,1] [,2]
[1,]    7   10
[2,]   15   22

A aritmética célula a célula entre matrizes de tamanhos diferentes é interrompida com o erro non-conformable arrays, portanto compare dim() primeiro. O produto de matrizes tem sua própria regra de tamanho:

a <- matrix(1:4, nrow = 2)
b <- matrix(1:6, nrow = 2)
print(identical(dim(a), dim(b)))
print(dim(a %*% b))

Saída:

[1] FALSE
[1] 2 3

Desafio

Fácil

Complete cap_values(m, limit). Retorne duas linhas: Over: com o número de células acima de limit e, em seguida, Total: com a soma da matriz depois que todo valor acima de limit for substituído por limit.

O código fornecido lê uma matriz de uma linha: as linhas são separadas por ; e os valores de uma linha por ,, portanto 1,2;3,4 é uma matriz com as linhas 1 2 e 3 4. A linha seguinte contém limit. O código imprime cada linha retornada.

Experimente você mesmo

cap_values <- function(m, limit) {
  # Escreva seu código aqui
  c("Over: 0", "Total: 0")
}

# Código de entrada/saída fornecido: mantenha-o como está
input <- suppressWarnings(readLines(file("stdin")))
parse_matrix <- function(line) {
  rows <- strsplit(line, ";")[[1]]
  matrix(as.numeric(unlist(strsplit(rows, ","))), nrow = length(rows), byrow = TRUE)
}
cat(cap_values(parse_matrix(input[1]), as.numeric(input[2])), sep = "\n")