Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from collections import deque
- def mix_fireworks(mix_sum, fireworks, power):
- firework_type = ''
- if mix_sum % 3 == 0 and mix_sum % 5 != 0:
- firework_type = 'palm_fireworks'
- elif mix_sum % 3 != 0 and mix_sum % 5 == 0:
- firework_type = 'willow_fireworks'
- elif mix_sum % 3 == 0 and mix_sum % 5 == 0:
- firework_type = 'crossette_fireworks'
- return firework_type
- firework_effects = deque(int(x) for x in input().split(', '))
- explosive_power = [int(x) for x in input().split(', ')]
- palm_fireworks = 0
- willow_fireworks = 0
- crossette_fireworks = 0
- success = False
- while True:
- current_effect = firework_effects[0]
- current_power = explosive_power[-1]
- if current_effect <= 0:
- firework_effects.popleft()
- if not firework_effects:
- break
- current_effect = firework_effects[0]
- elif current_power <= 0 and len(explosive_power) > 1:
- explosive_power.pop()
- if not explosive_power:
- break
- current_power = explosive_power[-1]
- if current_effect > 0 and current_power > 0:
- attempt = current_effect + current_power
- result = mix_fireworks(attempt, firework_effects, explosive_power)
- if result == 'palm_fireworks':
- palm_fireworks += 1
- elif result == 'willow_fireworks':
- willow_fireworks += 1
- elif result == 'crossette_fireworks':
- crossette_fireworks += 1
- else:
- while True:
- current_effect -= 1
- firework_effects.popleft()
- if not firework_effects:
- break
- firework_effects.append(current_effect)
- current_effect = firework_effects[0]
- attempt = current_effect + current_power
- result = mix_fireworks(attempt, firework_effects, explosive_power)
- if result == 'palm_fireworks':
- palm_fireworks += 1
- break
- elif result == 'willow_fireworks':
- willow_fireworks += 1
- break
- elif result == 'crossette_fireworks':
- crossette_fireworks += 1
- break
- firework_effects.popleft()
- explosive_power.pop()
- if not firework_effects or not explosive_power:
- if palm_fireworks >= 3 and willow_fireworks >= 3 and crossette_fireworks >= 3:
- success = True
- break
- if palm_fireworks >= 3 and willow_fireworks >= 3 and crossette_fireworks >= 3:
- success = True
- break
- if success:
- print('Congrats! You made the perfect firework show!')
- else:
- print("Sorry. You can't make the perfect firework show.")
- if firework_effects:
- print(f"Firework Effects left: {', '.join(str(firework) for firework in firework_effects)}")
- if explosive_power:
- print(f"Explosive Power left: {', '.join(str(power) for power in explosive_power)}")
- print(f'Palm Fireworks: {palm_fireworks}')
- print(f'Willow Fireworks: {willow_fireworks}')
- print(f'Crossette Fireworks: {crossette_fireworks}')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement