前阵子app里需要实现抢红包的功能,后台需要个红包分配的算法,于是搜了下找到了实现的主要思想,参考文章http://blog.csdn.net/forever_cvank/article/details/54693000感谢博主的分享。
基本思想就是根据当前剩余金额和人数得到平均值,然后随机,为了防止最后的人分得最大金额需要调整平均值的倍数,这样就会导致后面红包不够分,于是就处理成如果不够分则按照正常的平均值进行随机,所以这个算法的弱点就是越到后面越抢不到大金额的红包,下面上代码
public class TestRedPackage {
//红包最小值
private static final BigDecimal MINVALUE = new BigDecimal("0.01");
//红包最大值
private static final BigDecimal MAXVALUE = new BigDecimal("200");
//这里为了避免某一个红包占用大量资金,我们需要设定非最后一个红包的最大金额,我们把他设置为红包金额平均值的N倍
private static final BigDecimal TIMES = new BigDecimal("2.1");
//小数点位数
private static final int MAX_DIGIT = 2;
private static void test(BigDecimal money, int people) {
List<BigDecimal> result = spiltRedPackets(money, people);
}
/**
* 分红包
*
* @param money
* @param count
* @return
*/
public List<BigDecimal> spiltRedPackets(BigDecimal money, int count) {
//首先判断红包是否合情理
if (!isRight(money, count)) {
return null;
}
List<BigDecimal> list = new ArrayList<>();
BigDecimal average = money.divide(new BigDecimal(count), MAX_DIGIT, BigDecimal.ROUND_UP);
BigDecimal max = TIMES.multiply(average).setScale(MAX_DIGIT, BigDecimal.ROUND_UP);
max = max.compareTo(money) > 0 ? money : max;
for (int i = 0; i < count; i++) {
BigDecimal value = randomRedPacket(money, MINVALUE, max, count - i);
list.add(value);
money = money.subtract(value);
}
return list;
}
/**
* 分红包核心算法
*
* @param money
* @param minS
* @param maxS
* @param count
* @return
*/
public BigDecimal randomRedPacket(BigDecimal money, BigDecimal minS, BigDecimal maxS, int count) {
//当人数剩余一个时,把当前剩余全部返回
if (count == 1) {
return money;
}
//如果当前最小红包等于最大红包,之间返回当前红包
if (minS.compareTo(maxS) == 0) {
return minS;
}
BigDecimal max = maxS.compareTo(money) > 0 ? money : maxS;
//随机产生一个红包
BigDecimal range = max.subtract(minS);
BigDecimal randomNumner = new BigDecimal(Math.random()).setScale(MAX_DIGIT, BigDecimal.ROUND_HALF_UP);
BigDecimal randomResult = range.multiply(randomNumner).setScale(MAX_DIGIT, BigDecimal.ROUND_UP);
BigDecimal one = randomResult.add(minS);
BigDecimal balance = money.subtract(one);
//判断此次分配后,后续是否合理
if (isRight(balance, count - 1)) {
return one;
} else {
//重新分配
BigDecimal avg = balance.divide(new BigDecimal(count - 1), 10, BigDecimal.ROUND_UP);
//本次红包过大,导致下次的红包过小;如果红包过大,下次就随机一个小值到本次红包金额的一个红包
if (avg.compareTo(MINVALUE) < 0) {
//递归调用,修改红包最大金额
BigDecimal average = money.divide(new BigDecimal(count), MAX_DIGIT, BigDecimal.ROUND_UP);
return randomRedPacket(money, minS, average, count);
} else {
//递归调用,修改红包最小金额
return randomRedPacket(money, one, maxS, count);
}
}
}
/**
* 判断红包是否合情理
*
* @param money
* @param count
* @return
*/
public boolean isRight(BigDecimal money, int count) {
if (count <= 0 || money == null || money.compareTo(BigDecimal.ZERO) <= 0) {
return false;
}
BigDecimal avg = money.divide(new BigDecimal(count), 10, BigDecimal.ROUND_UP); //防止舍入造成误差
if (avg.compareTo(MINVALUE) < 0) {
return false;
} else if (avg.compareTo(MAXVALUE) > 0) {
return false;
}
return true;
}
}