java计算概率

rates 为中奖概率值   比如:0.3,0.5,0.2

package com.ddb.activity.base.util;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;

public class LotteryProbabilityUtil {
	
	public static void main(String[] args) {
		List<BigDecimal> rates=new ArrayList<BigDecimal>();
		rates.add(new BigDecimal("0.01"));
		rates.add(new BigDecimal("0.2"));
		//rates.add(new BigDecimal("0.3"));
		//rates.add(new BigDecimal("0.2"));
		rates.add(new BigDecimal("0.2"));
		rates.add(new BigDecimal("0.09"));
//		int j=0;
		for(int i=0;i<10;i++){
			int a=LotteryProbabilityUtil.getRandomIndex(rates);
			System.out.println(a);
			if(a==0){
//				System.out.println(++j);
			}
//			
		}
	}
	/**
	 * 计算概率
	 * @param rates
	 * @return 抽中的位置
	 */
	public static int getRandomIndex(List<BigDecimal> rates) {
		Random random = new Random();
		BigDecimal ranNum = new BigDecimal(random.nextDouble());
		BigDecimal sumRate = new BigDecimal("0.0");
		for(BigDecimal rate:rates) {
			sumRate = sumRate.add(rate);
		}
		BigDecimal minVal = new BigDecimal("0.0");
		BigDecimal maxVal = new BigDecimal("0.0");
		for (int i = 0; i < rates.size(); i++) {
			BigDecimal rangeVal = rates.get(i).divide(sumRate, BigDecimal.ROUND_HALF_EVEN);
			maxVal = minVal.add(rangeVal);
			boolean exp1 = false;
			if(i == 0) {
				exp1 = (minVal.compareTo(ranNum)<=0);
			} else {
				exp1 = (minVal.compareTo(ranNum)<0);
			}
			if(exp1&&(ranNum.compareTo(maxVal)<=0)) {
				return i;
			}
			minVal = minVal.add(rangeVal);
		}
		throw new RuntimeException("Get Random Index Error");
	}
}

猜你喜欢

转载自huweiyi.iteye.com/blog/2366874