超级码力初赛第一场第四题 —— 对称前后缀 20-9-1
补充上上篇博客,这题题目的例子出的也不是很好。
我来解释一下题目的权值含义:每个字符串的权值是从端到端的回文字符的个数。
例如“aa”,权值为2,“aba”,权值为3,“aaa”,权值也为3,“abcba”,权值为5,“abdfba”,权值为2。
解答
其实这道题是比较简单的动态规划题目,只需遍历每一个字串即可,对于每一个长度大于1字串(长度为1的字串默认权值为1),如果该字串的首字符等于尾字符,则有权值,其权值取决于去掉首尾字符的字串权值,如果去掉首尾字符后的字串权值等于字串长度,说明该字串完全回文,权值等于该字串权值加2,否则加1。
假设dp[a][b]
表示索引a到b的字串的权值。
那么我们可以列出下面的式子:
则有dp[a][b] =
0 (首尾字符串不相等)
1 (a == b)
dp[a-1][b-1] + 2(首尾字符串相等 ——> 去掉首尾字符串 -->子串权值 == 子串长度)
dp[a-1][b-1] + 1 (首尾字符串相等 ——> 去掉首尾字符串 -->子串权值 != 子串长度)
特殊情况: 当字符串长度为2,去掉首尾字符串后长度为0,此时子串权值和子串长度都为0,条件,权值= 0+2。
代码实现如下:
public static long suffixQuery(String s){
int n = s.length();
long result = n;
//dp[a][b]表示字符串索引a到b的权值
long[][] dp = new long[n][n];
//单个字符字串权值为1
for(int i = 0 ; i<n ; i++){
dp[i][i] = 1;
}
//len为附加长度,最大为字串长度减一
for(int len=1; len < n; len++){
//从索引0的字符开始
for(int i = 0; i+len <n; i++){
//如果该字串的第一个字符和最后一个字符相等,那么该字符串有权值,权值与其内部字符串有关
if(s.charAt(i)== s.charAt(i+len)){
//内部字符串完全对称时权值为字符串长度
int length = len - 1;
//如果内部字符串权值等于最大权值说明对称,此字符串的权值等于其内部字符串权值加2
//否则说明内部字符串不是完全对称,此时字符串的权值等于其内部字符串权值加1
if(dp[i+1][i+len-1] == length)
dp[i][i+len] = dp[i+1][i+len-1]+2;
else
dp[i][i+len] = dp[i+1][i+len-1]+1;
}
result += dp[i][i+len];
}
}