Завдання

Напишіть програму або функцію, вхідною інформацією якої є список / масив 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).'}, що виводить відсортовані елементи в кожній групі і призводить до того, що результати для всіх груп будуть упаковані в масив комірок.

Пітон, 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]]

Математика, 36 байт

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

Пояснення

Для кожного nз {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, створює вектор від 1до max(x), неявно формує список довжини max(x)і перевіряє, які індекси xвідповідають тим, що в новому списку. Недбало друкує вихід.

Старіша версія:

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

Трохи інший підхід до іншої відповіді R. Займає вектор 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}

Спробуй це

Розширено:

{  # 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 ｢$_｣

}

Повертає нульові індекси, щоб отримати оператор кросуX+ на основі 1 ( ) у поєднанні з op . (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-ψ}¯

Спробуйте в Інтернеті!

Рода , 51 байт

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

Це порт відповіді Python від Уріеля .

Інша версія (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(виклик, виконаний при оцінці правої частини).

Лист звичайний, 91 байт

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

Індексація на основі 1, повертає набори як списки.

Спробуйте в Інтернеті!

k , 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