Devuelve el índice de vecinos en una cuadrícula de 3x3


11

Muy bien, mi segundo intento en un código de golf, veamos cómo va esto.

Imagina que tienes una matriz de 9 valores. Ahora imagine esa matriz en una cuadrícula de 3x3.

Debe devolver a los vecinos que ese número tiene como índices de la matriz.

0 | 1 | 2

3 | 4 | 5 5

6 | 7 | 8

Reglas:

  • Es el código de golf, por lo que gana la respuesta más corta.
  • El índice de la matriz de simulación puede comenzar en 0 o 1. (todos los ejemplos usan 0)
  • Solo devolver valores valores está mal visto (como if 3: return 046)
  • El envío puede ser solo un procedimiento / función / método, pero un ejemplo sería bueno
  • El valor devuelto puede estar en cualquier orden (como si la entrada es 0, podría ser 13 o 31)
  • si lo desea, la salida puede ser una lista de números, por ejemplo, en [0,4,6]lugar de046
  • las diagonales no cuentan, como se ve en los ejemplos.

Ejemplos:

entrada:

0 0

salida:

13

entrada:

3

salida:

046

entrada:

4 4

salida:

1357


44
Parece que este desafío podría beneficiarse de algún tiempo en el Sandbox . Puede publicar su desafío allí para que otros puedan revisarlo y ayudarlo antes de publicarlo en main. De tus ejemplos, supongo que no estás contando diagonales. Es posible que desee agregar esto a la pregunta en sí. También menciona el requisito de generar los índices de la matriz que son vecinos. Creo que esto podría ser codificado para una cuadrícula de 3x3. ¿Sería mejor sacar a los vecinos?
Poke

77
Para que lo sepas, mal visto no es realmente algo que hacemos aquí; la codificación de la salida está permitida o no. Como generalmente es bastante difícil definir qué es exactamente lo que cuenta como hardcoding, personalmente lo permitiría o le daría el tamaño de la cuadrícula como una entrada adicional.
Dennis

1
¿Puede la salida ser una lista de números, por ejemplo, en [0,4,6]lugar de 046?
Laikoni

@Laikoni Sí, un poco tarde porque ya lo has respondido.
hcorion

@ Dennis Sí, no estaba muy seguro de cómo decirlo. Me gusta cómo lo hicieron las respuestas C y python, proporcionando ambas, pero teniendo la respuesta no codificada como final. Quería alentar algoritmos en lugar de codificar, pero no estaba seguro de si era posible (sin respuestas demasiado largas), y no quería no tener respuestas a mi pregunta.
hcorion

Respuestas:


2

Jalea , 16 13 bytes

9Ḷ,d3ạ/S€=1T’

Pruébalo en línea!

Cómo funciona

9Ḷ,d3ạ/S€=1T’  Main link. Argument: n (0, ..., 8)

9              Set the return value to 9.
 Ḷ             Unlength; yield [0, ..., 8].
  ,            Pair; yield [[0, ..., 8], n].
   d3          Divmod 3; yield [[[0, 0], ..., [2, 2]], [n:3, n%3]]].
     ạ/        Reduce by absolute difference, yielding
               [[|0 - n:3|, |0 - n%3|], ..., [[|2 - n:3|, |2 - n%3|]].
       S€      Sum each, yielding
               [|0 - n:3| + |0 - n%3|, ..., [|2 - n:3| + |2 - n%3|].
         =1    Compare the sums with 1.
           T   Truth; yield all 1-based indices of 1.
            ’  Decrement to yield all 0-based indices of 1.

Las reglas establecen: "El índice de la matriz de simulación puede comenzar en 0 o 1." - Puedes soltar el Decremento al final.
steenbergh

@steenbergh Supongo que también tendría que tomar una entrada basada en 1, que cuesta tantos bytes como ahorra.
Dennis

9

MATL , 17 16 bytes

9:qWIe1Y6Z+i)BPf

La matriz está basada en 1, es decir, contiene números desde 1hasta 9.

Pruébalo en línea! O verificar todos los casos de prueba .

Explicación

Considere la entrada 2como un ejemplo.

9:q  % Push [0 1 2 ... 8]
     % STACK: [0 1 2 ... 8]
W    % Rise to 2, element-wise
     % STACK: [1 2 4 ... 256]
Ie   % Reshape as 3-row matrix (column-major order)
     % STACK: [1   8  64;
               2  16 128;
               4  32 256]
1Y6  % Push [0 1 0; 1 0 1; 0 1 0]
     % STACK: [1   8  64;
               2  16 128;
               4  32 256],
              [0   1   0;
               1   0   1;
               0   1   0]
Z+   % Convolution, maintaining size
     % STACK: [10  81 136;
               21 170 336;
               34 276 160]
i    % Take input, n
     % STACK: [10  81 136;
               21 170 336;
               34 276 160],
               2
 )   % Get n-th entry (1-based; column-major order)
     % STACK: 21
B    % Convert to binary
     % STACK: [1 0 1 0 1]
P    % Flip
     % STACK: [1 0 1 0 1]
f    % Find: gives indices of nonzeros. Implicitly display
     % STACK: [1 3 5]

1
Wat? ¿Cómo se te ocurrió esto?
Robert Fraser

1
@RobertFraser Estos desafíos para encontrar vecinos siempre me sugieren un enfoque de convolución. Pero la convolución inherentemente agrega los valores de los vecinos, por lo que necesitaba ser capaz de separarlos al final --- esa es la potencia de dos y las partes de expansión binarias
Luis Mendo

5

Mathematica, 32 bytes

GridGraph@{3,3}~AdjacencyList~#&

Utiliza un gráfico en lugar de una matriz. GridGraph@{3,3}construye un gráfico en forma de cuadrícula de 3x3, que se muestra a continuación, que Mathematica etiqueta útilmente con los números 1–9 para los vértices de forma predeterminada. Luego ~AdjacencyList~#&te dice los vecinos de un vértice.

El gráfico de cuadrícula de 3x3


Tengo que amar esas construcciones ...
Neil

4

Mathematica, 40 bytes

{24,135,26,157,2468,359,48,579,68}[[#]]&

1 indexado. Solo busca la respuesta. ¿Alguien puede mejorar en Mathematica?


3
Estoy sorprendido de que no haya nada incorporado para esto. Como esperaría que hubiera un generador incorporado para encontrar a todos los vecinos de un elemento en una matriz 2D, pero no estoy seguro, no sé nada sobre Mathematica aparte del hecho de que tiene demasiados componentes incorporados.
HyperNeutrino

2
Puede guardar un byte utilizando 0-indexing y 31[420,51,...,75][[#]]&.
Martin Ender

1
Puede usar GridGraph@{3,3}~AdjacencyList~#&para 32 bytes, con 1 indexación.
No es un árbol

@ lanlock4 ¡Impresionante! ¡Por favor, responda eso para que pueda votarlo!
Greg Martin

4

Octava, 42 40 39 bytes

@(n,x=~e(3),y=x(n)=1)find(bwdist(x)==1)

Índice basado en 1.

Verifique todos los casos de prueba.

Explicación:

x=~e(3);         % create a 3*3 matrix of zeros
x(n)=1;          % set the element with index n to 1
d=bwdist(x);     % compute the distance transform of the matrix
find(d == 1)     % find where the distance is 1.

Ejemplo: n = 2

x =

   0   0   0
   1   0   0
   0   0   0

(En Octave, los datos se almacenan en columnas).

d =

   1.00000   1.41421   2.23607
   0.00000   1.00000   2.00000
   1.00000   1.41421   2.23607

índice lógico donde la distancia es 1:

d == 1

 1   0   0
 0   1   0
 1   0   0

find(d ==1)

 1
 3
 5

3

Python 2, 71 bytes

lambda n:filter(abs,[(n-3)*(n>3),(n+3)*(n<7),~-n*(n%3!=1),-~n*(n%3>0)])

1 indexado ¡
Pruébelo en línea!


Obtener el resultado de una lista predefinida de resultados es más corto (46 bytes):

[13,204,15,406,1357,248,37,468,57].__getitem__

0 indexado ¡
Pruébelo en línea!


2

Haskell , 74 71 68 bytes

f n=[x|x<-[n-3,n-1..n+3],0<x,x<10,gcd 3x<2||n-1/=x,gcd 3n<2||n+1/=x]

Pruébalo en línea! Utiliza una cuadrícula indexada 1. Ejemplo de uso: f 3devoluciones [2,6].

Editar: ¡Guardado 3 6 bytes gracias a Ørjan Johansen!


Para 77 75 bytes, la siguiente función #funciona para un tamaño de cuadrícula arbitrario m:

n#m=[x|x<-[n-m,n-1,n+1,n+m],0<x,x<=m*m,gcd x m<m||n-1/=x,gcd n m<m||n+1/=x]

Pruébalo en línea! Para cada uno, nla lista [n-m,n-1,n+1,n+m]contiene los cuatro vecinos. Para cada entrada xen esta lista, verificamos -1<xy x<m*mpara asegurarnos de que xno esté encima o debajo de la cuadrícula, mod n 3>0||n-1/=xpara hacer cumplir el borde izquierdo de la cuadrícula y mod(n+1)m>0||n+1/=xpara el borde izquierdo.


1
Puedes usar [n-3,n-1..n+3]y gcd 3n>1.
Ørjan Johansen

Vaya, no importa esa gcdparte. Debería haber sido <3, y luego se rompe por n==0. Es posible que pueda usar ese truco si cambia todo a 1indexado.
Ørjan Johansen

Ah, y n/=2&&n/=5puede ser reemplazado por mod x 3>0. (O la gcdversión con reindexación, que ahora podría usarse dos veces.)
Ørjan Johansen

2

Ruby , 51 48 45 bytes

->a{[a+3,a-3][a/6..a/3]+[a+1,a-1][a%-3..a%3]}

Pruébalo en línea!

Cree 2 matrices, con vecinos verticales y horizontales, luego seleccione uno o más de ellos.

Ruby codificado, 44 ​​bytes

->a{%w(13 024 15 046 1357 248 37 468 57)[a]}

... No vale la pena.


2

C, 100 92 91 83 78 74 bytes

p(n){putchar(n+48);}f(n){n>3&&p(n-3);n<7&&p(n+3);n%3&&p(n+1);--n%3&&p(n);}

1 indexado. Gracias a @Neil por guardar 4 bytes.

Pruébalo en línea!

Versión codificada, 56 bytes

l[]={13,204,15,406,1357,248,37,468,57};
#define L(n)l[n]

0 indexado


2
En la primera versión, ¿no puedes escribir, n>3&&p(n-3)etc. para guardar 4 bytes? En la segunda versión, ¿no puedes escribir l[]=para guardar un byte?
Neil

@Neil Sí, puedo. ¡Gracias!
Steadybox

¿Estás seguro de que tu código es correcto actualmente? Cuando pruebo los casos de prueba falla para los tres ...: S Pruébelo aquí. ¿Podría proporcionar un enlace TIO que funcione, tal vez estoy haciendo algo mal?
Kevin Cruijssen

1
Enlace @KevinCruijssen TIO agregado, y parece que olvidé editar el código real en la última edición ... Oh, bueno. Su enlace también funciona correctamente, pero observe que mi respuesta está indexada en 1, mientras que los casos de prueba de ejemplo están indexados en 0.
Steadybox

@ Steadybox Ah, de hecho tienes razón. Me perdí la parte indexada, mi mal. Gracias por agregar el TIO. +1
Kevin Cruijssen

1

Python 2, 51 bytes

lambda x:[x+3,x-3][x/6:x/3+1]+[x+1,x-1][x%-3:x%3+1]

Basado en una versión anterior de mi respuesta de Ruby , me pareció interesante porque era principalmente el mismo código, usando un truco diferente, y produce el mismo resultado. Conseguir esto correctamente me ayudó a responder un poco más al golf.

Básicamente, ruby ​​lo tiene más corto porque el índice de corte de matriz es inclusivo, python necesita un +1para compensar.

Explicación

Obtenga las 2 matrices (vecinas verticales y horizontales), luego seleccione una o ambas en función de algunos cálculos.


1

Java 7, 63 bytes (codificado)

int c(int i){return new int[]{31,420,51,640,7531,842,73,864,75}[i];}

0 indexados
(salida orden inverso debido 024y 046no son números enteros válidos.)
Seguimos trabajando en una versión no codificada, pero les puedo asegurar que no será más corto ..

Pruébalo aquí.


82 bytes

String c(int n){return""+(n>3?n-3:"")+(n<7?n+3:"")+(n%3>0?n+1:"")+(--n%3>0?n:"");}

1 indexado
Basado en la respuesta de @Steadybox 'C

Pruébalo aquí.


0

JavaScript + lodash, 71 bytes

f=a=>_.range(9).filter(b=>a>b?f(b).includes(a):[,1,,1][b-a]&&b%3|a%3<2)

0

Lote, 116 bytes

@set c=cmd/cset/a%1
@set/ar=%1%%3
@if %1 gtr 2 %c%-3
@if %r% gtr 0 %c%-1
@if %r% lss 2 %c%+1
@if %1 lss 6 %c%+3

0 indexado.

Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.