此类需要使用之前讲到的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());
}
}