[0118]ADD:LC-721

Author: Coder-AndyLee

readme.md

@@ -6,6 +6,7 @@
 [[20210108]189. 旋转数组-Medium](数组/189.%20旋转数组-Medium.md)  
 [[20210113]1202. 交换字符串中的元素-Medium](并查集/1202.%20交换字符串中的元素-Medium.md)  
 [[20210114]1018. 可被 5 整除的二进制前缀-Easy](数组/1018.%20可被%205%20整除的二进制前缀-Easy.md)   
+[[20210118]721. 账户合并-Medium](并查集/721.%20账户合并-Medium.md)  
 ## 2020.12
 [[20201201]34. 在排序数组中查找元素的第一个和最后一个位置-Medium](数组/34.%20在排序数组中查找元素的第一个和最后一个位置-Medium.md)  
 [[20201202]292. Nim 游戏-Easy](动态规划/292.%20Nim%20游戏-Easy.md)  
@@ -103,6 +104,7 @@
 ## 并查集
 ### 并查集详解, [link](https://leetcode-cn.com/problems/number-of-provinces/solution/python-duo-tu-xiang-jie-bing-cha-ji-by-m-vjdr/)
 [684. 冗余连接-Medium](并查集/684.%20冗余连接-Medium.md)  
+[721. 账户合并-Medium](并查集/721.%20账户合并-Medium.md)  
 [990. 等式方程的可满足性-Medium](并查集/990.%20等式方程的可满足性-Medium.md)  
 [1202. 交换字符串中的元素-Medium](并查集/1202.%20交换字符串中的元素-Medium.md)  
 

并查集/721. 账户合并-Medium.md

@@ -0,0 +1,82 @@
+# [Description](https://leetcode-cn.com/problems/accounts-merge)
+给定一个列表 accounts，每个元素 accounts[i] 是一个字符串列表，其中第一个元素 accounts[i][0] 是 名称 (name)，其余元素是 emails 表示该账户的邮箱地址。
+
+现在，我们想合并这些账户。如果两个账户都有一些共同的邮箱地址，则两个账户必定属于同一个人。请注意，即使两个账户具有相同的名称，它们也可能属于不同的人，因为人们可能具有相同的名称。一个人最初可以拥有任意数量的账户，但其所有账户都具有相同的名称。
+
+合并账户后，按以下格式返回账户：每个账户的第一个元素是名称，其余元素是按字符 ASCII 顺序排列的邮箱地址。账户本身可以以任意顺序返回。
+
+ 
+
+示例 1：
+```python
+输入：
+accounts = [["John", "[email protected]", "[email protected]"], ["John", "[email protected]"], ["John", "[email protected]", "[email protected]"], ["Mary", "[email protected]"]]
+输出：
+[["John", '[email protected]', '[email protected]', '[email protected]'],  ["John", "[email protected]"], ["Mary", "[email protected]"]]
+解释：
+第一个和第三个 John 是同一个人，因为他们有共同的邮箱地址 "[email protected]"。 
+第二个 John 和 Mary 是不同的人，因为他们的邮箱地址没有被其他帐户使用。
+可以以任何顺序返回这些列表，例如答案 [['Mary'，'[email protected]']，['John'，'[email protected]']，
+['John'，'[email protected]'，'[email protected]'，'[email protected]']] 也是正确的。
+```
+
+提示：
+
+- accounts的长度将在[1，1000]的范围内。
+- accounts[i]的长度将在[1，10]的范围内。
+- accounts[i][j]的长度将在[1，30]的范围内。
+
+
+# Solution
+- 本题的不同之处在于，**并查集存储的是每条account的index之间的father关系**
+```python
+class UnionFind:
+    def __init__(self, length):
+        self.father = [i for i in range(length)]
+        self.mail2index = {}
+        
+    def find(self, x):
+        root = x
+        
+        while self.father[root] != root:
+            root = self.father[root]
+        
+        # 路径压缩
+        while x != root:
+            original_father = self.father[x]
+            self.father[x] = root
+            x = original_father
+        
+        return root
+    
+    def merge(self, x, y):
+        root_x, root_y = self.find(x), self.find(y)
+        
+        if root_x != root_y:
+            self.father[root_x] = root_y
+
+class Solution:
+    def accountsMerge(self, accounts: List[List[str]]) -> List[List[str]]:
+        uf = UnionFind(len(accounts))
+        index2name = {}
+        sets = {}
+
+        for i, c in enumerate(accounts):
+            index2name[i] = c[0]
+            for j in range(1, len(c)):
+                mail = c[j]
+                if mail in uf.mail2index:
+                    uf.merge(uf.mail2index[mail], i)
+                uf.mail2index[mail] = i
+				
+        for l, i in enumerate(uf.father):
+            index = uf.find(i)
+            if index not in sets: sets[index] = set()  # 其实这个的set可以使用defaultdict
+            sets[index].update(accounts[l])
+
+        res = []
+        for k, v in sets.items():
+            res.append(sorted(list(v)))
+        return res
+
+```