Biorąc pod uwagę liczbę n, wypisz uporządkowaną listę indeksów 1 opartych na przekątnej n*nmacierzy kwadratowej.

Przykład:

Dla danych wejściowych 3:

Kwadrat powinien być:

1 2 3
4 5 6
7 8 9

Teraz możemy wybrać wszystkie indeksy reprezentowanych przez \, /lub X( #lub pozycje spoza przekątne są odrzucane)

\ # /
# X #
/ # \

Dane wyjściowe wynoszą:

[1,3,5,7,9]

Przypadki testowe:

1=>[1]
2=>[1,2,3,4]
3=>[1,3,5,7,9]
4=>[1,4,6,7,10,11,13,16]
5=>[1,5,7,9,13,17,19,21,25]

Nie będzie akceptowanej odpowiedzi. Chcę znać najkrótszy kod dla każdego języka.

Oktawa , 28 bajtów

@(n)find((I=eye(n))+flip(I))

Anonimowa funkcja, która wprowadza liczbę i wyprowadza wektor kolumnowy liczb.

Wypróbuj online!

JavaScript (ES6), 48 bajtów

Zwraca rozdzieloną myślnikiem listę liczb całkowitych w postaci łańcucha.

f=(n,k=n*n)=>--k?f(n,k)+(k%~-n&&k%-~n?'':~k):'1'

Sformatowane i skomentowane

f = (n, k = n * n) => // given n and starting with k = n²
  --k ?               // decrement k; if it does not equal zero:
    f(n, k) + (       //   return the result of a recursive call followed by:
      k % ~-n &&      //     if both k % (n - 1) and
      k % -~n ?       //             k % (n + 1) are non-zero:
        ''            //       an empty string
      :               //     else:
        ~k            //       -(k + 1) (instantly coerced to a string)
    )                 //   end of iteration
  :                   // else:
    '1'               //   return '1' and stop recursion

Przypadki testowe

f=(n,k=n*n)=>--k?f(n,k)+(k%~-n&&k%-~n?'':~k):'1'

;[1,2,3,4,5].forEach(n => console.log(n, '->', f(n)))

R , 38 35 34 38 bajtów

3 bajty zapisane, gdy przypomniałem sobie o istnieniu whichfunkcji ..., 1 bajt zapisany dzięki @Rift

d=diag(n<-scan());which(d|d[n:1,])

+4 bajty dla argumentu ec=Twywołanego przez program jako pełny programsource()

Wypróbuj online!

Wyjaśnienie:

n<-scan()            # take input
d=diag(n);           # create an identity matrix (ones on diagonal, zeros elsewhere)
d|d[n:1,]            # coerce d to logical and combine (OR) with a flipped version
which([d|d[n:1,]])   # Find indices for T values in the logical expression above

Galaretka , 8 bajtów

⁼þ`+Ṛ$ẎT

Wypróbuj online!

Wykorzystuje algorytm Luisa Mendo do odpowiedzi MATL.

Oktawa , 41 37 bajtów

Nawiasem mówiąc, działa to również w MATLAB. Brak podstępnej funkcji specyficznej dla Octave :)

@(x)unique([x:x-1:x^2-1;1:x+1:x*x+1])

Wypróbuj online!

Wyjaśnienie:

Zamiast tworzyć kwadratową macierz i znaleźć dwie przekątne, pomyślałem, że raczej zamiast tego obliczam przekątne bezpośrednio. To było 17 bajtów krótszych! =)

@(x)                                   % Anonymous function that takes 'x' as input
    unique(...                   ...)  % unique gives the unique elements, sorted
           [x:x-1:x^2-1                % The anti-diagonal (is that the correct word?)
                       ;               % New row
                        1:x+1:x*x+1])  % The regular diagonal

Tak to wygląda bez unique:

ans =    
    6   11   16   21   26   31
    1    8   15   22   29   36

_{Tak, prawdopodobnie powinienem zmienić kolejność przekątnych, aby uczynić go bardziej przyjaznym dla człowieka.}

MATL , 6 bajtów

XytP+f

Wypróbuj online!

Wyjaśnienie

Takie samo podejście jak moja odpowiedź Octave.

Rozważ dane wejściowe 3jako przykład.

Xy   % Implicit input. Identity matrix of that size
     % STACK: [1 0 0;
               0 1 0;
               0 0 1]
t    % Duplicate
     % STACK: [1 0 0
               0 1 0
               0 0 1],
              [1 0 0
               0 1 0
               0 0 1]
P    % Flip vertically
     % STACK: [1 0 0
               0 1 0
               0 0 1],
              [0 0 1
               0 1 0
               1 0 0]
+    % Add
     % STACK: [1 0 1
               0 2 0
               1 0 1]
f    % Linear indices of nonzero entries. Implicit display  
     % STACK:[1; 3; 5; 7; 9]

Indeksowanie liniowe jest główne , oparte na 1. Aby uzyskać więcej informacji, zobacz fragment długości 12 tutaj .

Python 2 , 54 53 bajtów

lambda n:[i+1for i in range(n*n)if i%-~n<1or i%~-n<1]

Wypróbuj online!

Oktawa, 68 54 bajtów

Dzięki @Stewie Griffin za uratowanie 14 bajtów!

@(x)unique([diag(m=reshape(1:x^2,x,x)),diag(flip(m))])

Wypróbuj online!

MATLAB, 68 bajtów

x=input('');m=reshape([1:x*x],x,x);unique([diag(m) diag(flipud(m))])

Wyjaśnienie:

@(x)                               % Anonymous function
m=reshape([1:x*x],x,x);            % Create a vector from 1 to x^2 and
                                   % reshape it into an x*x matrix.
diag(m)                            % Find the values on the diagonal.
diag(flip(m))                      % Flip the matrix upside down and
                                   % find the values on the diagonal.
unique([])                         % Place the values from both diagonals
                                   % into a vector and remove duplicates.

Mathematica, 42 bajty

Union@Flatten@Table[{i,#+1-i}+i#-#,{i,#}]&

Wypróbuj online!

@ KellyLowder grał w golfa do ...

Mathematica, 37 bajtów

##&@@@Table[{i-#,1-i}+i#,{i,#}]⋃{}&

a @alephalpha wyrzuciło stół!

Mathematica, 34 bajty

Union@@Range[{1,#},#^2,{#+1,#-1}]&

Proton , 41 bajtów

n=>[i+1for i:0..n*n if!(i%-~n)or!(i%~-n)]

Wypróbuj online!

MATL , 14 bajtów

U:GeGXytPY|*Xz

Wypróbuj online!

C (gcc) , 65 58 bajtów

-7 bajtów dzięki Titusowi!

f(n,i){for(i=0;i<n*n;i++)i%-~n&&i%~-n||printf("%d ",i+1);}

Wypróbuj online!

C # (.NET Core) , 97 83 bajtów

f=>{var s="[";for(int i=0;i<n*n-1;)s+=i%-~n<1|i++%~-n<1?i+",":"";return s+n*n+"]";}

Wypróbuj online!

Zmiana tutaj opiera się na przesunięciu między liczbami do znalezienia. Dwie zmiany zaczynające się od 0 są, n-1a n+1więc jeśli n=5liczby dla n-1byłyby 0,4,8,12,16,20i n+1byłyby 0,6,12,18,24. Łącząc je i dając 1-indeksowanie (zamiast 0-indeksowanie) daje 1,5,7,9,13,17,19,21,25. Przesunięcie od njest uzyskiwane za pomocą negacji bitowej (operacja bitowego uzupełnienia), gdzie ~-n==n-1i-~n==n+1 .

Stara wersja

f=>{var s="[";for(int i=0;i<n*n-1;i++)s+=(i/n!=i%n&&n-1-i/n!=i%n?"":i+1+",");return s+$"{n*n}]";}

Wypróbuj online!

Podejście to wykorzystuje indeksy kolumn i wierszy do ustalenia, czy liczby są na przekątnych. i/ndaje indeks wiersza i i%ndaje indeks kolumny.

Zwracanie tylko tablicy liczb

Jeśli uważa się, że konstruowanie tylko tablicy liczb liczy się do kosztu bajtu, wówczas można wykonać następujące czynności, w oparciu o sugestię Dennis.Verweij ( using System.Linq;dodaje dodatkowe 18 bajtów):

C # (.NET Core) , 66 + 18 = 84 bajtów

x=>Enumerable.Range(1,x*x).Where(v=>~-v%~-x<1|~-v%-~x<1).ToArray()

Wypróbuj online!

JavaScript, 73 63 bajty

stara wersja

n=>[...Array(y=n*n).keys(),y].filter(x=>(--x/n|0)==x%n||(x/n|0)==n-x%n-1)

Zaoszczędź 10 bajtów dzięki @Shaggy

n=>[...Array(n*n)].map((_,y)=>y+1).filter(x=>!(--x%-~n&&x%~-n))

Pierwszy raz w golfa! mam nadzieję, że nie zepsułem się tak bardzo.

let f=n=>[...Array(n*n)].map((_,y)=>y+1).filter(x=>!(--x%-~n&&x%~-n));
[1,2,3,4,5].forEach(n=>console.log(f(n)))

Pyth , 20 18 bajtów

^{( Oto wersja początkowa .)}

hMf|<%ThQ1<%TtQ1U*

Pakiet testowy.

Pyth , 18 bajtów

hMf>1*%ThQ%T|tQ1U*

Pakiet testowy.

Perl 5 , 56 + 1 (-n) = 57 bajtów

!(($_+1+$_/$,)%$,&&$_%($,+1))&&say++$_ for 0..($,=$_)**2

Wypróbuj online!

Java (OpenJDK 8) , 71 bajtów

n->{for(int i=0;i<n*n;i++)if(i%-~n<1||i%~-n<1)System.out.println(i+1);}

Wypróbuj online!

Port szkockiej odpowiedzi .

Japt , 16 bajtów

Wydaje się, że nie radzę sobie lepiej, ale jestem pewien, że to możliwe. Musiałem poświęcić 2 bajty z powodu niepotrzebnego wymagania, że ​​używamy 1-indeksowania.

²õ f@´XvUÉ ªXvUÄ

Sprawdź to

Oktawa, 32 bajty

@(n)find((m=abs(--n:-2:-n))==m')

Wypróbuj online!

PHP, 56 54 + 1 bajtów

+1 bajt dla -Rflagi

for(;$z**.5<$n=$argn;$z++)$z%-~$n&&$z%~-$n||print~+$z;

drukuje liczby poprzedzone myślnikami. Uruchom jako potok z -nRlub spróbuj online .

wymaga PHP w wersji 5.6 lub nowszej dla **operatora.
Dodaj jeden bajt dla starszych PHP: Wymienić ;$z**.5<$n=$argnz $z=$argn;$z<$n*$n.

Rubinowy, 45 bajtów

->n{(n*n).times{|i|i%-~n>0&&i%~-n>0||p(i+1)}}

Działa wewnętrznie jako indeksowany zerem. sprawdza, czy imodulo n+1lub n-1ma wartość 0, jeśli tak drukuje i+1.