一、题目描述:
给定非空字符串s,将该字符串分割成一些子串,使每个子串的ASCII码值的和均为水仙花数。
1、若分割不成功,则返回0。
2、若分割成功且分割结果不唯一,则返回-1。
3、若分割成功且分割结果唯一,则返回分割后子串的数目。
输入描述:1、输入字符串的最大长度为200。
输出描述:根据题目描述中情况,返回相应的结果。
补充说明:“水仙花数Q“是指一个三位数,每位上数字的立方和等于该数字本身,如371是"水仙花数",因为:371=33+73+1^3。
示例1,输入:,abc,输出:0,说明:分割不成功。
示例2,输入:f3@d5a8,输出:-1,说明:分割成功但分割结果不唯一,可以分割为两组,一组"f3"和"@d5a8",另外一组"f3@d5"。
二、解法
首先枚举得到所有可能子串
使用递归回溯搜索分割组合
全局计数器控制输出逻辑,避免重复统计
时间复杂度为O(n³),主要来自子串验证和矩阵构建
public class Solution {
static int sample = 0;
public static boolean isValid(String s){
int acc = 0;
for(char c : s.toCharArray()){
acc += c;
}
int check = acc;
while(acc!=0){
check -= Math.pow(acc%10,3);
acc /= 10;
}
return check == 0;
}
public static int search(boolean[][] state,int cury,int cnt){ //按索引逆序循迹
if(cury < 0){
sample += 1;
return cnt;
}
for(int i = 0; i < state.length;++i){
if(state[i][cury]){
search(state,i-1,cnt+1);
}
}
return cnt;
}
public static void main(String[] args) {
String s = "f3@d5a8";
int n = s.length();
boolean[][] state = new boolean[n][n];
for(int i = 0;i < n;++i){
for(int j = i + 1;j < n;++j){
if(isValid(s.substring(i,j+1))){
System.out.println(s.substring(i,j+1));
state[i][j] = true;
}
}
}
int cnt = 0;
for(int i = 0;i < n;++i){
if(state[i][n-1]){
cnt = search(state,i-1,1); //已经有一段,故cnt=1,i-1则是开始下一段的匹配
}
}
if(sample > 1){
System.out.println(-1);
}else if(sample == 1){
System.out.println(cnt);
}else{
System.out.println(0);
}
}
}