Печать массивов из середины

10

Это вопрос кода-гольфа.

Для заданных целых чисел s и n задача состоит в том, чтобы вывести все массивы длины n, которые принимают значения от -s до s. Единственный поворот заключается в том, что вы должны вывести их в следующем порядке.

  • Массив всех нулей длины n.
  • Все массивы длины n с элементами от -1 до 1, за исключением любого массива, который вы выводили ранее.
  • Все массивы длины n с элементами от -2 до 2, за исключением любого массива, который вы выводили ранее.
  • И так до тех пор, пока вы не получите все массивы длины n с элементами от -s до s, исключая любой массив, который вы выводили ранее.

Вы должны вывести один массив на строку. Они могут быть разделены пробелом или запятой.

Вот некоторый несоответствующий код Python, который выводит массивы / списки / кортежи в правильном порядке.

import itertools

s =  3
n = 2

oldsofar = set()
newsofar = set()
for i in xrange(s):
    for k in itertools.product(range(-i,i+1), repeat = n):
        newsofar.add(k)
    print newsofar - oldsofar
    oldsofar = newsofar.copy()
    print "***"

Дополнительная слава (и возражение от меня) за ответы, которые не выполняют никакого установленного вычитания или эквивалентного.

Мартин Эндер
источник
1
Можем ли мы написать функцию, которая печатает результат?
LegionMammal978
@ LegionMammal978 Я бы предпочел полную программу. Если это будет сочтено серьезно противоречивым, я, конечно,
Есть ли требуемый порядок в каждом из ваших пунктов?
Мартин Эндер
@ MartinBüttner Нет, совсем нет.

Ответы:

6

Желе, 9 байт

NRṗµAṀ€Ụị

Никакое вычитание списка не использовалось в создании этого сообщения. Попробуйте онлайн!

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

NRṗµAṀ€Ụị  Main link. Arguments: s, n

N          Negate; yield -s.
 R         Range; yield [-s, ..., s].
  ṗ        Cartesian power; push all vectors of length n of those elements.
   µ       Begin a new, monadic link. Argument: L (list of vectors)
    A      Compute the absolute values of all vector components.
     Ṁ€    Get the maximum component of each vector.
       Ụ   Sort the indices of A according to the maximal absolute value of the
           corresponding vector's components.
        ị  Retrieve the vectors of A at those indices.
Деннис
источник
Теперь это просто глупо!
2
«При создании этого поста не пострадали никакие списки»
Деннис ван Гилс
6

MATL , 18 байт

_G2$:iZ^t!|X>4#SY)

Первый вход s, второйn

Это работает в текущей версии (15.0.0) языка.

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

объяснение

_      % take input s implicitly. Negate to obtain -s
G      % push input s again
2$:    % inclusive range from -s to s
i      % take input n
Z^     % Cartesian power. Gives 2D array, with each result on a row
t!     % duplicate and transpose
|      % absolute value
X>     % maximum of each column 
4#S    % sort and push the indices of the sorting
Y)     % apply as row indices into the 2D array. Display implicitly
Луис Мендо
источник
1
18 байт возмутительно :)
4

Haskell, 61 60 байт

n#s=[c|b<-[0..s],c<-mapM id$[-b..b]<$[1..n],any((b==).abs)c]

Пример использования: 2#2-> [[0,0],[-1,-1],[-1,0],[-1,1],[0,-1],[0,1],[1,-1],[1,0],[1,1],[-2,-2],[-2,-1],[-2,0],[-2,1],[-2,2],[-1,-2],[-1,2],[0,-2],[0,2],[1,-2],[1,2],[2,-2],[2,-1],[2,0],[2,1],[2,2]].

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

   b<-[0..s]                           -- loop b through 0 .. s
        c<-mapM id$[-b..b]<$[1..n]     -- loop c through all lists of length n
                                       -- made out of the numbers -b .. b
                                       -- ("[-b..b]<$[1..n]" is "replicate n [-b..b]";
                                       --  "mapM id" is "sequence")
[c|                 ,any((b==).abs)c]  -- keep c if it contains b or -b

Редактировать: @xnor указал, что mapM idэто sequence.

Ними
источник
mapM idкороче чем sequence.
xnor
@xnor: правда. Спасибо!
Ними
2

Mathematica, 83 байта

Print/@Select[Range[-#,b=#]~Tuples~a,Abs@#~MemberQ~b&]&/@Range[0,a=Input[];Input[]]

Чтобы использовать, вставьте в скрипт и введите nзатем sв отдельных строках. Печатает каждый массив в виде списка в кавычках, разделенных запятыми (например, {-1, 0, 1}). Это работает, беря каждый список длины nс числами между [-cur..cur], и и печатая те, которые включают -curили cur. Затем он повторяет это для всех curв [0..s]. (Этот пост содержит 19 символов!)

LegionMammal978
источник
1

JavaScript (SpiderMonkey 30+), 134 байта

(s,n)=>n?[for(a of f(s,n-1))for(i of Array(s*2+1).keys())[i-n,...a]].sort((a,b)=>g(a)-g(b),g=a=>Math.max(...a,-Math.min(...a))):[[]]

Использует подход декартовой власти и сортировки, о котором я думал отдельно, но я перекомпилировал SpiderMonkey в то время, поэтому я не смог ответить на этот вопрос до @Dennis.

Нил
источник