pisya

def get_result(rocket_pos, rocket_speed):
    k = {i: {'pos': rocket_pos[i], 'speed': rocket_speed[i]} for i in range(len(rocket_pos))}
    res = len(k.keys())
    while True:
        for i in k.keys():
            for j in k.keys():
                if i == j:
                    continue
                if k[i]['pos'] >= 1000 or k[j]['pos'] >= 1000:
                    continue
                if k[i]['pos'] < 0 or k[j]['pos'] < 0:
                    continue
                if k[i]['pos'] == k[j]['pos']:
                    k[i]['speed'] += k[j]['speed']
                    k[j]['pos'] = -1
                    res -= 1
        ctr = 0
        for i in k.keys():
            if k[i]['pos'] >= 0:
                k[i]['pos'] += k[i]['speed']
            if k[i]['pos'] > 1000 or k[i]['pos'] < 0:
                ctr += 1
        if ctr == len(k.keys()):
            break
    return res


if __name__ == '__main__':
    rocket_pos_ = [3, 10, 5]
    rocket_speed_ = [1, 2, 3]

    print(get_result(rocket_pos_, rocket_speed_))