网易2017春招笔试--双核cpu任务分配问题

一种双核CPU的两个核能够同时的处理任务，现在有n个已知数据量的任务需要交给CPU处理，假设已知CPU的每个核1秒可以处理1kb，每个核同时只能处理一项任务。n个任务可以按照任意顺序放入CPU进行处理，现在需要设计一个方案让CPU处理完这批任务所需的时间最少，求这个最小的时间。 

输入描述:

输入包括两行：
第一行为整数n(1 ≤ n ≤ 50)
第二行为n个整数length[i](1024 ≤ length[i] ≤ 4194304)，表示每个任务的长度为length[i]kb，每个数均为1024的倍数。

输出描述:

输出一个整数，表示最少需要处理的时间

输入例子:

5
3072 3072 7168 3072 1024

输出例子:

9216

上代码：

def distribute(ls):
    ls.sort()
    sub_ls1 = []
    sub_ls2 = []
    sub_ls1.append(ls.pop())
    sub_ls2.append(ls.pop())
    while len(ls):
        if sum(sub_ls1) < sum(sub_ls2):
            sub_ls1.append(ls.pop())
        else:
            sub_ls2.append(ls.pop())
    return sub_ls1, sub_ls2

def compute_time(sub1, sub2):
    #time_spend = sum(sub1) > sum(sub2) ? sum(sub1) : sum(sub2)
    '''
    if sum(sub1) > sum(sub2):
        time_spend = sum(sub1)
    else:
        time_spend = sum(sub2)
    '''
    time_spend = sum(sub1) > sum(sub2) and sum(sub1) or sum(sub2)
    return time_spend

if __name__ == "__main__":
    n = int(raw_input("please input n:"))
    ls_list = [int(i) for i in raw_input("please input list of number: ").strip().split()]
    sub1, sub2 = distribute(ls_list)
    t = compute_time(sub1, sub2)
    print(t)