思路
- 读取n,计算出n位数可以有多少不同的组合方案(每一位都可以取0~9 共10种情况,所以能组合出10 ^ n种方案)
- 读取m以及接下来的m个不能用作组合开头的字符串(以下称为 “非法开头” )
- 针对每个 “非法开头”,计算出以该非法开头的组合有多少种?
- 从总的组合方案数(10 ^ n)中减去它们(“以该非法开头的组合有多少种”),就得到了最终答案
注意点
- 根据第三个测试用例可以得知:
“1”这个非法开头可以得到的组合方案,其实已经包含了“12”、“123”开头的组合方案
- 因此如果不处理这种情况,就会重复减去“12”、“123”开头的组合方案的数量,导致最终结果比正确答案要小一些
如何处理?
- 先对所有的非法开头按照它们自身长度进行排序
- 遍历排序后的非法开头数组,判断在当前这个非法开头之前是否有某个非法开头恰好是它的前缀
- 如果没有,计算当前这个非法开头的组合方案数并从总的方案数减去它
- 如果有,那么跳过当前这个非法开头,直接去判断下一个非法开头,以避免重复计算
代码
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
long allCount = (long)Math.pow(10,n);
if(m == 0){
System.out.println((long)Math.pow(10,n));
return;
}
String[] str = new String[m];
for (int i = 0; i < m; i++) {
str[i] = sc.next();
}
Arrays.sort(str,(s1,s2) -> s1.length() - s2.length());
for (int i = 0; i < m; i++) {
boolean flag = true;
for (int j = 0; j < i; j++) {
if (str[i].startsWith(str[j])) {
flag = false;
break;
}
}
if (flag) allCount -= (long) Math.pow(10, n - str[i].length());
}
System.out.println(allCount);
}
}