示例#1
0
def test1():
    n = 5
    k = 3
    h = [10, 30, 40, 50, 20]
    cost = B.calc_cost(n, k, h)
    assert cost[n - 1] == 30
示例#2
0
def test3():
    n = 2
    k = 100
    h = [10, 10]
    cost = B.calc_cost(n, k, h)
    assert cost[n - 1] == 0
示例#3
0
def test4():
    n = 10
    k = 4
    h = [40, 10, 20, 70, 80, 10, 20, 70, 80, 60]
    cost = B.calc_cost(n, k, h)
    assert cost[n - 1] == 40
示例#4
0
def test2():
    n = 3
    k = 1
    h = [10, 20, 10]
    cost = B.calc_cost(n, k, h)
    assert cost[n - 1] == 20