Задание

Напишите программу или функцию, чьи входные данные представляют собой список / массив X целых чисел, а выходные данные представляют собой список наборов целых чисел Y , так что для каждого элемента e в каждом наборе Y [ i ], X [ e ] = i и таким образом, что общее число элементов в множествах в Y равно числу элементов в X .

(Это в основном та же операция, что и обращение к хеш-таблице / словарю, за исключением того, что вместо этого применяется к массивам.)

Примеры

В этих примерах предполагается индексирование на основе 1, но вместо этого вы можете использовать индексацию на основе 0.

X             Y
[4]           [{},{},{},{1}]
[1,2,3]       [{1},{2},{3}]
[2,2,2]       [{},{1,2,3}]
[5,5,6,6]     [{},{},{},{},{1,2},{3,4}]
[6,6,5,5]     [{},{},{},{},{3,4},{1,2}]

Разъяснения

• Вы можете представлять набор в виде списка, если хотите. Если вы это сделаете, порядок элементов не имеет значения, но вы не можете повторять элементы.
• Вы можете использовать любой разумный однозначный формат ввода / вывода; Например, вы можете разделить элементы набора пробелами, а сами наборы - символом новой строки.
• Y должен быть конечной длины и, по крайней мере, достаточно длинным, чтобы все элементы X были индексами массива. Однако он может быть длиннее максимального элемента X (дополнительные элементы будут пустыми наборами).
• Все элементы X будут действительными индексами массива, то есть неотрицательными целыми числами, если вы используете индексацию на основе 0, или положительными целыми числами, если вы используете индексацию на основе 1.

Состояние победы

Как вызов коду-гольфу , чем короче, тем лучше.

MATL , 8 байт

tn:IXQ&D

Ввод - это вектор-столбец с ;разделителем (например [2;2;2]). Выход - это строковое представление массива ячеек векторов строк (например {[]; [1 2 3]}). Вектор строки одного элемента совпадает с числом (поэтому {1; 2; 3}выводится вместо {[1]; [2]; [3]}).

Попробуйте онлайн! Или проверьте все тестовые случаи .

объяснение

t     % Implicit input, say x. Duplicate
n     % Number of elements, say N
:     % Range: [1 2 ... N]
IXQ   % accumarray(x, [1 2 ... N], [], @(x){sort(x).'})
&D    % String representation

Большая часть работы выполняется функцией высшего порядка Matlab accumarray, которая группирует элементы во втором входе в соответствии с соответствующими значениями в первом и применяет указанную функцию к каждой группе. В этом случае используется функция @(x){sort(x).'}, которая выводит отсортированные элементы в каждой группе и обеспечивает упаковку результатов для всех групп в массив ячеек.

Python, 69 байт

lambda s:[[j for j,x in enumerate(s)if x==i]for i in range(max(s)+1)]

Использует индексирование на основе 0.

Желе , 7 5 байт

=þṀT€

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

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

=þṀT€  Main link. Argument: A (array)

  Ṁ    Yield m, the maximum of A.
=þ     Equals table; for each t in [1, ..., m], compare all elemnts of A with t,
       yielding a 2D Boolean array.
   T€  Truth each; for each Boolean array, yield all indices of 1.

Желе , 8 байт

Jẋ"Ṭ€;"/

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

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

Jẋ"Ṭ€;"/  argument: z           eg. [6,6,4,4]
J         [1 .. len(z)]             [1,2,3,4]
   Ṭ€     untruth each of z         [[0,0,0,0,0,1],
                                     [0,0,0,0,0,1],
                                     [0,0,0,1],
                                     [0,0,0,1]]
 ẋ"       repeat each of ^^         [[[],[],[],[],[],[1]],
          as many times as           [[],[],[],[],[],[2]],
          each of ^                  [[],[],[],[3]],
                                     [[],[],[],[4]]]
       /  reduce by...
     ;"   vectorized concatenation  [[],[],[],[3,4],[],[1,2]]

Mathematica, 36 байт

Join@@@#~Position~n~Table~{n,Max@#}&

объяснение

Для каждого nin {1, 2, ..., Max@#}, где Max@#наибольшее целое число в списке ввода, вычисляется Positions, где он nпоявляется в списке ввода #. Так как Position[{6,6,5,5},5](например) возвращает {{3},{4}}, мы затем Apply Joinко всем элементам на уровне {1}результата.

Haskell , 45 байт

sберет список целых чисел и возвращает список списков. 1-индексированный, чтобы сохранить входные данные тестового примера неизмененными (хотя выходные данные получают несколько дополнительных пустых списков).

s l=[[i|(i,y)<-zip[1..]l,y==x]|x<-[1..sum l]]

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

Это довольно простые вложенные списки. Единственным небольшим изменением является использование возможности сделать более длинный список с помощью sumвместо maximum.

PHP, 55 байт

<?while($i<=max($_GET))print_r(array_keys($_GET,$i++));

0 индексированные.

R 68 49 47 байт

lapply(1:max(x<-scan()),function(y)which(y==x)) 

Удивительно, но гораздо проще, чем более длинные решения. Извлекает вектор xиз STDIN, создает вектор из 1to max(x), неявно генерирует список длины max(x)и проверяет, какие индексы xсоответствуют указанным в новом списке. Неявно печатает вывод.

Старая версия:

o=vector('list',max(x<-scan()));for(i in x)o[[i]]=c(o[[i]],F<-F+1);o

Немного другой подход к другому ответу. Принимает вектор к STDIN, создает список с длиной, равной максимальному значению на входе. Перебирает ввод и добавляет индекс в нужное место.

Использует индексирование на основе 1.

Python 2 , 91 86 85 байт

Я программирую на своем телефоне, но мне очень понравился этот вызов. Я могу определенно играть в гольф дальше.

def f(a):
 r=[[]for i in range(max(a)+1)]
 for i,j in enumerate(a):r[j]+=[i]
 print r

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

Желе , 9 байт

Ṭ+\ịĠȧ@"Ṭ

1-индексированные, пустые наборы, представленные в виде 0наборов одного элемента, представленных в виде Nнаборов из нескольких элементов, представленных в виде[M,N,...]

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

Как?

Ṭ+\ịĠȧ@"Ṭ - Main link: list a        e.g. [6,6,4,4]
Ṭ         - untruth a                     [0,0,0,1,0,1]
  \       - cumulative reduce with:
 +        -   addition                    [0,0,0,1,1,2]
    Ġ     - group indices of a by value   [[3,4],[1,2]]
   ị      - index into                    [[1,2],[1,2],[1,2],[3,4],[3,4],[1,2]]
        Ṭ - untruth a                     [0,0,0,1,0,1]
       "  - zip with:
     ȧ@   -   and with reversed @rguments [0,0,0,[3,4],0,[1,2]]

JavaScript (ES6), 64 62 байта

Сохранено 2 байта благодаря @SteveBennett

Принимает 0-индексированный ввод. Возвращает разделенный запятыми список множеств.

a=>a.map((n,i)=>o[n]=[i,...o[n]||[]],o=[])&&`{${o.join`},{`}}`

Контрольные примеры

let f =

a=>a.map((n,i)=>o[n]=[i,...o[n]||[]],o=[])&&`{${o.join`},{`}}`

console.log(f([3]))
console.log(f([0,1,2]))
console.log(f([1,1,1]))
console.log(f([4,4,5,5]))
console.log(f([5,5,4,4]))

Альтернативная версия, 53 байта

Если упрощенный вывод, такой как '||||3,2|1,0'приемлемый, мы можем просто сделать:

a=>a.map((n,i)=>o[n]=[i,...o[n]||[]],o=[])&&o.join`|`

Баш , 109 байт

Жаль, что нет встроенного для максимального значения массива.

a=($@)
for((x=m=1;x<=m;x++)){ for((y=0;y<$#;)){((m<a[y]))&&((m=a[y]));((a[y++]==x))&&printf "%d " $y;};echo;}

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

Mathematica 62 байта

(Y={}~Table~Max@#;Y[[#[[j]]]]~AppendTo~j~Table~{j,Tr[1^#]};Y)&

Я буду управлять этим для вас

(Y={}~Table~Max@#;Y[[#[[j]]]]~AppendTo~j~Table~{j,Tr[1^#]};Y)&[{4,5,2,3,3,8,6,3}]

{{}, {3}, {4, 5, 8}, {1}, {2}, {7}, {}, {6}}

Попробуйте онлайн (просто вставьте код с помощью Ctrl-V и нажмите Shift + Enter)
, не забудьте вставить список ввода в конце, как в примере выше

Perl 6 ,  36 32  29 байт

->\a{map {a.grep(*==$_):k},1..a.max}

Попытайся

{map {.grep(*==$^a):k},1.. .max}

Попытайся

{map {.grep($^a):k},1.. .max}

Попытайся

Expanded:

{  # bare block lambda with implicit parameter ｢$_｣

  map

    {  # bare block lambda with placeholder parameter ｢$a｣

      .grep(  # grep for the values in ｢$_｣
        $^a   # that are equal to the currently tested value (and declare param)
      ) :k    # return the key (index) rather than the value
    },

    1 .. .max # Range from 1 to the maximum value in ｢$_｣

}

Возвращает нулевые индексы, чтобы получить 1 основанный перекрестный оператор ( X) в сочетании с +оп . (33 байта)

{1 X+.grep($^a):k}

Для того, чтобы заставить его вернуть Сета сек только надстройкуset в там (всего 37 байт)

{set 1 X+.grep($^a):k}

R, 80 72 байта

1-индексируется, берет Xиз stdin. Возвращает список векторов индексов с NULLпустым набором.

X=scan();Y=vector('list',max(X));Y[X]=lapply(X,function(x)which(X==x));Y

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

старая версия:

X=scan();Y=vector('list',max(X));for(i in 1:length(X))Y[[X[i]]]=c(Y[[X[i]]],i);Y

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

05AB1E , 10 байтов

ZFā¹N-ψ}¯

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

Röda , 51 байт

f s{seq(1,max(s))|[[s()|enum|[_2+1]if[_1=i]]]for i}

Это порт ответа Python от Uriel .

Другая версия (88 байт):

f a{I=indexOf;seq(1,max(a))|{|i|b=a[:]c=[]x=1{c+=x+I(i,b)b-=i;x++}until[I(i,b)<0];[c]}_}

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

Оба с 1 индексом.

PowerShell, 81 байт

$args[0]|%{if($_-gt($c=$r.count)){$r+=@($t)*($_-$c)}$r[--$_]+=,++$i};$r|%{"{$_}"}

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

1-индексироваться.

GNU Make , 214 213 208 204 байта

X=$(MAKECMDGOALS)
M=0
P=$(eval N=$(word $1,$X))$(if $N,$(if $(shell dc -e$Nd$Mds.\>.p),$(eval M=$N),)$(eval A$N+=$1$(call $0,$(shell expr $1 + 1))),)
$(call P,1)$(foreach K,$(shell seq $M),$(info $(A$K)))

Ввод / вывод: ввод массива через аргументы, вывод в stdout, по одному на строку, разделенные пробелами.

$ make -f swap.mk 2 2 2

3 2 1
make: *** No rule to make target `2'.  Stop.

объяснение

X=$(MAKECMDGOALS)     # Input array
M=0                   # Max value encountered in X
P=$(eval
    N=$(word $1,$X))  # Get next word from X
  $(if $N,$(if $(shell dc -e$Nd$Mds.\>.p),
    $(eval M=$N),)    # Update M
    $(eval A$N+=$1    # Append index to a variable named after value
      $(call $0,      # Recurse (call returns empty string)
        $(shell expr $1 + 1))),)
$(call P,1)           # Initial call to P. 1 is the first index
$(foreach K,          # Print all values of A* variables
  $(shell seq $M),
  $(info $(A$K)))     # Uninitialized ones default to empty strings

Порядок индексов в наборах меняется на обратный, потому что Pрекурсивно вызывает себя перед обновлением A$2(вызов выполняется при оценке правой части).

Common Lisp, 91 байт

(lambda(x &aux(l(make-list(eval`(max,@x))))(i 0))(dolist(y x l)(push(incf i)(nth(1- y)l))))

Индексирование на основе 1 возвращает наборы в виде списков.

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

к , 13 байт

{(=x)@!1+|/x}

Это 0-индексированный.

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

{           } /function(x)
 (=x)         /make a map/dictionary of values to their indices
         |/x  /get maximum value in x
      !1+     /make a range from 0 to the value, inclusive
     @        /get map value at each of the values in the range
              /    0N is given where there is no result