(LeetCode 每日一题)2901. 最长相邻不相等子序列 II (动态规划dp)

题目：2901. 最长相邻不相等子序列 II

思路：动态规划dp，时间复杂度0(n^2)，细节看注释。

C++版本：

class Solution {
public:
	// 判断两个字符串的汉明距离是否为1
    bool check(string &a,string &b){
        if(a.size()!=b.size()) return false;
        int ct=0;
        for(int i=0;i<a.size();i++){
            if(a[i]!=b[i]) ct++;
        }
        return ct==1;
    }
    vector<string> getWordsInLongestSubsequence(vector<string>& words, vector<int>& groups) {
        int n=words.size();
        // f[i]表示以word[i]结尾的子序列，最大的长度
        // ft[i]表示以word[i]结尾的子序列，word[i]前一个元素下标
        vector<int> f(n),ft(n,-1);
        // 维护子序列最大长度的最后一个元素下标
        int mx_id=0;
        for(int i=0;i<n;i++){
            f[i]=1;
            for(int j=i-1;j>=0;j--){
                if(f[i]<f[j]+1 && groups[j]!=groups[i]&&check(words[i],words[j])){
                    f[i]=f[j]+1;
                    ft[i]=j;
                }
            }
            if(f[i]>f[mx_id]){
                mx_id=i;
            }
        }
        // 最长子序列的长度
        int k=f[mx_id];
        // 答案
        vector<string> v(k);
        for(int i=mx_id;i!=-1;i=ft[i]){
            v[k-1]=words[i];
            k--;
        }
        return v;
    }
};

JAVA版本：

class Solution {
    boolean check(String a,String b){
        if(a.length()!=b.length()) return false;
        int ct=0;
        for(int i=0;i<a.length();i++){
            if(a.charAt(i)!=b.charAt(i)) ct++;
        }
        return ct==1;
    }  
    public List<String> getWordsInLongestSubsequence(String[] words, int[] groups) {
        int n=words.length;
        int[] f=new int[n];
        int[] ft=new int[n];
        int mx_id=0;
        for(int i=0;i<n;i++){
            f[i]=1;
            ft[i]=-1;
            for(int j=i-1;j>=0;j--){
                if(f[i]<f[j]+1 && groups[j]!=groups[i]&&check(words[i],words[j])){
                    f[i]=f[j]+1;
                    ft[i]=j;
                }
            }
            if(f[i]>f[mx_id]){
                mx_id=i;
            }
        }
        int k=f[mx_id];
        List<String> v = new ArrayList<>();
        for(int i=mx_id;i!=-1;i=ft[i]){
            v.add(words[i]);
        }
        Collections.reverse(v);
        return v;
    }
}

Go版本：

func getWordsInLongestSubsequence(words []string, groups []int) []string {
    n:=len(words)
    f:=make([]int,n)
    ft:=make([]int,n)
    mx_id:=0
    for i:=0;i<n;i++ {
        f[i]=1
        ft[i]=-1
        for j:=i-1;j>=0;j-- {
            if f[i]<f[j]+1 && groups[i]!=groups[j] && check(words[i],words[j]) {
                f[i]=f[j]+1
                ft[i]=j
            }
        }
        if f[i]>f[mx_id] {
            mx_id=i
        }
    }
    k:=f[mx_id]
    v:=make([]string,k)
    for i:=mx_id;i!=-1;i=ft[i] {
        v[k-1]=words[i]
        k--
    }
    return v
}
func check(a,b string) bool{
    if len(a)!=len(b) {
        return false
    }
    ct:=0
    for i:=0;i<len(a);i++ {
        if a[i]!=b[i] {
            ct++
        }
    }
    if ct==1 {
        return true
    }
    return false
}