句子相似性 II

// Problem: Sentence Similarity II
// Given two sentences and a list of similar word pairs, determine if the two sentences are similar.
// Sentences are represented as arrays of strings, and similar word pairs are given as a 2D array.
// Two sentences are similar if the words in one sentence can be replaced with similar words to get the other sentence.

import java.util.*;
import java.util.stream.Collectors;

public class SentenceSimilarityIIStream {
    private Map<String, String> parent;

    public SentenceSimilarityIIStream() {
        this.parent = new HashMap<>();
    }

    private String find(String word) {
        parent.putIfAbsent(word, word);
        if (!parent.get(word).equals(word)) {
            parent.put(word, find(parent.get(word)));
        }
        return parent.get(word);
    }

    private void union(String word1, String word2) {
        String root1 = find(word1);
        String root2 = find(word2);
        if (!root1.equals(root2)) {
            parent.put(root1, root2);
        }
    }

    public boolean areSentencesSimilarTwo(String[] sentence1, String[] sentence2, List<List<String>> similarPairs) {
        if (sentence1.length != sentence2.length) {
            return false;
        }

        similarPairs.forEach(pair -> union(pair.get(0), pair.get(1)));

        return IntStream.range(0, sentence1.length)
                .allMatch(i -> find(sentence1[i]).equals(find(sentence2[i])));
    }

    public static void main(String[] args) {
        SentenceSimilarityIIStream similarityChecker = new SentenceSimilarityIIStream();
        String[] sentence1 = {"great", "acting", "skills"};
        String[] sentence2 = {"fine", "drama", "talent"};
        List<List<String>> similarPairs = Arrays.asList(
                Arrays.asList("great", "good"),
                Arrays.asList("fine", "good"),
                Arrays.asList("drama", "acting"),
                Arrays.asList("skills", "talent")
        );
        System.out.println(similarityChecker.areSentencesSimilarTwo(sentence1, sentence2, similarPairs)); // true
    }
}