Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from collections import deque
- size = 6
- matrix = []
- water = 0
- metal = 0
- concrete = 0
- for _ in range(size):
- matrix.append(list(input().split()))
- rover_row = 0
- rover_col = 0
- for row in range(size):
- for col in range(size):
- if matrix[row][col] == "E":
- rover_row = row
- rover_col = col
- directions = deque(input().split(", "))
- while directions:
- current_direction = directions.popleft()
- if current_direction == "right":
- rover_col += 1
- if rover_col == size:
- rover_col = 0
- if matrix[rover_row][rover_col] == "W":
- water += 1
- print(f"Water deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "M":
- metal += 1
- print(f"Metal deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "C":
- concrete += 1
- print(f"Concrete deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "R":
- print(f"Rover got broken at ({rover_row}, {rover_col})")
- break
- elif current_direction == "left":
- rover_col -= 1
- if rover_col < 0:
- rover_col = size - 1
- if matrix[rover_row][rover_col] == "W":
- water += 1
- print(f"Water deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "M":
- metal += 1
- print(f"Metal deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "C":
- concrete += 1
- print(f"Concrete deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "R":
- print(f"Rover got broken at ({rover_row}, {rover_col})")
- break
- elif current_direction == "down":
- rover_row += 1
- if rover_row == size:
- rover_row = 0
- if matrix[rover_row][rover_col] == "W":
- water += 1
- print(f"Water deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "M":
- metal += 1
- print(f"Metal deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "C":
- concrete += 1
- print(f"Concrete deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "R":
- print(f"Rover got broken at ({rover_row}, {rover_col})")
- break
- elif current_direction == "up":
- rover_row -= 1
- if rover_row < 0:
- rover_col = size - 1
- if matrix[rover_row][rover_col] == "W":
- water += 1
- print(f"Water deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "M":
- metal += 1
- print(f"Metal deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "C":
- concrete += 1
- print(f"Concrete deposit found at ({rover_row}, {rover_col})")
- elif matrix[rover_row][rover_col] == "R":
- print(f"Rover got broken at ({rover_row}, {rover_col})")
- break
- if water >= 1 and metal >= 1 and concrete >= 1:
- print("Area suitable to start the colony.")
- else:
- print("Area not suitable to start the colony.")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement