python3：判断一个数是否可以拆解为n个连续的正整数的和

#!/usr/bin/python#--**--coding:utf8--*--'''输入一个数num， 1 < num < 2 ** 30 + 1，判断这个数是否能拆解为连续的整数和，如输入5 #5 = 2 + 3输出’YES‘'''import sysimport mathnum = int(sys.stdin.readline().strip())...

qq_41572664

1376人浏览 · 2020-05-08 17:03:07

qq_41572664 · 2020-05-08 17:03:07 发布

#!/usr/bin/python
#--**--coding:utf8--*--
'''
输入一个数num， 1 < num < 2 ** 30 + 1，
判断这个数是否能拆解为连续的整数和，
如输入5 #5 = 2 + 3
输出’YES‘
'''

import sys
import math

num = int(sys.stdin.readline().strip())
if 1 < num < 2 ** 30 + 1:
    flag = 0
    if num % 2 == 1:
        print('YES')
        sys.exit(0)
    max_num = math.ceil(num / 2)
    for y in range(1, max_num + 1):#假设首项y1,末项yn，能满足条件
        a = 1           #(y1 + yn) * n / 2 = num，yn = (y1 + n - 1) * 1
        b = 1 - 2 * y   #其中y1=y解一元二次方程
        c = -2 * num
        n1 = (-b + math.sqrt( b ** 2 - 4 * a * c)) / （2 * a）#求根公式
        n2 = (-b + math.sqrt( b ** 2 - 4 * a * c)) / （2 * a）#求根公式
        if str(n1).strip('.0').isdigit() \
          or str(n2).strip('.0').isdigit(): #项数为正整数，则表示假设成立
            print('YES')
            break
    else:
        print("NO")