Python无循环/推导式处理输入遇ValueError，求合规解法

问题概述 #

运行代码时触发ValueError: invalid literal for int() with base 10: '3 -1 1 14'，需在禁止使用循环、列表/集合/字典推导式的前提下，处理空格分隔的整数输入，计算非负整数的平方和，满足指定的输入输出规则。

输入示例 #

4
3
3 -1 1 14
2
-5 10
1
0
5
-1 -2 3 4 -5

题目要求 #

• 计算给定整数的平方和，排除负数；
• 第一行输入整数N（1≤N≤100），表示测试用例数量；
• 每个测试用例包含两行：第一行是整数X（0<X≤100），第二行是X个空格分隔的整数Yn（-100≤Yn≤100）；
• 接收所有输入后统一输出结果，测试用例输出间无空行，使用标准流输入输出。

Python实现规则 #

• 单文件源码，必须包含main函数；
• 禁止使用for/while循环或任何列表/集合/字典推导式；
• 兼容Python 3.11及以上版本。

错误代码分析 #

你提供的代码存在以下核心问题：

def helper2(n: int, sum: int):
    m = int(input()).split()  # 错误：带空格的字符串无法直接转整数，触发ValueError
    if m > 0:  # 错误：列表与整数比较逻辑无效
        sum += m * m  # 错误：列表不能直接做乘法运算
    else:
        sum += 0
    helper2(n - 1, sum)

def helper(n: int):
    if n == 0:
        return 0
    m = int(input())
    sum = 0
    helper2(m, sum)  # 错误：递归无返回值，sum无法被更新
    helper(n - 1)
    return sum

def main():
    n = int(input())
    sum = helper(n)
    print(sum)

if __name__ == "__main__":
    main()

1. ValueError根源：int(input()).split()试图将包含空格的输入字符串直接转为整数，必然触发类型转换错误；
2. 逻辑错误：将列表与整数比较、对列表执行乘法运算，均属于无效操作；
3. 递归缺陷：递归函数无返回值，无法传递累加的平方和，最终返回的sum始终为初始值0。

正确实现方案 #

核心思路是通过递归逐段处理输入字符串，避免使用循环和推导式：

def process_single_number(remaining_str: str, count: int, current_sum: int) -> int:
    """递归处理单个测试用例的数字行，返回平方和"""
    if count == 0:
        return current_sum
    # 手动查找空格位置，分割出第一个数字
    space_pos = remaining_str.find(' ')
    if space_pos != -1:
        num_str = remaining_str[:space_pos]
        rest_str = remaining_str[space_pos+1:].strip()
    else:
        num_str = remaining_str
        rest_str = ''
    # 转换数字并累加非负数的平方
    num = int(num_str)
    if num >= 0:
        current_sum += num ** 2
    # 递归处理剩余数字
    return process_single_number(rest_str, count - 1, current_sum)

def process_test_cases(case_count: int, result_str: str) -> None:
    """递归处理所有测试用例，最终统一输出结果"""
    if case_count == 0:
        print(result_str.strip())
        return
    # 读取当前测试用例的数字个数X
    x = int(input())
    # 读取X个数字的行
    num_line = input().strip()
    # 计算当前用例的平方和
    case_result = process_single_number(num_line, x, 0)
    # 拼接结果字符串（避免使用列表存结果）
    new_result_str = f"{result_str}{case_result}\n"
    # 递归处理下一个测试用例
    process_test_cases(case_count - 1, new_result_str)

def main():
    total_cases = int(input())
    process_test_cases(total_cases, "")

if __name__ == "__main__":
    main()

代码说明 #

1. process_single_number：通过字符串find方法手动分割出第一个数字，判断非负后累加平方，递归处理剩余字符串，直到完成X个数字的计算；
2. process_test_cases：递归遍历每个测试用例，将每个用例的结果拼接成字符串，所有用例处理完成后统一输出；
3. 无列表/循环依赖：全程仅使用字符串操作和递归，完全符合题目禁止循环、推导式的要求。

测试输出 #

对应上述输入示例，输出为：

206
100
0
25

内容的提问来源于stack exchange，提问作者Titiksha Gupta