如何编写Python循环实现指定序列生成，值为负或已存在时切换逻辑

问题核心修正点 #

• 现有代码的两个核心问题：一是初始值不符合规则要求的x₀=0，二是缺少高效的重复值校验逻辑
• 重复值检测推荐用set集合实现，查询时间复杂度为O(1)，远优于列表的遍历查询

可直接运行的实现代码 #

# 初始化：x0=0，初始序列和查重集合
sequence = [0]
exists = {0}

# 计算n从1到100的序列值（x0对应n=0，共生成101个元素）
for n in range(1, 101):
    prev = sequence[-1]
    candidate = prev - n
    # 同时满足大于0、未存在于序列两个条件则用减法结果，否则用加法结果
    if candidate > 0 and candidate not in exists:
        cur = candidate
    else:
        cur = prev + n
    sequence.append(cur)
    exists.add(cur)

# 打印最终序列
print(sequence)

逻辑说明 #

1. 初始化阶段直接按规则设置首个元素x₀=0，同时创建集合exists同步存储已生成的序列值，用于快速查重
2. 循环从n=1运行到n=100，每轮先计算减法候选值，校验两个触发条件：
   • 候选值大于0
   • 候选值未在已有序列中出现
3. 条件不满足时自动切换为加法逻辑，生成的新值同时存入序列和查重集合

适配numpy的实现版本（如果你需要继续使用numpy） #

import numpy as np

# 初始化101个元素的数组（对应n=0到n=100），x0=0
x = np.zeros(101, dtype=int)
exists = {0}

for n in range(1, 101):
    candidate = x[n-1] - n
    if candidate > 0 and candidate not in exists:
        x[n] = candidate
    else:
        x[n] = x[n-1] + n
    exists.add(x[n])

print(x.tolist())

内容的提问来源于stack exchange，提问作者Mackenzie Watts