优化 JavaScript 中的代码（当前使用太多 for 循环）

Question

我一直在研究一个数据格式问题，终于得到了答案。

但是，我的答案循环太多，在实现的时候会减慢时间复杂度。

下面是我的代码：

const userInfo = [];

for (let i = 0; i < projects.length; i++) {
  const userProject = projects[i].userProjects;

  for (let j = 0; j < userProject.length; j++) {
    const userExist = userInfo.some((item) =>
      item.userId.includes(userProject[j].user.userId)
    );
    if (!userExist) userInfo.push(userProject[j].user);
  }
}

userInfo.forEach((user) => {
  let tempArr = [];
  for (let i = 0; i < projects.length; i++) {
    const userProject = projects[i].userProjects;
    for (let j = 0; j < userProject.length; j++) {
      if (user.userId === userProject[j].userId)
        tempArr.push({
          projectId: projects[i].projectId,
          name: projects[i].name,
        });
    }
  }
  user.project = tempArr;
});

console.log(userInfo);

有什么方法可以优化我的代码吗？我试图将嵌套的 for 循环组合在一起，但没有工作......

请帮忙。非常感谢！！！

Answer 1

在顶部，外部两个循环看起来是必要的（因为您有 N 个项目 * M 个 userProjects），但第三个 .some 不是 - 将 userInfo 从数组更改为 userIds 到他们的映射用户，因此在第二个循环中，您只需查看 userId 是否已存在（O(1)），如果不存在则分配。

不要分配用户，而是在最后分配一个您需要的形状的对象 - [{ projectId, name }]。这将使您不必再次遍历整个数组 - 一旦您检查是否需要为 userId 创建周围数组，无论您是否必须将对象推送到数组是否在该迭代中创建数组。

const userProjectsByUserId = new Map();
for (const { userProjects, projectId, name } of projects) {
    for (const { user } of projects) {
        if (!userProjectsByUserId.has(user.userId)) {
            userProjectsByUserId.set(user.userId, { user, projects: [] }]);
        }
        userProjectsByUserId.get(user.userId).projects.push({ projectId, name });
    }
}

这会给你一个形状的地图

userId => {
  user,
  projects: [
    { projectId, name },
    ...

如果您需要当前代码提供给您的确切数据结构，那么之后，请执行以下操作：

const output = [...userProjectsByUserId.values()]
  .map(({ user, projects }) => ({ ...user, project: projects }));

（尽管您可能会考虑使用projects 而不是project，因为它是项目的集合，而不是单个项目）