leetcode 2900. 最长相邻不相等子序列 I 简单
给你一个下标从 0 开始的字符串数组 words
,和一个下标从 0 开始的 二进制 数组 groups
,两个数组长度都是 n
。
你需要从 words
中选出 最长子序列。如果对于序列中的任何两个连续串,二进制数组 groups
中它们的对应元素不同,则 words
的子序列是不同的。
正式来说,你需要从下标 [0, 1, ..., n - 1]
中选出一个 最长子序列 ,将这个子序列记作长度为 k
的 [i0, i1, ..., ik - 1]
,对于所有满足 0 <= j < k - 1
的 j
都有 groups[ij] != groups[ij + 1]
。
请你返回一个字符串数组,它是下标子序列 依次 对应 words
数组中的字符串连接形成的字符串数组。如果有多个答案,返回 任意 一个。
注意:words
中的元素是不同的 。
示例 1:
输入:words = ["e","a","b"], groups = [0,0,1] 输出:["e","b"] 解释:一个可行的子序列是 [0,2] ,因为 groups[0] != groups[2] 。 所以一个可行的答案是 [words[0],words[2]] = ["e","b"] 。 另一个可行的子序列是 [1,2] ,因为 groups[1] != groups[2] 。 得到答案为 [words[1],words[2]] = ["a","b"] 。 这也是一个可行的答案。 符合题意的最长子序列的长度为 2 。
示例 2:
输入:words = ["a","b","c","d"], groups = [1,0,1,1] 输出:["a","b","c"] 解释:一个可行的子序列为 [0,1,2] 因为 groups[0] != groups[1] 且 groups[1] != groups[2] 。 所以一个可行的答案是 [words[0],words[1],words[2]] = ["a","b","c"] 。 另一个可行的子序列为 [0,1,3] 因为 groups[0] != groups[1] 且 groups[1] != groups[3] 。 得到答案为 [words[0],words[1],words[3]] = ["a","b","d"] 。 这也是一个可行的答案。 符合题意的最长子序列的长度为 3 。
提示:
1 <= n == words.length == groups.length <= 100
1 <= words[i].length <= 10
groups[i]
是0
或1
。words
中的字符串 互不相同 。words[i]
只包含小写英文字母。
分析:最长子序列的选择实际上只与 groups 数组有关,问题转化为,在 groups 数组中,最长的相邻不同的 01 串是什么,也就是最长子序列的变式。在求这个 01 串时,要用一个数组记录每一个元素的前驱,这样最后用 dfs,才能找到最长子序列的每一个对应下标。
/*** Note: The returned array must be malloced, assume caller calls free().*/
char** getLongestSubsequence(char** words, int wordsSize, int* groups, int groupsSize, int* returnSize) {int pre[wordsSize+5],dp[wordsSize+5];for(int i=0;i<=wordsSize;++i)pre[i]=-1,dp[i]=1;for(int i=0;i<wordsSize;++i){for(int j=i+1;j<wordsSize;++j){if(groups[j]!=groups[i]){if(dp[j]<dp[i]+1)dp[j]=dp[i]+1,pre[j]=i;}}}int index=0,temp=0;for(int i=0;i<wordsSize;++i){if(dp[i]>temp)temp=dp[i],index=i;}// printf("temp=%d index=%d\n",temp,index);char **ans=(char**)malloc(sizeof(char*)*(temp+5));for(int i=0;i<=temp;++i)ans[i]=(char*)malloc(sizeof(char)*12);int t=temp-1;while(index!=-1){strcpy(ans[t],words[index]);// printf("t=%d ans=%s word=%s index=%d pre=%d\n",t,ans[t],words[index],index,pre[index]);index=pre[index];t--;}*returnSize=temp;return ans;
}