1287: [蓝桥杯2016初赛]四平方和-----二分查找

题目

题目链接
四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去，就正好可以表示为4个数的平方和。
比如：
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2（符号表示乘方的意思）
对于一个给定的正整数N，可能存在多种平方和的表示法。
要求你对4个数排序：0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

分析

刚开始写的是四层循环，但是超时，后来想到最后一个数可以用二分查找，另外前三层循环也可以缩小范围

#include<bits/stdc++.h>
#include<cstring>
using namespace std;
#define ll long long
ll bin(ll l,ll r,ll m){//二分的边界一定要确定好 
	ll mm;
	while(l<=r){
		ll mid=(l+r)/2;
		mm=mid*mid;
		if(mm==m){
			return mid;
		}else if(mm<m){
			l=mid+1;
		}else{
			r=mid-1;
		}
	}
	return -1;
}
int main(){
	ll n;
	ll flag=0;
	ll a;
   // a=bin(0,702244,702244);
   //cout<<a<<endl;
	while(cin>>n){
		flag=0;
		for(int i=0;i<=n;i++){
			for(int j=i;j*j*3<=n-i*i;j++){
				for(int k=j;k*k*2<=n-i*i-j*j;k++){//限制边界，因为想想，四个数是按照非递减顺序的 
					a=bin(0,n-i*i-j*j-k*k,n-i*i-j*j-k*k);   //最后一个数使用了二分来查找 
					if(a==-1){
					continue;
					}else{
						cout<<i<<" "<<j<<" "<<k<<" "<<a<<endl;
						flag=1;
						break;
					}
				}
				if(flag==1){
						break;
					}
			}
			if(flag==1){
						break;
					}
		}
		
	}
}