from median_of_stream import MedianOfStream def main(input_1, input_2): result = [] median_num = MedianOfStream() result.append(None) for i in range(1, len(input_1)): if input_1[i] == "insert_num": median_num.insert_num(input_2[i][0]) result.append(None) elif input_1[i] == "find_median": res = median_num.find_median() result.append(res) return result