feat: md-table weighted random

This commit is contained in:
2026-03-21 18:42:50 +08:00
parent 0261061bcb
commit 66e8564a34
3 changed files with 106 additions and 3 deletions
+88
View File
@@ -0,0 +1,88 @@
/**
* 解析整数范围字符串
* @param label 标签字符串,可能是单个整数(如 "1")或整数范围(如 "1-3"
* @returns 如果成功解析返回 [min, max],否则返回 null
*/
export function parseIntegerRange(label: string): [number, number] | null {
const trimmed = label.trim();
// 尝试匹配整数范围格式 (如 "1-3")
const rangeMatch = trimmed.match(/^(\d+)\s*-\s*(\d+)$/);
if (rangeMatch) {
const min = parseInt(rangeMatch[1]!, 10);
const max = parseInt(rangeMatch[2]!, 10);
if (min <= max) {
return [min, max];
}
}
// 尝试匹配单个整数格式 (如 "5")
const intMatch = trimmed.match(/^(\d+)$/);
if (intMatch) {
const num = parseInt(intMatch[1]!, 10);
return [num, num];
}
return null;
}
/**
* 计算加权随机的总权重
* @param labels 标签数组
* @returns 总权重值
*/
export function calculateTotalWeight(labels: string[]): number {
let total = 0;
for (const label of labels) {
const range = parseIntegerRange(label);
if (range) {
const [min, max] = range;
total += max - min + 1;
}
}
return total;
}
/**
* 根据权重随机选择一个索引
* @param labels 标签数组
* @param totalWeight 总权重(可选,会自行计算)
* @returns 选中的索引,如果没有有效权重则返回 -1
*/
export function weightedRandomIndex(labels: string[], totalWeight?: number): number {
const weight = totalWeight ?? calculateTotalWeight(labels);
if (weight === 0) {
return -1;
}
// 生成一个随机权重值 (1 到 totalWeight)
const roll = Math.floor(Math.random() * weight) + 1;
let cumulative = 0;
for (let i = 0; i < labels.length; i++) {
const range = parseIntegerRange(labels[i]!);
if (range) {
const [min, max] = range;
const labelWeight = max - min + 1;
cumulative += labelWeight;
if (roll <= cumulative) {
return i;
}
}
}
// 理论上不会到这里,但为了安全返回最后一个
return labels.length - 1;
}
/**
* 检查所有 label 是否都是整数或整数范围格式
* @param labels 标签数组
* @returns 如果所有 label 都是整数/整数范围格式返回 true
*/
export function areAllLabelsNumeric(labels: string[]): boolean {
if (labels.length === 0) return false;
return labels.every(label => parseIntegerRange(label) !== null);
}