魔法师四种元素平衡(Python)

题目描述

《XX魔法书》中记载，只有魔法师体内地火水风四种元素平衡的时候才是绝对安全的。某魔法师在给自己做了一个体检，发现自己体内的四种元素数量分别是A，B，C，D，魔法师慌了，这时他看到魔法书中记载了一种转换魔法，可以把任意两个单位的元素转换成任意一种元素。

输入描述

输入仅包含一行，仅包含4个正整数A,B,C,D，分别表示四种元素的数量。（1<=A,B,C,D<=10^17）

输出描述

输出仅包含一个整数，如题所示

样例输入
1 2 2 4
样例输出
8
提示：只要把第四种元素中的两个转换为第一种元素即平衡。

def method(nums):
    sumnum = sum(nums)
    avg = sumnum // 4
    sub = 0
    for i in range(avg, -1, -1):
        sub = 0
        for item in nums:
            if item > i:
                sub += (item - i)

        count = 0
        for item in nums:
            if i>item:
                count += (i - item)*2
        if sub == count:
            return i
    return -1


if __name__ == '__main__':
    nums = list(map(int, input().strip().split()))
    nums.sort()
    res = method(nums)
    if res == -1:
        print(res)
    else:
        print(res*4)