コード例 #1
0
ファイル: rosgraph.py プロジェクト: orenlivne/euler
def read_edgelist(edgelist, create_using=None):
    '''Convert edge list format into a networkx graph object.'''
    g = nx.convert._prep_create_using(create_using)
    i = iter(edgelist)
    n, m = ro.to_int_list(i.next())
    g.add_nodes_from(xrange(1, n + 1))
    g.add_edges_from((int(x[0]), int(x[1])) for x in (x.split() for x in i))
    if g.number_of_edges() != m: raise ValueError('Problem reading edge list: wrong number of edges')
    return g
コード例 #2
0
ファイル: rosalind_bins.py プロジェクト: orenlivne/euler
def bins(f):
    '''Main driver to solve this problem.'''
    lines = ro.read_lines(f)
    a, k = ro.to_int_list(lines[2]), ro.to_int_list(lines[3])
    for v in k: print '%s ' % (to_one_based(bin_search(a, v))),
    print ''