难度困难15收藏分享切换为英文关注反馈
给你两个长度为 n
的字符串 s1
和 s2
,以及一个字符串 evil
。请你返回 好字符串 的数目。
好字符串 的定义为:它的长度为 n
,字典序大于等于 s1
,字典序小于等于 s2
,且不包含 evil
为子字符串。
由于答案可能很大,请你返回答案对 10^9 + 7 取余的结果。
示例 1:
输入:n = 2, s1 = "aa", s2 = "da", evil = "b"
输出:51
解释:总共有 25 个以 'a' 开头的好字符串:"aa","ac","ad",...,"az"。还有 25 个以 'c' 开头的好字符串:"ca","cc","cd",...,"cz"。最后,还有一个以 'd' 开头的好字符串:"da"。
示例 2:
输入:n = 8, s1 = "leetcode", s2 = "leetgoes", evil = "leet"
输出:0
解释:所有字典序大于等于 s1 且小于等于 s2 的字符串都以 evil 字符串 "leet" 开头。所以没有好字符串。
示例 3:
输入:n = 2, s1 = "gx", s2 = "gz", evil = "x"
输出:2
提示:
s1.length == n
s2.length == n
1 <= n <= 500
1 <= evil.length <= 50
- 所有字符串都只包含小写英文字母。
思路:感觉是打周赛以来最难的一道压轴题了,比赛时考虑到了用数位dp,然而出现了一些问题,对于判定不含evil串的过程中出现了重复计数的问题,之后卡到了比赛结束,太菜了哇~~~
看了两篇讲解,觉得讲的不错,这里我直接贴出链接供大家(自己)学习参考:
https://leetcode-cn.com/problems/find-all-good-strings/solution/kmpshang-de-dpc-by-zhu-mang-4/
class Solution {
public int findGoodStrings(int n, String s1, String s2, String evil) {
int mod=(int)1e9+7;
int len=evil.length();
//第二维为0表示s1和s2都有限制,1表示s1有限制,
//2表示s2有限制,3表示s1和s2都没限制。
long[][][] dp=new long[n+1][4][len+1];
for(int i=0;i<len;i++) {
dp[n][0][i]=1;
dp[n][1][i]=1;
dp[n][2][i]=1;
dp[n][3][i]=1;
}
char[] c=evil.toCharArray();
int[] prefix=calcuPrefix(c); //计算前缀数组
for(int i=n-1;i>=0;i--) {
for(int j=0;j<len;j++) {
for(char k=s1.charAt(i);k<=s2.charAt(i);k++) {
int state=0;
if(k==s1.charAt(i) && k==s2.charAt(i))
state=0;
else if(k==s1.charAt(i))
state=1;
else if(k==s2.charAt(i))
state=2;
else
state=3;
dp[i][0][j]+=dp[i+1][state][getNext(prefix,c,k,j)];
dp[i][0][j]%=mod;
}
for(char k=s1.charAt(i);k<='z';k++) {
int state=k==s1.charAt(i)?1:3;
dp[i][1][j]+=dp[i+1][state][getNext(prefix,c,k,j)];
dp[i][1][j]%=mod;
}
for(char k='a';k<=s2.charAt(i);k++) {
int state=k==s2.charAt(i)?2:3;
dp[i][2][j]+=dp[i+1][state][getNext(prefix,c,k,j)];
dp[i][2][j]%=mod;
}
for(char k='a';k<='z';k++) {
int state=3;
dp[i][3][j]+=dp[i+1][state][getNext(prefix,c,k,j)];
dp[i][3][j]%=mod;
}
}
}
return (int)dp[0][0][0];
}
private int[] calcuPrefix(char[] c) {
int n=c.length;
int[] next=new int[n];
for(int j=0,i=1;i<n;i++) {
while(j>0 && c[i]!=c[j])
j=next[j-1];
if(c[i]==c[j])
j++;
next[i]=j;
}
return next;
}
private int getNext(int[] next,char[] p,char c,int j) {
while(j>0 && c!=p[j])
j=next[j-1];
if(c==p[j])
j++;
return j;
}
}