Skip to content

Accounts Merge

Jump to bottom
Linda Zhou edited this page Oct 16, 2022 · 21 revisions

🔗 Leetcode Link: https://leetcode.com/problems/accounts-merge

⏰ Time to complete: 17 mins

  1. U-nderstand

  1. M-atch

    For graph problems, some things we want to consider are:

    How is this a graph problem?

    We can apply a graph data structure where we build a map that maps an email to a list of accounts, which can be used to track which email is linked to which account. Emails can be represented as nodes, and an edge between nodes will signify that they belong to the same person. Then we can add an edge between the two connected components, effectively merging them into one connected component. This is essentially our graph. For example:

    # emails_accounts_map of email to account ID
{
  "[email protected]": [0, 2],
  "[email protected]": [0],
  "[email protected]": [1],
  "[email protected]": [2],
  "[email protected]": [3]
}

    We can use a DFS on each account in accounts list and look up the emails accounts map to tell us which accounts are linked to that particular account via common emails. This will make sure we visit each account only once. This is a recursive process and we should collect all the emails that we encounter along the way. Lastly, it will allow us to sort the collected emails and add it to final results.

  2. P-lan

    General Description of plan (1-2 sentences)

    Build a graph with an adjacency list of emails. Every email should have an edge to the connected email (including itself). From this, we can maintain a list of emails to account name list. Next, do a DFS for the unique email (using a hashset 'visited') to fill the emails for the given account name. Then, we can add the account name to the email address. Add the resultant account to end result.

  3. I-mplement

    class Solution {
    public List<List<String>> accountsMerge(List<List<String>> accounts) {
        Map<String, List<Integer>> names = new HashMap<>(); // map email to names using indexes
        for (int i = 0; i < accounts.size(); i++) {
            List<String> data = accounts.get(i);
            for (int j = 1; j < data.size(); j++) {
                String email = data.get(j);
                List<Integer> list = names.get(email);
                if (list == null) {
                    list = new ArrayList<Integer>();
                    names.put(email, list);
                }
                list.add(i);
            }
        }
        boolean[] visited = new boolean[accounts.size()];
        List<List<String>> res = new LinkedList<>();
        for (int i = 0; i < accounts.size(); i++) {
            Set<String> set = new TreeSet<String>();
            dfs(i, accounts, names, visited, set);
            if (!set.isEmpty()) {
                List<String> list = new LinkedList<String>(set);
                list.add(0, accounts.get(i).get(0));
                res.add(list);
            }
        }
        return res;
    }
    private void dfs(int cur, List<List<String>> accounts, Map<String, List<Integer>> names, 
                     boolean[] visited, Set<String> set) {
        if (visited[cur]) {
            return;
        }
        visited[cur] = true;
        for (int i = 1; i < accounts.get(cur).size(); i++) {
            String email = accounts.get(cur).get(i);
            set.add(email);
            for (int index : names.get(email)) {
                dfs(index, accounts, names, visited, set);
            }
        }
    }
}
    class Solution(object):
    def accountsMerge(self, accounts):
        from collections import defaultdict
        visited_accounts = [False] * len(accounts)
        emails_accounts_map = defaultdict(list)
        res = []
      
        for i, account in enumerate(accounts):
            for j in range(1, len(account)):
                email = account[j]
                emails_accounts_map[email].append(i)
      
        def dfs(i, emails):
            if visited_accounts[i]:
                return
            visited_accounts[i] = True
            for j in range(1, len(accounts[i])):
                email = accounts[i][j]
                emails.add(email)
                for neighbor in emails_accounts_map[email]:
                    dfs(neighbor, emails)
        
        for i, account in enumerate(accounts):
            if visited_accounts[i]:
                continue
            name, emails = account[0], set()
            dfs(i, emails)
            res.append([name] + sorted(emails))
        return res

  4. R-eview

    Verify the code works for the happy and edge cases you created in the “Understand” section

  5. E-valuate

    • Time Complexity: O(V + E), where V is the number accounts (can contain duplicates) and E is the number of accounts (without any duplicates)
    • Space Complexity: O(V + E), where V is the number accounts (can contain duplicates) and E is the number of accounts (without any duplicates)
Clone this wiki locally