Вычислить сумму Кронекера двух матриц

9

В приведенных ниже примерах, Aи Bбудет 2-на-2 матрицы, и матрицы являются одним индексированные.

Кронекера продукт имеет следующие свойства:

A⊗B =  A(1,1)*B   A(1,2)*B
        A(2,1)*B   A(2,2)*B

     =  A(1,1)*B(1,1)   A(1,1)*B(1,2)   A(1,2)*B(1,1)   A(1,2)*B(1,2)
        A(1,1)*B(2,1)   A(1,1)*B(2,2)   A(1,2)*B(2,1)   A(1,2)*B(2,2)
        A(2,1)*B(1,1)   A(2,1)*B(1,2)   A(2,2)*B(1,1)   A(2,2)*B(1,2)
        A(2,2)*B(2,1)   A(2,2)*B(1,2)   A(2,2)*B(2,1)   A(2,2)*B(2,2)

Кронекера сумма обладает следующими свойствами:

A⊕B = A⊗Ib + Ia⊗B

Iaи Ibявляются единичными матрицами с размерами Aи Bсоответственно. Aи Bявляются квадратными матрицами. Обратите внимание, что Aи Bмогут быть разных размеров.

A⊕B =  A(1,1)+B(1,1)  B(1,2)         A(1,2)         0
        B(2,1)         A(1,1)+B(2,2)  0              A(1,2)
        A(2,1)         0              A(2,2)+B(1,1)  B(1,2)
        0              A(2,1)         B(2,1)         A(2,2)+B(2,2)

Учитывая две квадратные матрицы, Aи Bвычислить сумму Кронекера двух матриц.

  • Размер матриц будет как минимум 2-by-2. Максимальный размер будет соответствовать тому, что ваш компьютер / язык может обрабатывать по умолчанию, но минимальный 5-by-5ввод (5 МБ).
  • Все входные значения будут неотрицательными целыми числами
  • Встроенные функции, которые вычисляют сумму Кронекера или произведения Кронекера, не допускаются
  • В целом: стандартные правила, касающиеся формата ввода / вывода, программы и функций, лазеек и т. Д.

Тестовые случаи:

A =
     1     2
     3     4
B =
     5    10
     7     9

A⊕B =
     6    10     2     0
     7    10     0     2
     3     0     9    10
     0     3     7    13

----

A =
    28    83    96
     5    70     4
    10    32    44
B =
    39    19    65
    77    49    71
    80    45    76

A⊕B =
    67    19    65    83     0     0    96     0     0
    77    77    71     0    83     0     0    96     0
    80    45   104     0     0    83     0     0    96
     5     0     0   109    19    65     4     0     0
     0     5     0    77   119    71     0     4     0
     0     0     5    80    45   146     0     0     4
    10     0     0    32     0     0    83    19    65
     0    10     0     0    32     0    77    93    71
     0     0    10     0     0    32    80    45   120

----

A =
    76    57    54
    76     8    78
    39     6    94
B =
    59    92
    55    29

A⊕B =
   135    92    57     0    54     0
    55   105     0    57     0    54
    76     0    67    92    78     0
     0    76    55    37     0    78
    39     0     6     0   153    92
     0    39     0     6    55   123
Стьюи Гриффин
источник

Ответы:

2

Желе , 26 21 20 19 байт

æ*9Bs2¤×€€/€S;"/€;/

Ввод - это список двух 2D-списков, вывод - это один 2D-список. Попробуйте онлайн! или проверьте все контрольные примеры .

Как это работает

æ*9Bs2¤×€€/€S;"/€;/  Main link.
                     Argument: [A, B] (matrices of dimensions n×n and m×m)

      ¤              Evaluate the four links to the left as a niladic chain.
  9B                 Convert 9 to base 2, yielding [1, 0, 0, 1].
    s2               Split into sublists of length 2, yielding [[1, 0], [0, 1]].
æ*                   Vectorized matrix power.
                     This yields [[A¹, B⁰], [A⁰, B¹]], where B⁰ and A⁰ are the
                     identity matrices of dimensions m×m and n×n.
          /€         Reduce each pair by the following:
        €€             For each entry of the first matrix:
       ×                 Multiply the second matrix by that entry.
            S        Sum the two results, element by element.
                     This yields the Kronecker sum, in form of a n×n matrix of
                     m×m matrices.
               /€    Reduce each row of the outer matrix...
             ;"        by zipwith-concatenation.
                     This concatenates the columns of the matrices in each row,
                     yielding a list of length n of n×nm matrices.
                 ;/  Concatenate the lists, yielding a single nm×nm matrix.
Деннис
источник
Так много евро ... ваша программа богата!
Луис Мендо
5

CJam, 40 39 38 байт

9Yb2/q~f{.{_,,_ff=?}:ffff*::.+:~}:..+p

Формат ввода представляет собой список, содержащий Aи Bкак 2D списки, например

[[[1 2] [3 4]] [[5 10] [7 9]]]

Выходной формат представляет собой один 2D-список в стиле CJam.

Тестирование. (С более читаемым форматом вывода.)

объяснение

Этот код является упражнением в составных (или инфиксных) операторах. Они обычно полезны для манипулирования массивами, но эта проблема усугубляет необходимость в них. Вот краткий обзор:

  • fожидает список и что-то еще в стеке и отображает следующий двоичный оператор над списком, передавая другой элемент в качестве второго аргумента. Например [1 2 3] 2 f*и то и 2 [1 2 3] f*другое дают [2 4 6]. Если оба элемента являются списками, первый сопоставляется, а второй используется для каррирования бинарного оператора.
  • :имеет два использования: если оператор, следующий за ним, унарный, это простая карта. Например , [1 0 -1 4 -3] :zэто [1 0 1 4 3], где zполучает модуль числа. Если оператор, следующий за ним, является двоичным, то вместо этого он сложит оператор. Например, [1 2 3 4] :+есть 10.
  • .векторизует бинарный оператор. Он ожидает два списка в качестве аргументов и применяет оператор к соответствующим парам. Например [1 2 3] [5 7 11] .*дает [5 14 33].

Обратите внимание, что :сам по себе всегда унарный оператор, тогда как fи .сами всегда бинарные операторы. Они могут быть произвольно вложены (при условии, что они имеют правильные арты). И это то, что мы будем делать ...

9Yb      e# Push the binary representation of 9, i.e. [1 0 0 1].
2/       e# Split into pairs, i.e. [[1 0] [0 1]]. We'll use these to indicate
         e# which of the two inputs we turn into an identity matrix.
q~       e# Read and evaluate input, [A B].
f{       e# This block is mapped over the [[1 0] [0 1]] list, also pushing
         e# [A B] onto the stack for each iteration.
  .{     e#   The stack has either [1 0] [A B] or [0 1] [A B]. We apply this
         e#   block to corresponding pairs, e.g. 1 A and 0 B.
    _,   e#     Duplicate the matrix and get its length/height N.
    ,_   e#     Turn into a range [0 1 ... N-1] and duplicate it.
    ff=  e#     Double f on two lists is an interesting idiom to compute an
         e#     outer product: the first f means that we map over the first list
         e#     with the second list as an additional parameter. That means for
         e#     the remaining operator the two arguments are a single integer
         e#     and a list. The second f then maps over the second list, passing
         e#     in the the number from the outer map as the first parameter.
         e#     That means the operator following ff is applied to every possible
         e#     pair of values in the two lists, neatly laid out in a 2D list.
         e#     The operator we're applying is an equality check, which is 1
         e#     only along the diagonal and 0 everywhere else. That is, we've
         e#     created an NxN identity matrix.
    ?    e#     Depending on whether the integer we've got along with the matrix
         e#     is 0 or 1, either pick the original matrix or the identity.
  }
         e#   At this point, the stack contains either [A Ib] or [Ia B]. 
         e#   Note that A, B, Ia and Ib are all 2D matrices.
         e#   We now want to compute the Kronecker product of this pair.
  :ffff* e#   The ffff* is the important step for the Kronecker product (but
         e#   not the whole story). It's an operator which takes two matrices
         e#   and replaces each cell of the first matrix with the second matrix
         e#   multiplied by that cell (so yeah, we'll end up with a 4D list of
         e#   matrices nested inside a matrix).
         e#   The leading : is a fold operation, but it's a bit of a degenerate
         e#   fold operation that is only used to apply the following binary operator
         e#   to the two elements of a list.
         e#   Now the ffff* works essentially the same as the ff= above, but
         e#   we have to deal with two more dimensions now. The first ff maps
         e#   over the cells of the first matrix, passing in the second matrix
         e#   as an additional argument. The second ff then maps over the second
         e#   matrix, passing in the cell from the outer map. We multiply them
         e#   with *.
         e#   Just to recap, we've essentially got the Kronecker product on the
         e#   stack now, but it's still a 4D list not a 2D list.
         e#   The four dimensions are:
         e#     1. Columns of the outer matrix.
         e#     2. Rows of the outer matrix.
         e#     3. Columns of the submatrices.
         e#     4. Rows of the submatrices.
         e#   We need to unravel that into a plain 2D matrix.
  ::.+   e#   This joins the rows of submatrices across columns of the outer matrix.
         e#   It might be easiest to read this from the right:
         e#     +    Takes two rows and concatenates them.
         e#     .+   Takes two matrices and concatenates corresponding rows.
         e#     :.+  Takes a list of matrices and folds .+ over them, thereby
         e#          concatenating the corresponding rows of all matrices.
         e#     ::.+ Maps this fold operation over the rows of the outer matrix.
         e#   We're almost done now, we just need to flatten the outer-most level
         e#   in order to get rid of the distinction of rows of the outer matrix.
  :~     e#   We do this by mapping ~ over those rows, which simply unwraps them.
}
         e# Phew: we've now got a list containing the two Kronecker products
         e# on the stack. The rest is easy, just perform pairwise addition.
:..+     e# Again, the : is a degenerate fold which is used to apply a binary
         e# operation to the two list elements. The ..+ then simply vectorises
         e# addition twice, such that we add corresponding cells of the 2D matrices.
p        e# All done, just pretty-print the matrix.
Мартин Эндер
источник
fffffffffff что на земле ... Я надеюсь, что выживет в гольф, чтобы вы в конечном итоге объяснить это: P
FryAmTheEggman
@FryAmTheEggman :ffff*может быть самым длинным (соединение) оператор я когда - либо использовал в CJam ... Для один более байт один может идти даже безумнее , хотя: 9Yb2/Q~f.{\{,,_ff=}&}::ffff*:::.+::~:..+p(и да, добавим объяснение , когда я закончу играть в гольф).
Мартин Эндер
4

J - 38 33 31 байт

i=:=@i.@#
[:,./^:2(*/i)+(*/~i)~

Применение

   f =: [:,./^:2(*/i)+(*/~i)~
   (2 2 $ 1 2 3 4) f (2 2 $ 5 10 7 9)
6 10 2  0
7 10 0  2
3  0 9 10
0  3 7 13
   (3 3 $ 28 83 96 5 70 4 10 32 44) f (3 3 $ 39 19 65 77 49 71 80 45 76)
67 19  65  83   0   0 96  0   0
77 77  71   0  83   0  0 96   0
80 45 104   0   0  83  0  0  96
 5  0   0 109  19  65  4  0   0
 0  5   0  77 119  71  0  4   0
 0  0   5  80  45 146  0  0   4
10  0   0  32   0   0 83 19  65
 0 10   0   0  32   0 77 93  71
 0  0  10   0   0  32 80 45 120
   (3 3 $ 76 57 54 76 8 78 39 6 94) f (2 2 $ 59 92 55 29)
135  92 57  0  54   0
 55 105  0 57   0  54
 76   0 67 92  78   0
  0  76 55 37   0  78
 39   0  6  0 153  92
  0  39  0  6  55 123
миль
источник
Использование матричного деления не удастся, если одна из матриц будет единственной. Например, (2 2 $ 1 2 3 4) f (2 2 $ 1 1 1 1)возникнет ошибка домена.
Деннис
@ Денис хороший улов, я проверял только случайные значения ? 4 4 $ 100. Я не уверен, есть ли способ использовать композицию диады x f&g y = (g x) f (g y)или что-то еще здесь.
миль
2

Юлия, 60 59 58 56 байт

A%B=hvcat(sum(A^0),sum(i->map(a->a*B^i,A'^-~-i),0:1)...)

Попробуйте онлайн!

Как это работает

  • Для матриц A и B , map(a->a*B,A')вычисляет произведение Кронекера A⊗B .

    Результат является вектором матричных блоков с размерами B .

    Мы должны транспонировать A'), так как матрицы хранятся в главном порядке столбцов.

  • Поскольку побитовое НЕ с дополнением до двух удовлетворяет тождеству ~ n = - (n + 1) для всех целых чисел n , имеем - ~ -n = - (~ (-n)) = - ((- n) + 1) = 1 - n , поэтому - ~ -0 = 1 и - ~ -1 = 0 .

    Таким образом, анонимная функция i->map(a->a*B^i,A'^-~-i)применяет приведенную выше карту к B⁰ (единичная матрица с размерами B ) и A¹ = A, когда i = 0 , и к и A⁰, когда i = 1 .

  • sum(i->map(a->a*B^i,A'^-~-i),0:1)Суммирует более {0,1} с помощью указанной выше анонимной функции, вычисляя сумму Кронекера A⊕B как A¹⊗B⁰ + A⁰⊗B¹ .

    Результат является вектором матричных блоков с размерами B .

  • sum(A^0)вычисляет сумму всех элементов матрицы идентичности А размеров «s. Для матрицы A n × n это дает n .

  • Наконец, hvcat(sum(A^0),sum(i->map(a->a*B^i,A'^-~-i),0:1)...)объединяет матричные блоки, которые образуют A⊕B .

    С первым аргументом п , hvcatсцепляет п матричных блоков по горизонтали, и в результате (больше) блоков по вертикали.

Деннис
источник
0

Руби, 102

->a,b{r=0..-1+a.size*q=b.size
r.map{|i|r.map{|j|(i/q==j/q ?b[i%q][j%q]:0)+(i%q==j%q ?a[i/q][j/q]:0)}}}

В тестовой программе

f=->a,b{r=0..-1+a.size*q=b.size
r.map{|i|r.map{|j|(i/q==j/q ?b[i%q][j%q]:0)+(i%q==j%q ?a[i/q][j/q]:0)}}}

aa =[[1,2],[3,4]]
bb =[[5,10],[7,9]]
f[aa,bb].each{|e|p e}
puts

aa =[[28,83,96],[5,70,4],[10,32,44]]
bb =[[39,19,65],[77,49,71],[80,45,76]]
f[aa,bb].each{|e|p e}
puts

aa =[[76,57,54],[76,8,78],[39,6,94]]
bb =[[59,92],[55,29]]
f[aa,bb].each{|e|p e}
puts

Требуется два 2D-массива в качестве входных данных и возвращает 2D-массив.

Вероятно, есть лучшие способы сделать это: использовать функцию, чтобы избежать повторения; используя одну петлю и распечатывая вывод. Посмотрим на них позже.

Уровень реки St
источник
0

JavaScript (ES6), 109

Построен на ответ на другой вызов

(a,b)=>a.map((a,k)=>b.map((b,i)=>a.map((y,l)=>b.map((x,j)=>r.push(y*(i==j)+x*(k==l))),t.push(r=[]))),t=[])&&t

Тестовое задание

f=(a,b)=>a.map((a,k)=>b.map((b,i)=>a.map((y,l)=>b.map((x,j)=>r.push(y*(i==j)+x*(k==l))),t.push(r=[]))),t=[])&&t

console.log=x=>O.textContent+=x+'\n'

function show(label, mat)
{
  console.log(label)
  console.log(mat.join`\n`)
}

;[ 
  {a:[[1,2],[3,4]], b:[[5,10],[7,9]]},
  {a:[[28,83,96],[5,70,4],[10,32,44]], b:[[39,19,65],[77,49,71],[80,45,76]]},
  {a:[[76,57,54],[76,8,78],[39,6,94]], b:[[59,92],[55,29]]}
].forEach(t=>{
  show('A',t.a)  
  show('B',t.b)
  show('A⊕B',f(t.a,t.b))
  show('B⊕A',f(t.b,t.a))  
  console.log('-----------------')
})
<pre id=O></pre>

edc65
источник