Want to share your content on R-bloggers? click here if you have a blog, or here if you don't. This time we want to solve the following simple task with R: Take the numbers 1 to 100, square them, and add all the even numbers while subtracting the odd ones!

If you want to see how to do that in at least seven different ways in R, read on!

There are many different solutions possible, making use of several aspects of the R language. So this blog post can be seen as a fun exercise to recap some of the concepts explained in our introduction to R: Learning R: The Ultimate Introduction (incl. Machine Learning!).

First, as usual, you should try this for yourself… Ok, so if you didn’t have any ideas whatsoever you could have done it by hand, i.e. use R as a calculator:

2^2 + 4^2 + 6^2 + 8^2 + 10^2 + 12^2 + 14^2 + 16^2 + 18^2 + 20^2 + 22^2 + 24^2 + 26^2 + 28^2 + 30^2 + 32^2 + 34^2 + 36^2 + 38^2 + 40^2 + 42^2 + 44^2 + 46^2 + 48^2 + 50^2 + 52^2 + 54^2 + 56^2 + 58^2 + 60^2 + 62^2 + 64^2 + 66^2 + 68^2 + 70^2 + 72^2 + 74^2 + 76^2 + 78^2 + 80^2 + 82^2 + 84^2 + 86^2 + 88^2 + 90^2 + 92^2 + 94^2 + 96^2 + 98^2 + 100^2
- 99^2 - 97^2 - 95^2 - 93^2 - 91^2 - 89^2 - 87^2 - 85^2 - 83^2 - 81^2 - 79^2 - 77^2 - 75^2 - 73^2 - 71^2 - 69^2 - 67^2 - 65^2 - 63^2 - 61^2 - 59^2 - 57^2 - 55^2 - 53^2 - 51^2 - 49^2 - 47^2 - 45^2 - 43^2 - 41^2 - 39^2 - 37^2 - 35^2 - 33^2 - 31^2 - 29^2 - 27^2 - 25^2 - 23^2 - 21^2 - 19^2 - 17^2 - 15^2 - 13^2 - 11^2 - 9^2 - 7^2 - 5^2 - 3^2 - 1^2
##  5050


The result is 5050. But there are of course many much more elegant solutions. The first solution I thought of was the following, it makes use of the seq function:

sum(seq(2, 100, 2)^2 - seq(99, 1, -2)^2)
##  5050


An integral part is splitting the numbers into 50 even and 50 odd ones. There are several ways to do that. One way is to create both with the formulas 2n for even and 2n-1 for odd numbers:

n <- 1:50
even <- 2 * n
odd <- 2 * n - 1
sum(even^2 - odd^2)
##  5050


Another possibility is by subsetting with recycling

x <- 1:100
even <- x[c(FALSE, TRUE)] # subsetting with recycling
odd <- x[c(TRUE, FALSE)]
sum(even^2 - odd^2)
##  5050


…or elegantly by creating a matrix:

M <- matrix(1:100, nrow = 2)
sum(M[2, ]^2 - M[1, ]^2)
##  5050


If you come from another language, especially C and its derivatives you might have wanted to use a loop. This is of course also possible but discouraged in R (some say that you then “speak R with a C accent”):

s <- 0
for (x in 1:100) {
if (x %% 2) s <- s - x^2
else s <- s + x^2
}
s
##  5050


As you can see, inside of the loop is a conditional statement and the modulo operator (%%) to get the remainder of a division. The loop can easily be vectorized which is the preferred method in R:

x <- 1:100
sum(ifelse(x %% 2, -x^2, x^2)) # vectorized if statement
##  5050


Those were seven ways to get to the same correct result… and now for the bonus: if you think long enough about this little riddle you will see that is equivalent to adding up the original numbers (I leave this as an exercise, it is not too hard to see). The resulting code out of this analysis couldn’t be any simpler:

sum(1:100) # analytical
##  5050


That was fun, wasn’t it! If you want to share your own solution please do so in the comments below. If it is an especially clever, elegant, or creative one you will get an honorary mention in an update of this post!

P.S.: The Bart Simpson blackboard pic was created with the code provided in this post: Create Bart Simpson Blackboard Memes with R.