首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 Javascript

js数组实现权重概率分配

Posted in Javascript onSeptember 12, 2017

今天写了一个js控制页面轮播的功能，如果仅仅使用队列很简单，但是考虑到为每一个页面分配权重的是否变的异常复杂，使用switch和if else也无法解决，于是想到使用js数组实现，思路是将各个轮播的页面抽象成一个对象，各个对象需要手动指定权重值，然后组成一个数组，使用下面封装的函数，将会根据各个对象相应的权重概率返回一个对象，代码如下：

/**
* js数组实现权重概率分配
* @param  Array  arr    js数组，参数类型[Object,Object,Object……]
* @return  Array        返回一个随机元素，概率为其percent/所有percent之和，参数类型Object
* @author  shuiguang
*/
function weight_rand(arr){
  //参数arr元素必须含有percent属性，参考如下所示
  /*
  var arr = [{
      name : '1',
      percent : 1
    }, {
      name : '2',
      percent : 2
    }, {
      name : '3',
      percent : 1
    }, {
      name : '4',
      percent : 2
    }
  ];
  */
  var total = 0;
  var i, j, percent;
  //下标标记数组，按照上面的例子，单倍情况下其组成为[1,2,2,3,4,4]
  var index = new Array();
  for (i = 0; i < arr.length; i++) {
    //判断元素的权重，为了实现小数权重，先将所有的值放大100倍
    percent = 'undefined' != typeof(arr[i].percent) ? parseInt(arr[i].percent*100) : 0;
    for (j = 0; j < percent; j++) {
      index.push(i);
    }
    total += percent;
  }
  //随机数值，其值介于0-5的整数
  var rand = Math.floor(Math.random() * total);
  return arr[index[rand]];
}

上面的方法虽然可行，可是遇到这样一个问题：对于一般复杂的分配情况如1:1:1分配（相对值）可以满足，如果遇到15%,25%,35%剩余等精确权重分配（绝对值）无法满足。因为去计算15%:25%:35%:剩余的比例很是麻烦，于是我将上面的函数继续修改，添加了百分比模式，比如上面的例子，分配了上面明确的百分数之后，剩余的百分比将给最后一个元素，而不用计算最后一个元素占的百分数，也不用计算各个元素的比例。代码如下：

/**
* js数组实现权重概率分配，支持数字比模式(支持2位小数)和百分比模式(不支持小数，最后一个元素多退少补)
* @param  Array  arr  js数组，参数类型[Object,Object,Object……]
* @return  Array      返回一个随机元素，概率为其weight/所有weight之和，参数类型Object
* @author  shuiguang
*/
function weight_rand(arr){
	//参数arr元素必须含有weight属性，参考如下所示
	//var arr=[{name:'1',weight:1.5},{name:'2',weight:2.5},{name:'3',weight:3.5}];
	//var arr=[{name:'1',weight:'15%'},{name:'2',weight:'25%'},{name:'3',weight:'35%'}];
	//求出最大公约数以计算缩小倍数，perMode为百分比模式
	var per;
	var maxNum = 0;
	var perMode = false;
	//自定义Math求最小公约数方法
	Math.gcd = function(a,b){
		var min = Math.min(a,b);
		var max = Math.max(a,b);
		var result = 1;
		if(a === 0 || b===0){
			return max;
		}
		for(var i=min; i>=1; i--){
			if(min % i === 0 && max % i === 0){
				result = i;
				break;
			}
		}
		return result;
	};
	
	//使用clone元素对象拷贝仍然会造成浪费，但是使用权重数组对应关系更省内存
	var weight_arr = new Array();
	for (i = 0; i < arr.length; i++) {
		if('undefined' != typeof(arr[i].weight))
		{
			if(arr[i].weight.toString().indexOf('%') !== -1) {
				per = Math.floor(arr[i].weight.toString().replace('%',''));
				perMode = true;
			}else{
				per = Math.floor(arr[i].weight*100);
			}
		}else{
			per = 0;
		}
		weight_arr[i] = per;
		maxNum = Math.gcd(maxNum, per);
	}
	//数字比模式，3:5:7，其组成[0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2]
	//百分比模式，元素所占百分比为15%，25%，35%
	var index = new Array();
	var total = 0;
	var len = 0;
	if(perMode){
		for (i = 0; i < arr.length; i++) {
			//len表示存储arr下标的数据块长度，已优化至最小整数形式减小索引数组的长度
			len = weight_arr[i];
			for (j = 0; j < len; j++){
				//超过100%跳出，后面的舍弃
				if(total >= 100){
					break;
				}
				index.push(i);
				total++;
			}
		}
		//使用最后一个元素补齐100%
		while(total < 100){
			index.push(arr.length-1);
			total++;
		}
	}else{
		for (i = 0; i < arr.length; i++) {
			//len表示存储arr下标的数据块长度，已优化至最小整数形式减小索引数组的长度
			len = weight_arr[i]/maxNum;
			for (j = 0; j < len; j++){
				index.push(i);
			}
			total += len;
		}
	}
	//随机数值，其值为0-11的整数，数据块根据权重分块
	var rand = Math.floor(Math.random()*total);
	//console.log(index);
	return arr[index[rand]];
}

var arr=[{name:'1',weight:1.5},{name:'2',weight:2.5},{name:'3',weight:3.5}];
console.log(weight_rand(arr));
var arr=[{name:'1',weight:'15%'},{name:'2',weight:'25%'},{name:'3',weight:'35%'}];
console.log(weight_rand(arr));
var prize_arr = [
	{'id':1, 'prize':'平板电脑', 'weight':1},
	{'id':2, 'prize':'数码相机', 'weight':2},
	{'id':3, 'prize':'音箱设备', 'weight':10},
	{'id':4, 'prize':'4G优盘', 'weight':12},
	{'id':5, 'prize':'10Q币', 'weight':22},
	{'id':6, 'prize':'下次没准就能中哦', 'weight':50}    
];

var times = 100000;
var prize;
var pingban = 0;
var shuma = 0;
var yinxiang = 0;
var youpan = 0;
var qb = 0;
var xc = 0;
var start = new Date().getTime();

for($i=0; $i<times; $i++){
	prize = weight_rand(prize_arr);
	if(prize.prize == '平板电脑')
	{
		pingban++;
	}else if(prize.prize == '数码相机'){
		shuma++;
	}else if(prize.prize == '音箱设备'){
		yinxiang++;
	}else if(prize.prize == '4G优盘'){
		youpan++;
	}else if(prize.prize == '10Q币'){
		qb++;
	}else if(prize.prize == '下次没准就能中哦'){
		xc++;
	}
}

var stop = new Date().getTime();
console.log('平板电脑：'+pingban/times+', 数码相机：'+shuma/times+', 音箱设备：'+yinxiang/times+', 4G优盘：'+youpan/times+', 10Q币：'+qb/times+', 下次没准就能中哦：'+xc/times);
console.log('耗费时间：'+(stop-start)/1000+'秒');

该代码已经通过最大公约数对下标数组进行优化，使用数字比模式已经优化到最小数值比例，百分比模式考虑性能消耗暂不支持2位小数。

写完js版，于是很轻松改为php版本，经过10万次循环测试，发现for循环比foreach省时间，而非网上传的foreach比for更快。但是总体来说，js的执行速度是php的20倍左右，php的执行时间约6秒，js的执行时间约为0.346秒。

/**
* php数组实现权重概率分配，支持数字比模式(支持2位小数)和百分比模式(不支持小数，最后一个元素多退少补)
* @param  array  $arr  php数组，参数类型array(array(),array(),array()……)
* @return  array      返回一个随机元素，概率为其percent/所有percent之和，参数类型array()
* @author  shuiguang
*/
function weight_rand($arr)
{
  //参数arr元素必须含有percent属性，参考如下所示
  //$arr=array(array('name'=>'1','weight'=>1.5),array('name'=>'2','weight'=>1.5),array('name'=>'3','weight'=>1.5));
  //$arr=array(array('name'=>'1','weight'=>'15%'),array('name'=>'2','weight'=>'25%'),array('name'=>'3','weight'=>'35%'));
  //求出最大公约数以计算缩小倍数，perMode为百分比模式
  $perMode = false;
  $maxNum = 0;
  //自定义求最小公约数方法
  $gcd = function($a, $b)
  {
    $min = min($a, $b);
    $max = max($a, $b);
    $result = 1;
    if($a === 0 || $b === 0)
    {
      return $max;
    }
    for($i=$min; $i>=1; $i--)
    {
      if($min % $i === 0 && $max % $i === 0)
      {
        $result = $i;
        break;
      }
    }
    return $result;
  };
  //使用传地址可能会影响后面的结果，但是使用权重数组对应关系更省内存
  $weight_arr = array();
  $arr_len = count($arr);
  for($i=0; $i<$arr_len; $i++)
  {
    if(isset($arr[$i]['weight']))
    {
      if(strpos($arr[$i]['weight'], '%') !== false)
      {
        $per = floor(str_replace('%', '', $arr[$i]['weight']));
        $perMode = true;
      }else{
        $per = floor($arr[$i]['weight']*100);
      }
    }else{
      $per = 0;
    }
    $weight_arr[$i] = $per;
    $maxNum = call_user_func($gcd, $maxNum, $per);
  }
  //数字比模式，3:5:7，其组成[0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2]
  //百分比模式，元素所占百分比为15%，25%，35%
  $index = array();
  $total = 0;
  if($perMode)
  {
    for($i=0; $i<$arr_len; $i++)
    {
      //$len表示存储$arr下标的数据块长度，已优化至最小整数形式减小索引数组的长度
      $len = $weight_arr[$i];
      for ($j = 0; $j < $len; $j++)
      {
        //超过100%跳出，后面的舍弃
        if($total >= 100)
        {
          break;
        }
        $index[] = $i;
        $total++;
      }
    }
    //使用最后一个元素补齐100%
    while($total < 100)
    {
      $index[] = $arr_len-1;
      $total++;
    }
  }else{
    for($i=0; $i<$arr_len; $i++)
    {
      //len表示存储arr下标的数据块长度，已优化至最小整数形式减小索引数组的长度
      $len = $weight_arr[$i]/$maxNum;
      for ($j = 0; $j < $len; $j++)
      {
        $index[] = $i;
      }
      $total += $len;
    }
  }
  //随机数值，其值为0-11的整数，数据块根据权重分块
  $rand = floor(mt_rand(0, $total));
	//修复php随机函数可以取临界值造成的bug
  $rand = $rand == $total ? $total-1 : $rand;
  return $arr[$index[$rand]];
}

$arr=array(array('name'=>'1','weight'=>1.5),array('name'=>'2','weight'=>1.5),array('name'=>'3','weight'=>1.5));
p(weight_rand($arr));
$arr=array(array('name'=>'1','weight'=>'15%'),array('name'=>'2','weight'=>'25%'),array('name'=>'3','weight'=>'35%'));
p(weight_rand($arr));

$prize_arr = array(
	'0' => array('id'=>1, 'prize'=>'平板电脑', 'weight'=>1),
	'1' => array('id'=>2, 'prize'=>'数码相机', 'weight'=>5),
	'2' => array('id'=>3, 'prize'=>'音箱设备', 'weight'=>10),
	'3' => array('id'=>4, 'prize'=>'4G优盘', 'weight'=>12),
	'4' => array('id'=>5, 'prize'=>'10Q币', 'weight'=>22),
	'5' => array('id'=>6, 'prize'=>'下次没准就能中哦', 'weight'=>50),
);

$start = time();
$result = array();
$times = 100000;
for($i=0; $i<$times; $i++)
{
	$row = weight_rand($prize_arr);
	if(array_key_exists($row['prize'], $result))
	{
		$result[$row['prize']] ++;
	}else{
		$result[$row['prize']] = 1;
	}
}
$cost = time() - $start;


p($result);
p('耗费时间：'.$cost.'秒');
function p($var)
{
  echo "<pre>";
  if($var === false)
  {
    echo 'false';
  }else if($var === ''){
    print_r("''");
  }else{
    print_r($var);
  }
  echo "</pre>";
}

php版本如果只是使用整数数字比模式，完全不用考虑数字的放大与求最小公倍数的算法，只需要做简单的累加即可，可以大大缩短执行时间。

js数组实现权重概率分配

- Author -

shuiguang

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Javascript 相关文章推荐
jquery判断单个复选框是否被选中的代码
Sep 03 Javascript
JavaScript 对象模型 执行模型
Oct 15 Javascript
js实现简单的左右两边固定广告效果实例
Apr 10 Javascript
JavaScript实现点击按钮直接打印
Jan 06 Javascript
JavaScript 浏览器兼容性总结及常用浏览器兼容性分析
Mar 30 Javascript
JS脚本实现动态给标签控件添加事件的方法
Jun 02 Javascript
jQuery实现为LI列表前3行设置样式的方法【2种方法】
Sep 04 Javascript
js仿新浪微博消息发布功能
Feb 17 Javascript
从零学习node.js之详解异步控制工具async（八）
Feb 27 Javascript
微信小程序调用PHP后台接口 解析纯html文本
Jun 13 Javascript
基于 Vue.js 2.0 酷炫自适应背景视频登录页面实现方式
Jan 17 Javascript
Vue实现搜索 和新闻列表功能简单范例
Mar 16 Javascript
js 概率计算(简单版)
Sep 12 #Javascript
JavaScript面向对象精要(下部)
Sep 12 #Javascript
在Vue.js中使用Mixins的方法
Sep 12 #Javascript
JavaScript面向对象精要(上部)
Sep 12 #Javascript
JS库之ParticlesJS使用简介
Sep 12 #Javascript
关于预加载InstantClick的问题解决方法
Sep 12 #Javascript
提升页面加载速度的插件InstantClick
Sep 12 #Javascript
区块链(2) 扫雷(1) Date(1) 路由(1) NumPy(2) router(1) Scheduled(1) 模拟登录(1) 单向链表(1) 双向链表(6)
You might like
php连接mssql的一些相关经验及注意事项
2013/02/05 PHP
php中session与cookie的比较
2015/01/27 PHP
Nodejs实现多人同时在线移动鼠标的小游戏分享
2014/12/06 NodeJs
Javascript中的apply()方法浅析
2015/03/15 Javascript
Javascript代码实现仿实例化类
2015/04/03 Javascript
充分发挥Node.js程序性能的一些方法介绍
2015/06/23 Javascript
基于Javascript实现倒计时功能
2016/02/22 Javascript
javascript中href和replace的比较(详解)
2016/11/25 Javascript
axios post提交formdata的实例
2018/03/16 Javascript
bootstrap 路径导航 分页 进度条的实例代码
2018/08/06 Javascript
Vue引用Swiper4插件无法重写分页器样式的解决方法
2018/09/27 Javascript
js Array.slice的8种不同用法示例
2019/07/10 Javascript
layui动态渲染生成左侧3级菜单的方法(根据后台返回数据)
2019/09/23 Javascript
解决nuxt 自定义全局方法,全局属性,全局变量的问题
2020/11/05 Javascript
[03:52]显微镜下的DOTA2第三期——英雄在无聊的时候干什么
2014/06/20 DOTA
浅谈python中copy和deepcopy中的区别
2017/10/23 Python
详解TensorFlow在windows上安装与简单示例
2018/03/05 Python
Python基于递归和非递归算法求两个数最大公约数、最小公倍数示例
2018/05/21 Python
Python后台开发Django的教程详解（启动）
2019/04/08 Python
Python发展简史 Python来历
2019/05/14 Python
PyCharm+Qt Designer+PyUIC安装配置教程详解
2019/06/13 Python
Python 字典中的所有方法及用法
2020/06/10 Python
解决keras加入lambda层时shape的问题
2020/06/11 Python
解决python 在for循环并且pop数组的时候会跳过某些元素的问题
2020/12/11 Python
欧洲顶级的童装奢侈品购物网站：Bambini Fashion（面向全球）
2018/04/24 全球购物
Lookfantastic西班牙官网：英国知名美妆购物网站
2018/06/13 全球购物
澳大利亚家具和家居用品购物网站：Zanui
2018/12/29 全球购物
英国123鲜花网站：123 Flowers
2019/07/07 全球购物
网络工程师的自我评价
2013/10/02 职场文书
大学毕业感言100字
2014/02/03 职场文书
道德模范先进事迹
2014/02/14 职场文书
班主任对学生的评语
2014/04/26 职场文书
秋季运动会广播稿（30篇）
2014/09/13 职场文书
小学教育见习报告
2014/10/31 职场文书
2015年世界无烟日活动方案
2015/05/04 职场文书
《山中访友》教学反思
2016/02/24 职场文书