updated to Hackerrank given code stab

Author: arsho

PythonFunctionals/MapandLambdaFunction.py

@@ -4,24 +4,19 @@
 Domain    : Python
 Author    : Ahmedur Rahman Shovon
 Created   : 15 July 2016
+Updated   : 08 February 2023
 Problem   : https://www.hackerrank.com/challenges/map-and-lambda-expression/problem
 """
-# Enter your code here. Read input from STDIN. Print output to STDOUT
-def sqr(a):
-    return a * a * a
+cube = lambda x: x * x * x
 
-
-n = int(input())
-if n == 0:
-    print("[]")
-elif n == 1:
-    print("[0]")
-else:
-    ar = [0] * n
-    ar[0] = 0
-    ar[1] = 1
+def fibonacci(n):
+    ar = [0, 1]
+    if n < 2:
+        return ar[:n]
     for i in range(2, n):
-        ar[i] = ar[i - 1] + ar[i - 2]
+        ar.append(ar[i-1] + ar[i-2])
+    return ar
 
-    ar = map(sqr, ar)
-    print(list(ar))
+if __name__ == '__main__':
+    n = int(input())
+    print(list(map(cube, fibonacci(n))))