並んだボール

下図ように、１辺にｎ個のボールが正方形の形に並んでいます。
このときのボ－ルの数をあらわす式を書きました。
下図は、どの式の考え方を図にしたものでしょう。