任意个整数尽量平均分配到多个组

group_number.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import pprint


class Node:

    def __init__(self, id, count):
        self.id = id
        self.count = count

    def __repr__(self):
        return f"({self.id}, {self.count})"


class NodeGroup(list):

    def __init__(self):
        self.value = 0

    def append(self, node):
        self.value += node.count
        return super().append(node)

    def __repr__(self):
        return "value: {}, {}".format(self.value, super().__repr__())


def group(data, amount):
    data.sort(key=lambda node: node.count, reverse=True)
    # init result group
    result = [NodeGroup() for i in range(amount)]

    for node in data:
        # find a most match position to insert this node
        result[minimum(result)].append(node)
    return result


def minimum(ldata):
    '''return the minimum data index of ldata'''
    if not ldata:
        raise Exception("param is empty")

    result = 0
    for i, node in enumerate(ldata):
        if ldata[result].value > node.value:
            result = i
    return result


if __name__ == "__main__":
    data = [Node(n, n) for n in [28, 25, 19, 18, 10, 9, 6, 4, 3, 1]]
    result = group(data, 3)
    pprint.pprint(result)