JAVA-数量进行平均分配

要求：

有一定数量，对一批用户进行平均分配，每人一次分配不得超过20。

分配方案：

1. 如果平均不够用户分，则循环给每人指定一个，直到分完为止；

2. 如果平均超过20，则每人直接20；

3. 如果平均(0,20)，除了将平均值分给每个人，如果存在剩余，则循环直接将每个人进行20补全。

package com.algorithm.demo;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author create by ***
 * @version v1.0
 * @date 2022/7/7 15:30
 */
public class AverageDistributionDemo1 {

    /**
     * 将一定量进行用户平均分配
     */
    public static void main(String[] args) {
        List<Integer> userList = Lists.newArrayList(1, 2, 3);
        int count = 10;
        Map<Integer, Integer> userAverage = calculateAverage(userList, count);
        System.out.println("平均分配结果：" + userAverage);
    }

    /**
     * 计算平均分配
     *
     * @param userList 用户
     * @param count    数量
     * @return 分配结果
     */
    private static Map<Integer, Integer> calculateAverage(List<Integer> userList, int count) {
        int assignNum = count / userList.size();
        // 不够均分，每人最多一个，分完为止
        if (assignNum == 0) {
            Map<Integer, Integer> result = Maps.newConcurrentMap();
            for (Integer userId : userList) {
                result.put(userId, 1);
                if (--count <= 0) {
                    return result;
                }
            }
        }
        // 均分超过20个，以20个为准
        if (assignNum >= 20)
            return userList.stream().collect(Collectors.toMap(Integer::intValue, v -> 20));
        // 均分不超过20个，余下的将每人最多进行20补全，直至将剩余补完为止
        int diff = 20 - assignNum;
        int remainder = count % userList.size();
        Map<Integer, Integer> userAssignNumMap = Maps.newConcurrentMap();
        for (Integer userId : userList) {
            userAssignNumMap.put(userId, assignNum + Math.min(remainder, diff));
            remainder = Math.max(remainder - diff, 0);
        }
        return userAssignNumMap;
    }
}