实例创建时增量选色的算法与数据结构优化方案咨询

需求明确 #

创建实例时需从[Red, Green, Blue, Purple]列表中按规则分配颜色：

• 每次分配上一次分配的下一个颜色；
• 实例删除后对应颜色释放，但下次仍按原顺序选色，直到释放的颜色被重新分配；
• 最多存在4个实例，且无重复颜色。

UI交互：界面设有Create按钮，创建实例后展示含对应颜色的实例及delete按钮，后台维护颜色分配逻辑。

现有伪代码的不足 #

当前实现需要遍历实例列表查找maxColor，再循环递增检查可用颜色，在实例数量接近上限时可能多次遍历，效率偏低且逻辑繁琐。

优化方案 #

通过维护三个核心元素即可简化逻辑并提升效率：

1. 固定顺序的颜色数组：const colors = ['Red', 'Green', 'Blue', 'Purple']；
2. 已使用颜色索引的集合：快速判断颜色是否被占用，支持O(1)的增删查操作；
3. 下一个待分配的索引指针：记录下一次尝试分配的起始位置，避免从头遍历。

具体逻辑 #

• 初始化：
  从数据库加载已存在的实例，将对应颜色的索引存入usedIndices集合；
  计算初始nextIndex：若有已使用索引，取最大值加1后对4取模；若无则设为0。

• 创建实例：
  从nextIndex开始循环检查：

  • 若当前索引未被占用，分配对应颜色，将索引加入集合，更新nextIndex为(当前索引 + 1) % 4；
  • 若已被占用，将nextIndex自增取模后继续检查，直到找到可用索引（最多循环4次）。

• 删除实例：
  直接从usedIndices集合中移除该实例对应的颜色索引即可，无需修改nextIndex——确保后续分配仍按原顺序进行，直到遇到释放的索引时重新分配。

示例代码（JavaScript） #

const colors = ['Red', 'Green', 'Blue', 'Purple'];
const usedIndices = new Set();
let nextIndex = 0;

// 从数据库加载已有实例初始化
function init(existingInstances) {
  existingInstances.forEach(instance => {
    const idx = colors.indexOf(instance.color);
    if (idx !== -1) usedIndices.add(idx);
  });
  if (usedIndices.size > 0) {
    const maxIdx = Math.max(...usedIndices);
    nextIndex = (maxIdx + 1) % colors.length;
  } else {
    nextIndex = 0;
  }
}

// 创建实例
function createInstance() {
  if (usedIndices.size >= colors.length) {
    console.log('已达最大实例数量');
    return null;
  }
  let currentIdx = nextIndex;
  while (usedIndices.has(currentIdx)) {
    currentIdx = (currentIdx + 1) % colors.length;
  }
  const assignedColor = colors[currentIdx];
  usedIndices.add(currentIdx);
  nextIndex = (currentIdx + 1) % colors.length;
  return { color: assignedColor, index: currentIdx };
}

// 删除实例
function deleteInstance(instanceIndex) {
  usedIndices.delete(instanceIndex);
}

方案优势 #

• 效率更高：分配操作最多循环4次，删除操作是O(1)的集合操作；
• 逻辑清晰：通过指针和集合直接维护状态，避免了冗余的遍历逻辑；
• 严格贴合需求：完全遵循“按原顺序分配、释放后仍按顺序直到重新分配”的规则。

内容的提问来源于stack exchange，提问作者user21357723