Java权重分配的实现算法

来源:互联网 发布:淘宝模板上传宝贝图 编辑:程序博客网 时间:2024/05/18 00:48

现有一需求,就是假设有若干任务执行者执行一定数目的任务,并且任务的分配需按一定的权重比来进行。任务的分配是随机的,分配完毕后需要为每条任务打上执行者的标签(即被谁执行。)分配算法不难,但是用java写起来还是蛮巧妙的。
算法设计:

1)总的任务数能被权重和整除,则每个人分配的数量就是:总任务数/权重和*权重

2)总的任务数不能被权重和整除,则先按整除的数按1)的方式分,然后余数再优先分给权重较小的执行者。

//总的任务List<TaskDTO> dataList = pm1.getDataList();int total = dataList.size();//待分配人员List<ExecutorDTO> autoList = pmAuto.getDataList();//总权重int totalWight = 0;for(int i= 0;i<autoList.size();i++){      ExecutorDTO dto = autoList.get(i);      int weight = dto.getWeight();      totalWight += weight;}//获得能整除的最大数Integer maxInt = getMaxInt(totalWight, total);int rest = total - maxInt;//要保存的结果集List<TaskDTO> resultList = new ArrayList<TaskDTO>();for(int i=0;i<autoList.size();i++){    int assignNum = maxInt/totalWight*(autoList.get(i).getWeight());    int count = 0;    while(count < assignNum){        if(dataList.size()>0){            Random random = new Random();            //根据对话记录总数产生随机数            int index =random.nextInt((int)dataList.size());            TaskDTO dto = dataList.get(index);            if(null != dto){                                                                  dto.setAssignOperator(autoList.get(i).getId());                 resultList.add(dto);                 count ++;                dataList.remove(index);//已分配的就从待分配的任务数中拿掉            }         }else{            break;        }}    autoList.get(i).setTask(autoList.get(i).getTask() + count);//更新每个任务执行者的任务数    }    System.out.println("dataList大小:" + dataList.size() + ", 余数为:" + rest);    if(dataList.size() > 0) {        //按权重从小到大分余数~        Collections.sort(autoList, new Comparator<ExecutorDTO>() {            @Override            public int compare(ExecutorDTO o1,                            ExecutorDTO o2) {                return o1.getWeight() - o2.getWeight();                    }            });            List<TaskDTO> restResultList = this.assignRest(autoList, dataList);                resultList.addAll(restResultList);    }

其中分配余数assignRest的方法为:

private List<TaskDTO> assignRest(List<ExecutorDTO> autoList, List<TaskDTO> dataList) {        List<TaskDTO> resultList = new ArrayList<TaskDTO>();        Map<String, Integer> map = new HashMap();        outter:        for(int i=0; i<autoList.size(); i++) {            ExecutorDTO dto = autoList.get(i);            int weight = dto.getWeight();            Random random = new Random();            for(int j=0; j<weight; j++) {                int index =random.nextInt((int)dataList.size());                TaskDTO chatDTO = dataList.get(index);                if(null != chatDTO){                    chatDTO.setAssignOperator(dto.getOperPk());//分配人的id                    dto.setTask(dto.getTask() + 1);//更新任务执行者所包含的任务数                    resultList.add(chatDTO);                    dataList.remove(index);                    if(dataList.size() == 0)                         break outter;                }            }        }        return resultList;    }

至此,按权重分配的一种java算法就写完了。感觉算法不难,写起来还是有点小困难的,哈哈。

0 0
原创粉丝点击