2021/01/18 每日一题 合并问题

LeetCode上合并问题,记录下解题思路

又是并查集,我现在看到这种类型的题目只会想到并查集了.....

可以拆分为以下2个问题:

  1. 怎么确认2个信息属于同一个人
  2. 确认完同一人之后如何合并信息

使用并查集确认同一人,需要判断两个用户的信息中间有一个相同,就可以确定为同一人,并且把相同人的父节点属性都设为一样
用map来保存每个相同账户的所有信息
之后再通过set来去重,不过要注意,这个根据题意需要排序输出所以还要加一道排序

// 定义一个节点
class user {
   constructor (i) {
     // 将自己的父亲设为自己 
     this.parent = this
     // val属性记录下当前节点在accounts中的位置
     this.val = i
   }
}
// 定义查找函数,不断查找父节点,直到父节点等于本身
const find = function (x) {
   while(x !== x.parent) {
      x = x.parent
   }
   return x 
}
var accountsMerge = function(accounts) {
 // 保存账户数量长度
 let len = accounts.length
 // 用一个map保存同一用户的所有信息
 let map = new Map()
 // 建立对应长度的节点数组
 let users = Array(len).fill(0).map((x,i) => new user(i))
 // 返回结果
 let res = []
 // 遍历整个accounts数组,确立好每个用户之间的关系
 for(let i = 0; i<len-1; i++) {
    // 双重循环遍历,每一个都要和后面的对比多次
    for(let j = i+1; j<len; j++) {
      // accounts[i]和accounts[j],名字不对比,后面的数据有一个相同就可以确认是同一账号
      // 用到数组的some方法,some() 方法用于检测数组中的元素是否满足指定条件
      var flag = accounts[i].some((item,k) => {
         // 排除第一位名字的可能
         // includes()是包含这个元素的话就返回true
         return k!==0 && accounts[j].includes(item)
      })
      if(flag) {
        // 如果标志位为true,那么就是包含相同的邮箱号
        // 就要把j的父亲设为i
        find(users[j]).parent = find(users[i])
      }
    }
 }
 // 然后遍历users数组,将父元素为同一节点的所有数据放在一起
 for(let i =0;i<users.length;i++) {
   // 查找到当前节点的父元素
   let root = find(users[i])
   if(map.has(root)) {
      // 如果map中已经有这个节点的数据的话
      map.get(root).push(accounts[i])
   } else {
      // 如果没有这个节点数据的话
      map.set(root,[])
      map.get(root).push(accounts[i])
   }
 }
 // map中就是的每一项就是确认相同的
 // 计算整合map就能得到结果,用set来去重
 map.forEach(item => {
   // 用flat来数据扁平化,sort进行排序,set来去重
   let u = [...new Set(item.flat().sort())]
   res.push(u)
 })
 return res
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容