LeetCode第182场周赛：5371. 找到所有好字符串（数位dp+kmp）

5371. 找到所有好字符串

难度困难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/shu-wei-dp-kmpqian-zhui-shu-zu-java-by-henrylee4/

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;
    }
}