Added solution for ThemePark. The solution only works for the small input