LeetCode-in-Java

2901. Longest Unequal Adjacent Groups Subsequence II

Medium

You are given an integer n, a 0-indexed string array words, and a 0-indexed array groups, both arrays having length n.

The hamming distance between two strings of equal length is the number of positions at which the corresponding characters are different.

You need to select the longest subsequence from an array of indices [0, 1, ..., n - 1], such that for the subsequence denoted as [i0, i1, ..., ik - 1] having length k, the following holds:

Return a string array containing the words corresponding to the indices (in order) in the selected subsequence. If there are multiple answers, return any of them.

A subsequence of an array is a new array that is formed from the original array by deleting some (possibly none) of the elements without disturbing the relative positions of the remaining elements.

Note: strings in words may be unequal in length.

Example 1:

Input: n = 3, words = [“bab”,”dab”,”cab”], groups = [1,2,2]

Output: [“bab”,”cab”]

Explanation: A subsequence that can be selected is [0,2].

So, a valid answer is [words[0],words[2]] = [“bab”,”cab”]. Another subsequence that can be selected is [0,1].

So, another valid answer is [words[0],words[1]] = [“bab”,”dab”]. It can be shown that the length of the longest subsequence of indices that satisfies the conditions is 2.

Example 2:

Input: n = 4, words = [“a”,”b”,”c”,”d”], groups = [1,2,3,4]

Output: [“a”,”b”,”c”,”d”]

Explanation: We can select the subsequence [0,1,2,3]. It satisfies both conditions. Hence, the answer is [words[0],words[1],words[2],words[3]] = [“a”,”b”,”c”,”d”]. It has the longest length among all subsequences of indices that satisfy the conditions. Hence, it is the only answer.

Constraints:

Solution

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class Solution {
    public List<String> getWordsInLongestSubsequence(int n, String[] words, int[] groups) {
        int[] check = new int[groups.length];
        int[] before = new int[groups.length];
        Arrays.fill(check, 1);
        Arrays.fill(before, -1);
        int index = 0;
        int max = 1;
        for (int i = 1; i < n; i++) {
            for (int j = i - 1; j >= 0; j--) {
                if (groups[i] != groups[j] && ham(words[i], words[j]) && check[j] + 1 > check[i]) {
                    check[i] = check[j] + 1;
                    before[i] = j;
                    if (check[i] > max) {
                        max = check[i];
                        index = i;
                    }
                }
            }
        }
        List<String> ans = new ArrayList<>();
        while (index >= 0) {
            ans.add(words[index]);
            index = before[index];
        }
        Collections.reverse(ans);
        return ans;
    }

    private boolean ham(String s1, String s2) {
        if (s1.length() != s2.length()) {
            return false;
        }
        int count = 0;
        for (int i = 0; i < s1.length(); i++) {
            if (s1.charAt(i) != s2.charAt(i)) {
                count++;
            }
            if (count > 1) {
                return false;
            }
        }
        return count == 1;
    }
}