【华为OD】D卷真题100分：字符串分割 python代码实现[思路+代码]

【华为OD】2024年C、D卷真题集：最新的真题集题库 C/C++/Java/python/JavaScript

【华为OD】2024年C、D卷真题集：最新的真题集题库 C/C++/Java/python/JavaScript-CSDN博客

JS、C、Java、Python、C++代码实现：

【华为OD】D卷真题100分：字符串分割 JavaScript代码实现[思路+代码]-CSDN博客

 【华为OD】D卷真题100分：字符串分割 C语言代码实现[思路+代码]-CSDN博客

【华为OD】D卷真题100分：字符串分割 Java代码实现[思路+代码]-CSDN博客 

【华为OD】D卷真题100分：字符串分割 python代码实现[思路+代码]-CSDN博客 

【华为OD】D卷真题100分：字符串分割 C++代码实现[思路+代码]-CSDN博客 

题目描述：

给定非空字符串s，将该字符串分割成一些子串，使每个子串的ASCII码值的和均为水仙花数。
1、若分割不成功，则返回0
2、若分割成功且分割结果不唯一，则返回-1
3、若分割成功且分割结果唯一，则返回分割后子串的数目

输入描述

1、输入字符串的最大长度为200

输出描述

根据题目描述中情况，返回相应的结果

示例1

输入输出示例仅供调试，后台判题数据一般不包含示例

输入

abc

输出

0

说明

分割不成功

示例2

输入输出示例仅供调试，后台判题数据一般不包含示例

输入

f3@d5a8

输出

-1

说明

分割成功但分割结果不唯一，可以分割为两组，一组"f3"和"@d5a8"，另外一组"f3@d5"和"a8"

示例3

输入输出示例仅供调试，后台判题数据一般不包含示例

输入

AXdddF

输出

2

说明

成功分割且分割结果唯一，可以分割“AX"(153)和"dddF"(370)成两个子串

备注

“水仙花数”是指一个三位数，每位上数字的立方和等于该数字本身，如371是“水仙花数”，因为：371 = 3^3 + 7^3 + 1^3

----------------------

     wumu1024                                                         
                                                            
              +---+                                                          
  3            |   |       ++                               +       +---|   
  |           |   | 3      +                6               +  |   +   |        +
  |      +     |   |       +         +                      +    |  +   |       +
  |      +    |   +---+    +        +        +++++          +   +  +   |        +
  |      +    | +      |   +   +----+        |   |          +   +  +   |        +
  |      +  3 | +      |   +   +    +      2 |   |     2    +   +  +   |        +
  |      +    | +      |   +   +    +        |   |          +   +  +   |        +
  |      +---+ +     |    |  |    +    ----+   |   +---+    |  |  +   |         +
  |      |     +     |    |  |    +    |       |   |   |    |  |  +   |         +
  |    1 |     +     | 8  |  |    +  1 |   |    | 1 |   | 1 |   |  +   |        +
  |      |     +     |    |  |    +    |   |    |   |   |   |   |  +   |        +
  |  +---+     +     +---+   |    ++---+    ++   +---+   +---+   |  +   |        +
  |  |         +         |   |    |         ++              |   |  |+   |        +
  |0 |         +         | 0 |  0 |         ++              | 0 |  |+   |        +
  |  |         +         |   |    |         ++              |   |  |+   |        +
  +---+         +          +-------+                       +---+| +|+   |        +
                +                                                    +   |        +
    0   1   2   3   4   5   6   7   8   9  10  11  12 + v:    w  u m    u 1 0 2 4
 

题目解析：

        先把水仙花数的函数按规则设计出来， 就是一个三位数等于其每个数字的立方和

然后再使用回溯算法，递归的方式来进行组合，求出合适的分割，来匹配这个函数即可

其实满足水仙花数的3位数经过计算之后，也就是这几个数值：153   370  371  407，也可以直接判断这几个值是否和对应的和相等

代码实现：

sub_res = []


def is_daffodil(s, left, right):
    sum = 0
    for i in range(left, right + 1):
        sum += ord(s[i])
    return sum == 407 or sum == 371 or sum == 370 or sum == 153


def get_find(s, index, res):
    if len(s) == index:
        res.append(list(sub_res))
        return None
    for i in range(index, len(s)):
        if is_daffodil(s, index, i):
            sub_res.append(s[index: i - index + 1])
            get_find(s, i + 1, res)
            sub_res.pop()


def result(res):
    if len(res) == 0:
        return 0
    elif len(res) > 1:
        return -1
    else:
        return len(res[0])


if __name__ == '__main__':
    s = input().strip()
    res = []
    get_find(s, 0, res)
    print(result(res))