Funky Numbers(二分)

本文介绍了一种算法,用于判断一个给定的整数是否可以表示为两个三角形数之和。通过使用二分查找的方法优化了遍历过程,提高了算法效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

As you very well know, this year's funkiest numbers are so called triangular numbers (that is, integers that are representable as , where k is some positive integer), and the coolest numbers are those that are representable as a sum of two triangular numbers.

A well-known hipster Andrew adores everything funky and cool but unfortunately, he isn't good at maths. Given number n, help him define whether this number can be represented by a sum of two triangular numbers (not necessarily different)!

Input

The first input line contains an integer n (1 ≤ n ≤ 109).

Output

Print "YES" (without the quotes), if n can be represented as a sum of two triangular numbers, otherwise print "NO" (without the quotes).

Examples
Input
256
Output
YES
Input
512
Output
NO
Note

In the first sample number .

In the second sample number 512 can not be represented as a sum of two triangular numbers.

两个循环搜索会超时,所以第二个使用二分搜索

#include <iostream>
#include <math.h>
using namespace std;
typedef long long ll;
int main(){
    ll n,i,A,B,mid,r,l;
    cin>>n;
    int flag=0;
    for(i=1;i<sqrt(2*n);i++){
        A=i*(i+1)/2;
        l=i;r=sqrt(2*n);mid=(l+r)/2;
        while(l<=r){
            B=mid*(mid+1)/2;
            if(A+B<n){
                l=mid+1;
                mid=(l+r)/2;
            }
            else if(A+B>n){
                r=mid-1;
                mid=(l+r)/2;
            }
            else{
                flag=1;
                break;
            }
            
        }
        if(flag==1)
            break;
    }
    if(flag==1)
        cout<<"YES"<<endl;
    else
        cout<<"NO"<<endl;
    
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值