按字典序排列最小的等效字符串

/*
题目思路：
1. 使用并查集（Union-Find）算法来管理字符之间的等价关系。
2. 初始化每个字符自己和自己等价。
3. 遍历 s1 和 s2，将对应字符进行合并。
4. 查找每个字符的最小等价字符，并用其替换 baseStr 中的字符。
5. 返回替换后的字符串。
*/

import java.util.stream.IntStream;

public class Solution {
    private int find(int[] parent, int index) {
        if (parent[index] != index) {
            parent[index] = find(parent, parent[index]);
        }
        return parent[index];
    }

    private void union(int[] parent, int index1, int index2) {
        int root1 = find(parent, index1);
        int root2 = find(parent, index2);
        if (root1 != root2) {
            if (root1 < root2) {
                parent[root2] = root1;
            } else {
                parent[root1] = root2;
            }
        }
    }

    public String smallestEquivalentString(String s1, String s2, String baseStr) {
        int[] parent = IntStream.range(0, 26).toArray();

        IntStream.range(0, s1.length()).forEach(i -> union(parent, s1.charAt(i) - 'a', s2.charAt(i) - 'a'));

        return baseStr.chars()
            .mapToObj(c -> String.valueOf((char) (find(parent, c - 'a') + 'a')))
            .collect(Collectors.joining());
    }
}