此类需要使用之前讲到的IntegerWeightRandom类;
示例:
给定"a"赋予权重100,"b"的权重为50,则getNextString()时取"a"的概率要更大;
package org.xiazdong.util; /*给定字符串集合,设定每个字符串的权重,返回随机字符串*/ public class StringRandom { private String[] datas; private IntegerWeightRandom random = new IntegerWeightRandom(); public StringRandom(String[]datas){ this.datas = datas; } public void setStringWeight(int weight,int idx){ if(datas.length>idx) random.addWeightNumber(weight, idx); } public String getNextString(){ int idx = random.getNextInt(); return datas[idx]; } }
测试类:
package test.org.xiazdong.util; import org.junit.Test; import org.xiazdong.util.StringRandom; import junit.framework.TestCase; public class StringRandomTest extends TestCase { @Test public void testGetNextString() { StringRandom random = new StringRandom(new String[] { "a", "b", "c" }); random.setStringWeight(10, 0); random.setStringWeight(20, 1); random.setStringWeight(30, 2); for (int i = 0; i < 10; i++) System.out.println(random.getNextString()); } }