Pythagorean Triples—CF1487D
思路
联立 a 2 = b + c 2 a^2 = b + c^2 a2=b+c2、 a 2 + b 2 = c 2 a^2 + b^2 = c^2 a2+b2=c2 得:
a 2 = 2 ∗ c − 1 a^2 = 2 * c - 1 a2=2∗c−1、 b = c − 1 b = c - 1 b=c−1。
对于一个固定的 a a a, b b b、 c c c 的值都是固定的,只要满足 a ≤ b a\le b a≤b 和 c ≤ n c\le n c≤n 即可。
使用二分可以求出对应的 c c c 的值满足条件的 a a a 的取值范围,另外要排除 a = 1 , 2 a = 1, 2 a=1,2 这两个值,因为这时候 a > b a > b a>b。
C o d e Code Code
#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;int n;// c的值是否符合条件(先不考虑奇数这个条件)
int judge(int a) {if ((a * a + 1) / 2 <= n&& a <=(a * a + 1) / 2 - 1) {return 1;}return 0;
}void solve() {cin >> n;cout << " ";if (n <= 2) {cout << 0 << "\n";return;}// 二分求a的最大值(先不考虑得到的c是否是整数)int l = 3, r = n;while (l < r) {int mid = (l + r + 1) / 2;if (judge(mid)) {l = mid;} else {r = mid - 1;}}if (l % 2 == 0) {l --;}// 现在满足条件的a的取值范围是[3, l]中的所有奇数if (judge(l)) {cout << l / 2 << "\n";} else {cout << "0\n";}
}signed main() {ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);int T = 1;cin >> T; cin.get();while (T --) solve();return 0;
}