| comments | difficulty | edit_url | tags | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
true |
Medium |
|
Given a list of accounts where each element accounts[i] is a list of strings, where the first element accounts[i][0] is a name, and the rest of the elements are emails representing emails of the account.
Now, we would like to merge these accounts. Two accounts definitely belong to the same person if there is some common email to both accounts. Note that even if two accounts have the same name, they may belong to different people as people could have the same name. A person can have any number of accounts initially, but all of their accounts definitely have the same name.
After merging the accounts, return the accounts in the following format: the first element of each account is the name, and the rest of the elements are emails in sorted order. The accounts themselves can be returned in any order.
Example 1:
Input: accounts = [["John","[email protected]","[email protected]"],["John","[email protected]","[email protected]"],["Mary","[email protected]"],["John","[email protected]"]] Output: [["John","[email protected]","[email protected]","[email protected]"],["Mary","[email protected]"],["John","[email protected]"]] Explanation: The first and second John's are the same person as they have the common email "[email protected]". The third John and Mary are different people as none of their email addresses are used by other accounts. We could return these lists in any order, for example the answer [['Mary', '[email protected]'], ['John', '[email protected]'], ['John', '[email protected]', '[email protected]', '[email protected]']] would still be accepted.
Example 2:
Input: accounts = [["Gabe","[email protected]","[email protected]","[email protected]"],["Kevin","[email protected]","[email protected]","[email protected]"],["Ethan","[email protected]","[email protected]","[email protected]"],["Hanzo","[email protected]","[email protected]","[email protected]"],["Fern","[email protected]","[email protected]","[email protected]"]] Output: [["Ethan","[email protected]","[email protected]","[email protected]"],["Gabe","[email protected]","[email protected]","[email protected]"],["Hanzo","[email protected]","[email protected]","[email protected]"],["Kevin","[email protected]","[email protected]","[email protected]"],["Fern","[email protected]","[email protected]","[email protected]"]]
Constraints:
1 <= accounts.length <= 10002 <= accounts[i].length <= 101 <= accounts[i][j].length <= 30accounts[i][0]consists of English letters.accounts[i][j] (for j > 0)is a valid email.
Based on the problem description, we can use a union-find data structure to merge accounts with the same email address. The specific steps are as follows:
First, we iterate through all the accounts. For the $i$th account, we iterate through all its email addresses. If an email address appears in the hash table
Next, we iterate through all the accounts again. For the $i$th account, we use the union-find to find its root node, and then add all the email addresses of that account to the hash table
The time complexity is
class UnionFind:
def __init__(self, n):
self.p = list(range(n))
self.size = [1] * n
def find(self, x):
if self.p[x] != x:
self.p[x] = self.find(self.p[x])
return self.p[x]
def union(self, a, b):
pa, pb = self.find(a), self.find(b)
if pa == pb:
return False
if self.size[pa] > self.size[pb]:
self.p[pb] = pa
self.size[pa] += self.size[pb]
else:
self.p[pa] = pb
self.size[pb] += self.size[pa]
return True
class Solution:
def accountsMerge(self, accounts: List[List[str]]) -> List[List[str]]:
uf = UnionFind(len(accounts))
d = {}
for i, (_, *emails) in enumerate(accounts):
for email in emails:
if email in d:
uf.union(i, d[email])
else:
d[email] = i
g = defaultdict(set)
for i, (_, *emails) in enumerate(accounts):
root = uf.find(i)
g[root].update(emails)
return [[accounts[root][0]] + sorted(emails) for root, emails in g.items()]class UnionFind {
private final int[] p;
private final int[] size;
public UnionFind(int n) {
p = new int[n];
size = new int[n];
for (int i = 0; i < n; ++i) {
p[i] = i;
size[i] = 1;
}
}
public int find(int x) {
if (p[x] != x) {
p[x] = find(p[x]);
}
return p[x];
}
public boolean union(int a, int b) {
int pa = find(a), pb = find(b);
if (pa == pb) {
return false;
}
if (size[pa] > size[pb]) {
p[pb] = pa;
size[pa] += size[pb];
} else {
p[pa] = pb;
size[pb] += size[pa];
}
return true;
}
}
class Solution {
public List<List<String>> accountsMerge(List<List<String>> accounts) {
int n = accounts.size();
UnionFind uf = new UnionFind(n);
Map<String, Integer> d = new HashMap<>();
for (int i = 0; i < n; ++i) {
for (int j = 1; j < accounts.get(i).size(); ++j) {
String email = accounts.get(i).get(j);
if (d.containsKey(email)) {
uf.union(i, d.get(email));
} else {
d.put(email, i);
}
}
}
Map<Integer, Set<String>> g = new HashMap<>();
for (int i = 0; i < n; ++i) {
int root = uf.find(i);
g.computeIfAbsent(root, k -> new HashSet<>())
.addAll(accounts.get(i).subList(1, accounts.get(i).size()));
}
List<List<String>> ans = new ArrayList<>();
for (var e : g.entrySet()) {
List<String> emails = new ArrayList<>(e.getValue());
Collections.sort(emails);
ans.add(new ArrayList<>());
ans.get(ans.size() - 1).add(accounts.get(e.getKey()).get(0));
ans.get(ans.size() - 1).addAll(emails);
}
return ans;
}
}class UnionFind {
public:
UnionFind(int n) {
p = vector<int>(n);
size = vector<int>(n, 1);
iota(p.begin(), p.end(), 0);
}
bool unite(int a, int b) {
int pa = find(a), pb = find(b);
if (pa == pb) {
return false;
}
if (size[pa] > size[pb]) {
p[pb] = pa;
size[pa] += size[pb];
} else {
p[pa] = pb;
size[pb] += size[pa];
}
return true;
}
int find(int x) {
if (p[x] != x) {
p[x] = find(p[x]);
}
return p[x];
}
private:
vector<int> p, size;
};
class Solution {
public:
vector<vector<string>> accountsMerge(vector<vector<string>>& accounts) {
int n = accounts.size();
UnionFind uf(n);
unordered_map<string, int> d;
for (int i = 0; i < n; ++i) {
for (int j = 1; j < accounts[i].size(); ++j) {
const string& email = accounts[i][j];
if (d.find(email) != d.end()) {
uf.unite(i, d[email]);
} else {
d[email] = i;
}
}
}
unordered_map<int, set<string>> g;
for (int i = 0; i < n; ++i) {
int root = uf.find(i);
g[root].insert(accounts[i].begin() + 1, accounts[i].end());
}
vector<vector<string>> ans;
for (const auto& [root, s] : g) {
vector<string> emails(s.begin(), s.end());
emails.insert(emails.begin(), accounts[root][0]);
ans.push_back(emails);
}
return ans;
}
};type unionFind struct {
p, size []int
}
func newUnionFind(n int) *unionFind {
p := make([]int, n)
size := make([]int, n)
for i := range p {
p[i] = i
size[i] = 1
}
return &unionFind{p, size}
}
func (uf *unionFind) find(x int) int {
if uf.p[x] != x {
uf.p[x] = uf.find(uf.p[x])
}
return uf.p[x]
}
func (uf *unionFind) union(a, b int) bool {
pa, pb := uf.find(a), uf.find(b)
if pa == pb {
return false
}
if uf.size[pa] > uf.size[pb] {
uf.p[pb] = pa
uf.size[pa] += uf.size[pb]
} else {
uf.p[pa] = pb
uf.size[pb] += uf.size[pa]
}
return true
}
func accountsMerge(accounts [][]string) (ans [][]string) {
n := len(accounts)
uf := newUnionFind(n)
d := make(map[string]int)
for i := 0; i < n; i++ {
for _, email := range accounts[i][1:] {
if j, ok := d[email]; ok {
uf.union(i, j)
} else {
d[email] = i
}
}
}
g := make(map[int]map[string]struct{})
for i := 0; i < n; i++ {
root := uf.find(i)
if _, ok := g[root]; !ok {
g[root] = make(map[string]struct{})
}
for _, email := range accounts[i][1:] {
g[root][email] = struct{}{}
}
}
for root, s := range g {
emails := []string{}
for email := range s {
emails = append(emails, email)
}
sort.Strings(emails)
account := append([]string{accounts[root][0]}, emails...)
ans = append(ans, account)
}
return
}class UnionFind {
private p: number[];
private size: number[];
constructor(n: number) {
this.p = new Array(n);
this.size = new Array(n);
for (let i = 0; i < n; ++i) {
this.p[i] = i;
this.size[i] = 1;
}
}
find(x: number): number {
if (this.p[x] !== x) {
this.p[x] = this.find(this.p[x]);
}
return this.p[x];
}
union(a: number, b: number): boolean {
let pa = this.find(a),
pb = this.find(b);
if (pa === pb) {
return false;
}
if (this.size[pa] > this.size[pb]) {
this.p[pb] = pa;
this.size[pa] += this.size[pb];
} else {
this.p[pa] = pb;
this.size[pb] += this.size[pa];
}
return true;
}
}
function accountsMerge(accounts: string[][]): string[][] {
const n = accounts.length;
const uf = new UnionFind(n);
const d = new Map<string, number>();
for (let i = 0; i < n; ++i) {
for (let j = 1; j < accounts[i].length; ++j) {
const email = accounts[i][j];
if (d.has(email)) {
uf.union(i, d.get(email)!);
} else {
d.set(email, i);
}
}
}
const g = new Map<number, Set<string>>();
for (let i = 0; i < n; ++i) {
const root = uf.find(i);
if (!g.has(root)) {
g.set(root, new Set<string>());
}
const emailSet = g.get(root)!;
for (let j = 1; j < accounts[i].length; ++j) {
emailSet.add(accounts[i][j]);
}
}
const ans: string[][] = [];
for (const [root, emails] of g.entries()) {
const emailList = Array.from(emails).sort();
const mergedAccount = [accounts[root][0], ...emailList];
ans.push(mergedAccount);
}
return ans;
}