Educational Codeforces Round 104 (Rated for Div. 2) D. Pythagorean Triples

题目

链接

思路

首先由c×c=a×a+b×b和c=a×a-b可得： a×a = 2×c-1,b = c-1;
因为a×a=2×c-1 所以a肯定是奇数 并且a×a = 2×c-1 <= 2×n-1
思路一：
a×a=2×c-1 只要是a是奇数 对于每一个a>=3的奇数 都存在a b c；
所以二分找到最大的a满足a×a<=2×n-1。
如果是偶数 --得到奇数。这时候ans/2就是相应的个数(奇数偶数除二对半分)
思路二：
t循环外面vector容器存下所有满足情况。然后对于每一个样例，利用函数
lower_bound() 来寻找最大的c满足c<=n的位置，自然下标就是个数啦

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<cstdio>
#include<cmath>
#include<stdlib.h>

using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int maxn = 1e6+50;

int main(){
	int t;
	cin >> t;
	while(t--){
		ll n;
		scanf("%lld",&n);
		ll l = 3,r = sqrt(n*2-1);
		ll ans;
		if(n < 5){
			puts("0");
			continue;
		}
		while(l <= r){
			ll mid = (l+r)/2;
			if(mid*mid <= 2*n-1){
				ans = mid;
				l = mid+1;
			}
			else r = mid-1;
		} 
		if(ans%2 == 0)	ans--;
		printf("%lld\n",ans/2);
	}
	return 0;
}

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<cstdio>
#include<cmath>
#include<stdlib.h>

using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int maxn = 1e6+50;

int main(){
	vector<ll>vec;
	for(int a = 3;a <= 1e9;a+=2){
		if((a*a+1)/2 > 1e9)	break;
		vec.push_back((a*a+1)/2);
	}
	int t;
	cin >> t;
	while(t--){
		ll n;
		cin >> n;
		int p = lower_bound(vec.begin(),vec.end(),n)-vec.begin();
		if(vec[p] == n)	p++;
		cout << p << endl;
	}
	return 0;
}