CasualMath - Weed Infestation

weed_infestation.py

rules = {2: 5, 7: 1}
weeds = [10]

def pick(w, pick, grow):
    if w == pick: return 0
    return w - pick + grow

step = 0
while 0 not in weeds:
    print(f"{step}:{weeds}")
    new_weeds = []
    for r in rules:
        for w in weeds:
            if w >= r:
                value = pick(w, r, rules[r])
                #print(f"{w} -> pick {r}, grow {rules[r]} -> {value}")
                new_weeds.append(value)
                
    # convert to set and back to remove duplicates.
    # if a number of weeds is arrived at by multiple courses of action, then it doesn't need to be considered more than once.
    weeds = list(set(new_weeds))
    step += 1
    if step > 10: break