Up and down, up and down

Question

Challenge:

Given a positive integer input n, create a vector that follows this pattern:

0  1  0 -1 -2 -1  0  1  2  3  2  1  0 -1 -2 -3 -4 -3 -2 -1 ... ±(n-1) ±n

Or, explained with words: The vector starts at 0, and makes increments of 1 until it reaches the smallest odd positive integer that isn't part of the sequence, then it makes decrements until it reaches the smallest (in magnitude) even negative integer that isn't part of the sequence. It continues this way until n is reached. The sequence will end on positive n if n is odd, and negative n if n is even.

The output format is flexible.

Test cases:

n = 1
0  1
-----------
n = 2
0  1  0 -1 -2
-----------
n = 3
0  1  0 -1 -2 -1  0  1  2  3
-----------
n = 4
0  1  0 -1 -2 -1  0  1  2  3  2  1  0 -1 -2 -3 -4
-----------
n = 5
0  1  0 -1 -2 -1  0  1  2  3  2  1  0 -1 -2 -3 -4 -3 -2 -1  0  1  2  3  4  5

You may choose to take the n zero-indexed. n = 1 would then give 0 1 0 -1 -2.

This is code-golf, so the shortest code in each language wins! Explanations are encouraged as always!

Answer 1

R, 58 54 50 48 43 bytes

-2 bytes thanks to MickyT

function(n)diffinv(rep(1:n%%2*2-1,1:n*2-1))

Try it online!

function(n)
 diffinv(                           # take cumulative sum, starting at 0 of
             1:n%%2*2-1,            # a vector of alternating 1,-1
         rep(                       # repeated
                        1:n*2-1))   # 1, 3, 5, etc. times

Answer 2

Perl 6,  60  26 bytes

{flat {((1,-*...*)ZX*(-$++...0...$++)xx$_)}(),$_*($_%2||-1)}

Try it

{[...] (-1,-*...*)Z*0..$_}

Try it

Expanded:

{  # bare block lambda with implicit parameter $_

  [...]  # reduce using &infix:«...» (sequence generator)

          ( -1, -* ... * ) # (-1, 1, -1, 1 ... *)

      Z*                   # zip multiplied with

          0 .. $_          # range up to and including input
}

(-1,-*...*)Z*0..$_ generates the sequence 0 1 -2 3 -4 5

Answer 3

Python 2, 69 57 56 bytes

f=lambda n:[0][n:]or f(n-1)+range(-n,n+1)[::n%2*2-1][2:]

Try it online!

For each n up to the input the range(-n,n) (inclusive) is calculated, inverted when n is an even number, has the fist two numbers (after the inversion) removed, and then appended to the output.

Answer 4

05AB1E, 9 7 bytes

Saved 2 bytes thanks to @Emigna

Ýā®sm*Ÿ

Try it online!

My first 05AB1E answer (I think), so I may be missing some tricks...

Explanation

Ý         # push range [0 ... n]   stack: [[0 ... n]]
 ā        # push range [1 ... len(prev)]  [[0 ... n], [1 ... n+1]]
  ®       # push value of register        [[0 ... n], [1 ... n+1], -1]
   s      # swap top two values           [[0 ... n], -1, [1 ... n+1]]
    m     # power                         [[0 ... n], [-1, 1, -1, 1, ...]]
     *    # multiply                      [[0, 1, -2, 3, -4, 5, ...]]
      Ÿ   # range interpolation           [[0, 1, 0, -1, -2, -1, ...]]

I have to thank @Dennis for the original usage of Ÿ, otherwise I may not probably would never have known about it...

Answer 5

JavaScript (ES6), 56 bytes

f=(n,b=d=1,k=0)=>[k,...k-d*n?f(n,k-b?b:(d=-d)-b,k+d):[]]

Try it online!

Commented

f = (               // f = recursive function taking:
  n,                //   n = input
  b =               //   b = boundary value, initialized to 1
  d = 1,            //   d = current direction, initialized to 1
  k = 0             //   k = current sequence value, initialized to 0
) =>                //
  [                 // update the sequence:
    k,              //   append the current value
    ...k - d * n ?  //   if |k| is not equal to |n|:
      f(            //     append the (spread) result of a recursive call:
        n,          //       use the original input
        k - b ?     //       if k has not reached the boundary value:
          b         //         leave b unchanged
        :           //       else:
          (d = -d)  //         reverse the direction
          - b,      //         and use a boundary of higher amplitude and opposite sign
        k + d       //       update k
      )             //     end of recursive call
    :               //   else:
      []            //     stop recursion and append nothing
  ]                 // end of sequence update

Answer 6

05AB1E, 15 14 bytes

ÝDÉ·<*Ý€û˜ÔsF¨

Try it online!

Explanation

Ý                # push range [0 ... n]
 D               # duplicate
  É·<            # (x % 2 == 1)*2-1 for each
     *           # multiply
      Ý          # range [0 ... a] for each
       €û        # palendromize each
         ˜       # flatten
          Ô      # connected uniqueified
           sF¨   # remove the last n elements

Answer 7

Jelly, 11 9 bytes

²Ḷƽ-*0;Ä

Try it online!

How it works

²Ḷƽ-*0;Ä  Main link. Argument: n

²          Square; yield n².
 Ḷ         Unlength; yield [0, ..., n²-1].
  ƽ       Take the integer square root of each k in the range.
    -*     Compute (-1)**r for each integer square root r.
      0;   Prepend a zero.
        Ä  Accumulate; take the sums of all prefixes.

Answer 8

Haskell, 48 42 bytes

f n=0:[(-1)^i*x|i<-[0..n-1],x<-[1-i..i+1]]

Try it online!

Thanks to Οurous for -1 byte

Even though it's kind of obvious in hindsight, it took me a while to arrive at (-1)^i*x which is x when i is even and -x when i is odd. Previous iterations where:

(-1)^i*x
x-2*mod i 2*x
(-1)^mod i 2*x
[x,-x]!!mod i 2
(1-sum[2|odd i])*x

Answer 9

Haskell, 43 bytes

f n=scanl(-)0[(-1)^k|k<-[1..n],_<-[2..2*k]]

Try it online!

Computes the negated cumulative sums of the list [(-1)^k|k<-[1..n],_<-[2..2*k]], which is the first n rows of

[-1,
 +1, +1, +1,
 -1, -1, -1, -1, -1,
 +1, +1, +1, +1, +1, +1, +1…

Answer 10

J, 25 bytes

-5 bytes thanks to FrownyFrog!

>:@*:$i.;@(<@i:@*_1&^)@,]

Try it online!

J, 30 bytes

>:@*:{.;@([:(i:@*_1&^)&.>i.,])

Explanation:

i.,] creates list 0..n

&.> for each number in the list execute the verb in (...) and box the result (I need boxing because the results have different length)

[:( _1&^) find -1 to the ith power (-1 or 1)

i:@* make a list -n..n or n..-n, depending on the sign of the above

;@ unbox

>:@*: find n^2 + 1

}. and take so many numbers from the list

Try it online!

Answer 11

C# (.NET Core), 300 167 bytes

I've never done any of these before, but this one seemed fun. I see why people use those "golfing" languages as 167 seems way higher than some of the other answers. But, you gotta go with what you know.

static int[] f(int n){if (n==1) return new int[]{0,1};var a=f(n-1);return a.Concat(a.Skip(a.Length-(n-1)*2).Select(x=>-x)).Concat(new int[]{(n%2)!=0?n:-n}).ToArray();}

Try it online!

// Recursive Worker Function
static public int[] f( int n )
{
    // Start with the simple case
    if ( n == 1 ) return new int[]{0,1};

    // Recusively build off of that
    var a = f(n-1);

    // To be added at the end
    int[] b = { (n%2) !=0 ? n : -n };

    // Skip some based on length
    int s = a.Length - (n-1)*2;

    // With the rest, multiply by -1 and then append to the end
    // And append the part
    return a.Concat( a.Skip(s).Select( x => -x ) ).Concat( b ).ToArray();
}

Answer 12

Python 2, 65 56 bytes

r=k=0;exec'print r;r+=1-k**.5//1%2*2;k+=1;'*-~input()**2

The output format is a bit ugly. :/

Try it online!

Answer 13

Java 8, 85 83 79 bytes

n->{for(int p=0,i=0;i<=n*n;p+=1-(int)Math.sqrt(i++)%2*2)System.out.println(p);}

-6 bytes thanks to @OlivierGrégoire.

Try it online.

Explanation:

n->{                            // Method with integer parameter and no return-type
  for(int p=0,                  //  Set both `p` to 0
      i=0;i<=n*n;               //  Loop `i` in the range [0, `n*n`]
      p+=                       //    After every iteration, increase `p` by:
         1-                     //     1, minus:
           (int)Math.sqrt(i++)  //     The square-root of `i`, truncated to its integer
           %2*2)                //     Modulo 2, and multiplied by 2
     System.out.println(p);}    //   Print integer `p` with a trailing new-line

Answer 14

R, 48 46 42 bytes

for(i in 1:scan())F=c(F,-(-1)^i*(2-i):i);F

Try it online!

A port of the Ruby answer by Kirill L. - and saved 6 bytes thanks to the same Kirill L.! Now shorter than Giuseppe's solution ;)

A port of this Octave answer by Luis Mendo using approx is less golfy. n=n^2+1 can be replaced by ,,n^2+1; or by 0:n^2+1(positional argument xout) for the same byte count :

R, 56 bytes

f=function(n)approx((0:n)^2+1,-(-1)^(0:n)*0:n,n=n^2+1)$y

Try it online!

Answer 15

APL (Dyalog Unicode), 17 bytes

+\0,¯1*⍳(/⍨)1+2×⍳

Try it online!

Golfed 2 bytes thanks to @FrownyFrog by converting to a train. See the older answer and its explanation below.

---

APL (Dyalog Unicode), 19 bytes

+\0,∊⊢∘-\⍴∘1¨1+2×⍳⎕

Try it online!

(Uses ⎕IO←0)

My first approach was to construct multiple ranges and concatenate them together, this easily went over 30 bytes. Then I started analysing the sequence

      +\⍣¯1⊢0  1  0 ¯1 ¯2 ¯1  0  1  2  3  2  1  0 ¯1 ¯2 ¯3 ¯4
0 1 ¯1 ¯1 ¯1 1 1 1 1 1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1

^{+\⍣¯1 denotes the inverse cumulative sum}

There is a repeating pattern of 1s and ¯1s, where the length of each consecutive sequence of 1s or ¯1s is 1+2×n. And each subsequence alternates between 1 and ¯1. What I can do now is to create the 1s and ¯1s list, and then scan by +

      ⍳4 ⍝ creates range 0..4
0 1 2 3
      2×⍳4
0 2 4 6
      1+2×⍳4
1 3 5 7
      ⍴∘1¨1+2×⍳4 ⍝ for-each create that many 1s
┌─┬─────┬─────────┬─────────────┐
│1│1 1 1│1 1 1 1 1│1 1 1 1 1 1 1│
└─┴─────┴─────────┴─────────────┘
      ⊢∘-\⍴∘1¨1+2×⍳4 ⍝ alternate signs
┌─┬────────┬─────────┬────────────────────┐
│1│¯1 ¯1 ¯1│1 1 1 1 1│¯1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1│
└─┴────────┴─────────┴────────────────────┘
      ∊⊢∘-\⍴∘1¨1+2×⍳4 ⍝ flatten
1 ¯1 ¯1 ¯1 1 1 1 1 1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1
      0,∊⊢∘-\⍴∘1¨1+2×⍳4
0 1 ¯1 ¯1 ¯1 1 1 1 1 1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1 ¯1
      +\0,∊⊢∘-\⍴∘1¨1+2×⍳4 ⍝ cumulative sum
0 1 0 ¯1 ¯2 ¯1 0 1 2 3 2 1 0 ¯1 ¯2 ¯3 ¯4

Answer 16

Haskell, 47 bytes

(#0)
m#n|m>n=[-n..n]++map(0-)(m#(n+1))|1>0=[-m]

Try it online!

Answer 17

MATL, 17 15 bytes

-2 bytes thanks to Luis Mendo!

0i:oEqG:EqY"Ysh

Try it online!

Explanation for n=3:

0		% push 0
 i:		% read input as integer, push range
		% stack: [0, [1 2 3]]
   o		% modulo 2, stack: [0, [1 0 1]]
    Eq		% double and decrement, stack: [0, [1 -1 1]]
      G:	% push input and range again
		% stack: [0, [1 -1 1], [1 2 3]]
        Eq	% double and decrement,
		% stack: [0, [1 -1 1], [1 3 5]]
	  Y"	% run-length decoding
		% stack: [0, [1 -1 -1 -1 1 1 1 1 1]]
	    Ys	% cumulative sum
		% stack: [0, [1  0 -1 -2 -1  0  1  2  3]]
	      h	% horizontally concatenate
		% end of program, automatically print the stack

Answer 18

Octave, 44 42 41 bytes

2 bytes removed thanks to @StewieGriffin, and 1 byte further removed thanks to @Giuseppe!

@(n)interp1((t=0:n).^2,-t.*(-1).^t,0:n^2)

Try it online!

Answer 19

Java (JDK 10), 98 bytes

n->{var s="0";for(int i=0,r=0,d=1;i++<n;s+=" "+r,d=-d)for(r+=d;r!=i&r!=-i;r+=d)s+=" "+r;return s;}

Try it online!

Answer 20

Ruby, 52 47 bytes

f=->n{n<1?[0]:f[n-1]+(2-n..n).map{|x|-~0**n*x}}

Try it online!

Below is the original 52-byte version with an explanation:

f=->n{n<1?[0]:f[n-1]+[(r=*2-n..n).map(&:-@),r][n%2]}

Try it online!

Walkthrough

f=->n{           #Recursive approach
 n<1?[0]         #Init with 0 if n=0
 :f[n-1]         #else make a recursive call
 +               #and append an array of numbers
 [(r=*2-n..n)    #Init r as splatted range from 2-n to n
 .map(&:-@)      #"-@" is unary minus, so this a fancy way to do map{|x|-x} for -1 byte
                 #For even n use this negated r, e.g. for n=4: [2, 1, 0, -1, -2, -3, -4]
 ,r]             #For odd n use r directly, e.g. for n=3: [-1, 0, 1, 2, 3]
 [n%2]           #Odd/even selector
}

Answer 21

Prolog (SWI), 113 bytes

0-I-[0|I].
N-[H|T]-R:-N is -H*(-1)^N,A is N-1,A-[H|T]-R;I is H-(-1)^N,N-[I|[H|T]]-R.
N-O:-X is -N*(-1)^N,N-[X]-O.

Try it online!

Answer 22

Python 3, 83 bytes

def c(n):print([(-1)**j*(abs(j-i)-j)for j in range(n+1)for i in range(2*j)][:-n+1])

Answer 23

Husk, 18 17 bytes

:ṁoṡ₁ŀ⁰_₁⁰
*^⁰_1⁰

Try it online!

Answer 24

Charcoal, 19 bytes

Ｆ⊕ＮＩ×∨﹪ι²±¹…·∧ι⁻²ιι

Try it online! Link is to verbose version of code. Explanation:

  Ｎ                 Input as a number
 ⊕                  Increment
Ｆ                   Loop over implicit range
                ²   Literal 2
                 ι  Current index
               ⁻    Subtract
              ι     Current index
             ∧      Logical And
                  ι Current index
           …·       Inclusive range
       ι            Current index
        ²           Literal 2
      ﹪             Modulo
          ¹         Literal 1
         ±          Negate
     ∨              Logical Or
    ×               Multiply
   Ｉ                Cast to string and implicitly print

Alternative explanation:

Ｆ⊕Ｎ

Loop over the integers from 0 to the input inclusive.

Ｉ

Cast the results to string before printing.

×∨﹪ι²±¹

Negate alternate sets of results.

…·∧ι⁻²ιι

Form a list from the previous index to the current index, excluding the previous index.

Answer 25

Jelly,  11  12 bytes

^{Bah, I thought I had 11 wih _2+ỊrN)N;¥/}

_2+ỊrN×-*$)Ẏ

Try it online!

How?

_2+ỊrN×-*$)Ẏ - Main Link: n           e.g. 4
          )  - for x in [1...n]:           1       2          3               4
_2           -   subtract 2 from x        -1       0          1               2
   Ị         -   is x insignificant?       1       0          0               0
  +          -   add                       0       0          1               2
     N       -   negate x                 -1      -2         -3              -4
    r        -   inclusive range          [0,-1]  [0,-1,-2]  [1,0,-1,-2,-3]  [2,1,0,-1,-2,-3,-4]
         $   -   last two links as a monad:
       -     -     minus one              -1      -1         -1              -1
        *    -     raised to the power x  -1       1         -1               1
      ×      -   multiply                 [0,1]   [0,-1,-2]  [-1,0,1,2,3]    [2,1,0,-1,-2,-3,-4]
           Ẏ - tighten                    [0,1,0,-1,-2,-1,0,1,2,3,2,1,0,-1,-2,-3,-4]

Answer 26

APL (Dyalog Unicode), 34 32 bytes

⎕CY'dfns'⋄-0,∘∊1↓¨2to/Sׯ1*S←⍳,⊢

Try it online!

Requires ⎕IO←0

-2 bytes thanks to @FrownyFrog

Answer 27

Stax, 10 bytes

╗)SΘ█☼₧ΘP(

Run and debug it

Answer 28

Scala, 119 Bytes

def a(n: Int)={lazy val s:Stream[Int]=0#::Stream.from(0).map{x=>s(x)+1 -2*(Math.sqrt(x).toInt%2)}
s.take(n*n+1).toList}

Ungolfed:

def a(n: Int)={
  lazy val s:Stream[Int]= 0#::Stream.from(0).map //Give the starting point and indexing scheme
  {
    x=>
    {
      val sign = 1-2*(Math.sqrt(x).toInt%2) //Determine whether we are adding or subtracting at the current index
      s(x)+sign
    }
  }
  s.take(n*n+1).toList //Take the desired values
}

This can probably be golfed much better, but I wanted a solution utilizing lazy Streams.

Answer 29

Stacked, 44 bytes

[~>0\:2%\#,2*1-tr[...rep]flatmap,$sumonpref]

Try it online! It's been a while since I programmed in Stacked, but I think I still got it.

Alternatives

73 bytes: [0\|>:2%tmo*2 infixes[:...|>\rev...|>rev#,$#'sortby 1#behead]flatmap 0\,]

This goes with the "ranges from generated indices" approach used in my Attache answer. This proved to be pretty long, since Stacked has no builtin for reversed ranges nor collapsing. (That's what :...|>\rev...|>rev#,$#'sortby 1#behead does.)

53 bytes: [0\|>:2%tmo _\tpo#,tr[...rep]flatmap 0\,inits$summap]

...so I decided to go for an approach which instead finds the cumulative sum (inits$summap) over 1 and -1 repeated by the odd integers, as in the R answer.

46 bytes: [~>0\:2%\#,2*1-tr[...rep]flatmap,inits$summap]

...but I realized that the negative integers and the odd integers could be made in one go, by multiplying both generated arrays (the mod 2 values of the range and the range itself) by 2 then subtracting 1. This gives alternating 1s and -1s for the first range and the odd integers for the second!

44 bytes: [~>0\:2%\#,2*1-tr[...rep]flatmap,$sumonpref]

... and then I remembered I had a builtin for mapping prefixes. ^-^

Answer 30

Julia 0.6, 44 bytes

n->[(i%2*2-1)*[0:i;(n>i)*~-i:-1:1]for i=1:n]

Try it online!

Since OP mentions "the output format is flexible", this prints an array of sub arrays, eg. U(3) => [[0, 1], [0, -1, -2, -1], [0, 1, 2, 3]].

i%2*2-1 decides the sign of the current subarray - negative for even numbers, positive for odd.

[0:i;(n>i)*~-i:-1:1] is in two parts. 0:i is straightforward, the range of values from 0 to the current i. In the next part, ~-i:-1:1 is the descending range from i-1 to 1. But we want to append this only if we're not yet at the final value, so multiply the upper end of the range by (n>i) so that when n==i, the range will be 0:-1:1 which ends up empty (so the array stops at n correctly).

---

And here's a version that can support random access - the inner lambda here returns the i'th term of the sequence without having to have stored any of the terms before it. This one gives the output as a single neat array too.

49 47 bytes

n->map(i->((m=isqrt(i))%2*2-1)*(m-i+m^2),0:n^2)

Try it online!

Answer 31

JavaScript (Node.js), 83 bytes

n=>[v=0].concat(...[...Array(n),c=d=-1].map(t=>[...Array(c+=2)].map(t=>v+=d,d=-d)))

Try it online!

Answer 32

Jelly, 17 16 15 bytes

-*N׌Ḅ)j00;ḣN‘$

Try it online!

Answer 33

Wolfram Language (Mathematica), 56 51 bytes

-5 bytes thanks to user202729

Join@@MapAt[-#&,Table[Range[-k,k],{k,#}],1;;-1;;2]&

Try it online!

Answer 34

Clean, 59 bytes

import StdEnv
$m=foldr(\a=(++)[~a..a]o map~)[][0..m-1]++[m]

Try it online!

Defines the function $ :: Int -> [Int] generating the required list.

A direct port of user1472751's Haskell solution ended up being slightly better:

Clean, 53 bytes

import StdEnv
$n m|m>n=[~n..n]++map~($(n+1)m)=[~m]

$0

Try it online!

Answer 35

Ruby, 59 bytes

->n{a,*b=-1;n.times{|z|b+=[*(z*a).step(z*a=-a,a)]};b+[n*a]}

Try it online!

Answer 36

C# (.NET Core), 97 bytes

n=>{var r="";int m=0,a=0,i=0;for(;++i<=n;)for(m=i%2>0?1:-1;a!=i*m;a+=m)r+=a+" ";r+=n*m;return r;}

Try it online!

Another different approach in C#.

Answer 37

CJam, 59 bytes

1]q~:X*ee{~[\_-1\#@*](2*)*}%:+W%~[]\_@+{@@+_@+}XX*(*W%0\+p;

Try it online!

Suggestions for improvement are welcome!

EDIT: forgot leading zero

Answer 38

Python 3, 83 bytes

lambda n:numpy.cumsum([0]+sum([[(-1)**i]*(i-~i)for i in range(n)],[]))
import numpy

Try it online!

I'll throw in a numpy solution, just for fun.

Answer 39

Swift, 293 bytes

let odd = n % 2 != 0
var i = 0
var inc = 1
var s : [Int] = [0]
repeat {
    i += inc
    if (inc == 1 && i % 2 != 0) || (inc == -1 && i % 2 == 0) {
        if !s.contains(i) {
            inc = -inc
        }
    }
    s.append(i)
} while !((odd && i == n) || (!odd && i == -n))
print("\(s)")

Answer 40

Attache, 38 bytes

Collapse@@{`::&>Pairs!On[Even,`-,0:_]}

Try it online!

I'm struggling to break 38 bytes... I'll post an explanation once I'm convinced I can't get lower than this (or if I golf it!).

Answer 41

JavaScript (Node.js), 51 bytes

n=>{for(p=i=0;i<=n*n;p+=1-(i++**.5|0)%2*2)alert(p)}

Try it online!

I stole toke ripped off used @KevinCruijssen's answer.

Thanks to @StewieGriffin for adding one more byte to my count (I forgot to output the last number) read the comments

Answer 42

Yabasic, 83 bytes

An anonymous function that takes Input, n as an integer and outputs to the console.

Input""n
For i=1To n
For j=-i To i-1
k=i-Abs(j)
?(-1)^(i+1)*k;
If k=n End
Next
Next

Try it online!