PTA Python 7-3 s=1+2+3+...+n直到s＞=m

s=1+2+3+...+n，当s刚超过某个数时，停止运算。求n和s的值

输入格式:

例如：输入1000

输出格式:

输出格式 ：1035=1+2+...+45

输入样例:

在这里给出一组输入。例如：

1000

输出样例:

在这里给出相应的输出。例如：

1035=1+2+...+45

输入样例:

在这里给出一组输入。例如：

10000

输出样例:

在这里给出相应的输出。例如：

10011=1+2+...+141

答案:

x = eval(input())  # 1035=1+2+...+45
i = 1
s = 0
while s < x:
    s += i
    i += 1
if s > 3:
    print(f"{s}=1+2+...+{i-1}")
elif s==1:
    print(f"{s}=1")
elif s==2:
    print(f"{s}=1+2")
elif s==3:
    print(f"{s}=1+2")
elif 3<s<6:
    print(f"{s}=1+2+3")