 # Problem code ZCO14001 solution

I have written a program for problem code ZCO14001 and submitted it but it didn’t even pass one test case. The thing is the I think the program is correct and I was also able to get the correct output for the two sample cases when I enter them myself so even if it was not fully correct, I should have some test cases pass atleast.

``````class Controls:
def __init__(self, end, max_h, stack):
self.position = 0
self.end = end
self.stack = stack
self.max = max_h

def move_left(self):
if self.position > 0:
self.position -= 1

def move_right(self):
if self.position < self.end-1:
self.position += 1

def pick(self):
if self.stack[self.position] > 0 and not self.is_loaded:
self.stack[self.position] -= 1

def drop(self):
if self.stack[self.position] < self.max and self.is_loaded:
self.stack[self.position] += 1

line1 = input()
line1 = line1.split(" ")
stack_no = int(line1)
max_height = int(line1)

line2 = input()
line2 = list(map(int, line2.split(" ")))

while len(line2) < stack_no:
line2.append(0)

line3 = input()
line3 = list(map(int, line3.split(" ")))

controller = Controls(stack_no, max_height, line2)

for instruction in line3:
if instruction == 0:
break
elif instruction == 1:
controller.move_left()
elif instruction == 2:
controller.move_right()
elif instruction == 3:
controller.pick()
elif instruction == 4:
controller.drop()
else:
print("Error")

print(controller.stack)
``````

I am using python 3.6. Can anyone tell me if my program is incoreect or if it just isn’t formatted correctly? I am a noob so thanks for help.

From the above code, your output will be in list format: [0, 1, 2, 3] etc.

The requirement is for output as a space-separated list. You can achieve that easily by unpacking your `.stack` by prefixing an asterisk:

``````print( *controller.stack )
``````

This feeds the iterable `stack` as a stream of parameters to the `print` function rather than as a single data item.

Thanks, It worked.

//