Skip to main content
deleted 9 characters in body
Source Link
ZaMoC
  • 25.5k
  • 2
  • 32
  • 87

Wolfram Language (Mathematica), 4642 bytes

Flatten[Array[{1,Table[2,2#+1]If[#==1||(d=NumberQ@*Sqrt)@#||d[#+1],1}&,#,0]][[#]]&2]&  

Try it online! returns 1 if n=1 or n is a perfect square or n+1 is a perfect square

Try it online!

Wolfram Language (Mathematica), 46 bytes

Flatten[Array[{1,Table[2,2#+1],1}&,#,0]][[#]]& 

Try it online!

Wolfram Language (Mathematica), 42 bytes

If[#==1||(d=NumberQ@*Sqrt)@#||d[#+1],1,2]&  

returns 1 if n=1 or n is a perfect square or n+1 is a perfect square

Try it online!

Source Link
ZaMoC
  • 25.5k
  • 2
  • 32
  • 87

Wolfram Language (Mathematica), 46 bytes

Flatten[Array[{1,Table[2,2#+1],1}&,#,0]][[#]]& 

Try it online!