def test_vs_brute_force_from_file(file_name): print 'Testing from file', file_name s = 0 with open(file_name, 'r') as f: f.next() # Skip n, k line for i, line in enumerate(f): _assert_results_equal(ru.to_int_list(line), s) print 'OK'
def three_sum_of_file(file_name, s): # Generates the 3SUM solutions for each array in the input file 'file_name'. with open(file_name, 'r') as f: f.next() # Skip n, k header line. for line in f: # Map 0-based to 1-based indices. Negative index (not found) are unchanged. result = three_sum_indices_fast(ru.to_int_list(line), s) yield (NOT_FOUND, ) if result is None else map(lambda x: x if x < 0 else x + 1, result)
def two_sum_of_file(file_name, s): with open(file_name, 'r') as f: f.next() # Skip n, k line for line in f: # Map 0-based to 1-based indices. Negative index (not found) are unchanged. yield map(lambda x: x if x < 0 else x + 1, two_sum_indices(ru.to_int_list(line), s))