给定一个列表 accounts,每个元素 accounts[i] 是一个字符串列表,其中第一个元素 accounts[i][0] 是 名称 (name),其余元素是 emails 表示该帐户的邮箱地址。
现在,我们想合并这些帐户。如果两个帐户都有一些共同的邮件地址,则两个帐户必定属于同一个人。请注意,即使两个帐户具有相同的名称,它们也可能属于不同的人,因为人们可能具有相同的名称。一个人最初可以拥有任意数量的帐户,但其所有帐户都具有相同的名称。
合并帐户后,按以下格式返回帐户:每个帐户的第一个元素是名称,其余元素是按顺序排列的邮箱地址。accounts 本身可以以任意顺序返回。
例子 1:
Input:
accounts = [["John", "johnsmith@mail.com", "john00@mail.com"], ["John", "johnnybravo@mail.com"], ["John", "johnsmith@mail.com", "john_newyork@mail.com"], ["Mary", "mary@mail.com"]]
Output: [["John", 'john00@mail.com', 'john_newyork@mail.com', 'johnsmith@mail.com'], ["John", "johnnybravo@mail.com"], ["Mary", "mary@mail.com"]]
Explanation:
第一个和第三个 John 是同一个人,因为他们有共同的电子邮件 "johnsmith@mail.com"。
第二个 John 和 Mary 是不同的人,因为他们的电子邮件地址没有被其他帐户使用。
我们可以以任何顺序返回这些列表,例如答案[['Mary','mary@mail.com'],['John','johnnybravo@mail.com'],
['John','john00@mail.com','john_newyork@mail.com','johnsmith@mail.com']]仍然会被接受。
注意:
accounts的长度将在[1,1000]的范围内。
accounts[i]的长度将在[1,10]的范围内。
accounts[i][j]的长度将在[1,30]的范围内。
思路:DFS或者并查集都可以,我采用并查集解决的这道题,耗时有点长,简单来说就是将具有相同邮件的人之间通过并查集把它们归到一个集合里即可。
class Solution {
private int[] f;
public List<List<String>> accountsMerge(List<List<String>> accounts) {
int len=0;
f=new int[accounts.size()];
List<String> res=new ArrayList<>();
int[] flag=new int[accounts.size()];
List<List<String>> ans=new ArrayList<>();
Map<String,Integer> map1=new HashMap<>();
for(int i=0;i<accounts.size();i++) {
f[i]=i;
for(int j=1;j<accounts.get(i).size();j++) {
if(map1.containsKey(accounts.get(i).get(j))) {
int t1=find(f[i]);
int t2=find(map1.get(accounts.get(i).get(j)));
if(t1!=t2) f[t1]=t2;
}
}
for(int j=1;j<accounts.get(i).size();j++)
map1.put(accounts.get(i).get(j), f[i]);
}
for(int i=0;i<accounts.size();i++) flag[i]=-1;
for(int i=0;i<accounts.size();i++) {
int index=flag[find(f[i])];
if(index==-1) {
res.clear();
for(int j=0;j<accounts.get(i).size();j++) {
if(res.contains(accounts.get(i).get(j)))
continue;
res.add(accounts.get(i).get(j));
}
ans.add(new ArrayList<>(res));
flag[find(f[i])]=len++;
}
else {
for(int j=1;j<accounts.get(i).size();j++) {
if(ans.get(index).contains(accounts.get(i).get(j)))
continue;
ans.get(index).add(accounts.get(i).get(j));
}
}
}
for(int i=0;i<ans.size();i++)
Collections.sort(ans.get(i));
return ans;
}
private int find(int x) {
if(f[x]==x)
return x;
return f[x]=find(f[x]);
}
}