4301 截断数列（枚举）

1. 问题描述：

给定一个由 n 位数字组成的序列 a1a2…an。其中，每个数字都是 0∼9 之一。请你判断，能否将数列从中间截断为两个或更多个非空部分，要求每一部分的各位数字之和都相等。例如，350178 可以截断为 3 个部分 350、17、8，并且满足 3+5+0=1+7=8。

输入格式

第一行包含一个整数 n。第二行包含 n 个数字 a1，a2，…，an，数字之间不含空格。

输出格式

如果可以按要求截断数列，则输出 YES，否则输出 NO。

数据范围

前 6 个测试点满足 2 ≤ n ≤ 10。
所有测试点满足 2 ≤ n ≤ 100，0 ≤ ai ≤ 9。

输入样例1：

5
73452

输出样例1：

YES

输入样例2：

4
1248

输出样例2：

NO
来源：https://www.acwing.com/problem/content/4304/

2. 思路分析：

分析题目可以知道数据规模不是特别大，所以我们可以枚举所有可能的段数，最少分成2段，最多分为n段，然后枚举当前段数为i的情况下判断是否可以分解成每一段相等的部分即可，如果存在前导0分为前一段或者后一段都可以，这里可以规定分为后一段，如果加上当前数字之后大于了sum / k之后说明当前分解为k段是无解的，如果等于sum / k那么判断下一段即可。

3. 代码如下：

class Solution:
    def process(self):
        n = int(input())
        s = input()
        _sum = 0
        for i in range(n):
            _sum += int(s[i])
        # 枚举段数
        for k in range(2, n + 1):
            # 当前能够分成k段
            if _sum % k == 0:
                _s, t = _sum // k, 0
                # flag用来标记当前是否存在不符合要求的一段
                flag = 1
                for i in range(n):
                    # 加上当前这个位置的数字
                    t += int(s[i])
                    # 大于0了之后说明不满足条件
                    if t > _s:
                        flag = 0
                        break
                    # 新的一段开始了
                    elif t == _s:
                        t = 0
                if flag: return "YES"
        return "NO"


if __name__ == "__main__":
    print(Solution().process())