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

#!/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")

 

 

 

你可能感兴趣的:(python3)